mirror of
https://github.com/OCA/contract.git
synced 2025-02-13 17:57:24 +02:00
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
# Copyright 2020 ACSONE SA/NV
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from odoo.addons.contract.tests.test_contract import TestContractBase
|
|
from odoo.addons.queue_job.tests.common import JobMixin
|
|
|
|
|
|
class TestContractQueueJob(TestContractBase, JobMixin):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestContractQueueJob, cls).setUpClass()
|
|
cls.contract3 = cls.contract2.copy()
|
|
|
|
def _get_related_invoices(self, contracts):
|
|
return (
|
|
self.env["account.invoice.line"]
|
|
.search(
|
|
[
|
|
(
|
|
"contract_line_id",
|
|
"in",
|
|
contracts.mapped("contract_line_ids.id"),
|
|
)
|
|
]
|
|
)
|
|
.mapped("invoice_id")
|
|
)
|
|
|
|
def test_contract_queue_job(self):
|
|
contracts = self.contract2 | self.contract3
|
|
job_counter = self.job_counter()
|
|
invoices = contracts._recurring_create_invoice()
|
|
self.assertFalse(invoices)
|
|
invoices = self._get_related_invoices(contracts)
|
|
self.assertFalse(invoices)
|
|
self.assertEqual(job_counter.count_created(), 2)
|
|
self.perform_jobs(job_counter)
|
|
invoices = self._get_related_invoices(contracts)
|
|
self.assertEqual(len(invoices), 2)
|
|
|
|
def test_contract_queue_job_1(self):
|
|
contracts = self.contract2
|
|
job_counter = self.job_counter()
|
|
count_all_jobs = job_counter.count_all()
|
|
self.assertEqual(job_counter.count_all(), count_all_jobs)
|
|
invoices_res = contracts._recurring_create_invoice()
|
|
self.assertTrue(invoices_res)
|
|
invoices = self._get_related_invoices(contracts)
|
|
self.assertEqual(invoices_res, invoices)
|
|
|
|
def test_contract_queue_job_2(self):
|
|
contracts = self.contract2 | self.contract3
|
|
job_counter = self.job_counter()
|
|
wizard = self.env["contract.manually.create.invoice"].create(
|
|
{"invoice_date": self.today, "contract_type": "purchase"}
|
|
)
|
|
wizard.create_invoice()
|
|
invoices = self._get_related_invoices(contracts)
|
|
self.assertFalse(invoices)
|
|
self.assertEqual(job_counter.count_created(), 2)
|
|
self.perform_jobs(job_counter)
|
|
invoices = self._get_related_invoices(contracts)
|
|
self.assertEqual(len(invoices), 2)
|