mdsfred/cashbook_planner
2
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages 1 Wiki Activity Actions

optimize call of run_booking() + tests

Browse source
This commit is contained in:
Frederik Jaeckel 2024-03-02 20:46:58 +01:00
parent 95f06e8e26
commit d800b17561
2 changed files with 38 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

42
tests/planner.py
View file

@ -3,6 +3,7 @@
# The COPYRIGHT file at the top level of this repository contains the
# full copyright notices and license terms.
from unittest.mock import MagicMock
from trytond.tests.test_tryton import with_transaction
from trytond.pool import Pool
from trytond.transaction import Transaction
@ -247,7 +248,9 @@ class PlannerTestCase(object):
def test_planner_run_cronjobs(self):
""" create job, check cron
"""
Planner = Pool().get('cashbook.planner')
pool = Pool()
Planner = pool.get('cashbook.planner')
IrDate = pool.get('ir.date')
job = self.prep_create_job()
self.assertEqual(
@ -255,18 +258,31 @@ class PlannerTestCase(object):
count=1, query_date=date(2022, 5, 1)), [
date(2022, 5, 1)])
with Transaction().set_context({
'nextrun_crondate': date(2022, 5, 24)}):
job, = Planner.search([])
self.assertEqual(job.nextrun[0].date, date(2022, 5, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
# job was not yet run after configure
IrDate.today = MagicMock(return_value=date(2022, 5, 24))
Planner.run_booking = MagicMock()
job, = Planner.search([])
self.assertEqual(job.nextrun[0].date, date(2022, 5, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
Planner.run_booking.assert_called_with([job])
with Transaction().set_context({
'nextrun_crondate': date(2022, 6, 1)}):
job, = Planner.search([])
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 7, 1))
# next call before due date - nothing should happen
IrDate.today = MagicMock(return_value=date(2022, 5, 30))
Planner.run_booking = MagicMock() # restart mock
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
Planner.run_booking.assert_not_called()
# next call at due date - calls booking and set due date
IrDate.today = MagicMock(return_value=date(2022, 6, 1))
Planner.run_booking = MagicMock() # restart mock
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 7, 1))
Planner.run_booking.assert_called_with([job])
IrDate.today = MagicMock(return_value=date.today())
# end PlannerTestCase