Julien CastiauxandRaphaël Collet
9a8cf18ddb
When testing cron triggers, it is common to get the newly created triggers in order to validate they exist and are scheduled are the right moment. This new helper is a context manager that capture all triggers or triggers created for a specific cron. The created triggers are accessible via the context's object `records` attribute. The various tests have been updated so they use that new helper. closes odoo/odoo#68163 Signed-off-by: Julien Castiaux <Julien00859@users.noreply.github.com> Co-authored-by: Raphaël Collet <rco@odoo.com>
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from unittest.mock import patch
|
|
|
|
from odoo import fields
|
|
from odoo.tests.common import TransactionCase, RecordCapturer
|
|
|
|
|
|
class CronMixinCase:
|
|
def capture_triggers(self, cron_id=None):
|
|
"""
|
|
Get a context manager to get all cron triggers created during
|
|
the context lifetime. While in the context, it exposes the
|
|
triggers created so far from the beginning of the context. When
|
|
the context exits, it doesn't capture new triggers anymore.
|
|
|
|
The triggers are accessible on the `records` attribute of the
|
|
returned object.
|
|
|
|
:param cron_id: An optional cron record id (int) or xmlid (str)
|
|
to only capture triggers for that cron.
|
|
"""
|
|
if isinstance(cron_id, str): # xmlid case
|
|
cron_id = self.env.ref(cron_id).id
|
|
|
|
return RecordCapturer(
|
|
model=self.env['ir.cron.trigger'].sudo(),
|
|
domain=[('cron_id', '=', cron_id)] if cron_id else []
|
|
)
|
|
|
|
|
|
class TestIrCron(TransactionCase, CronMixinCase):
|
|
|
|
def setUp(self):
|
|
super(TestIrCron, self).setUp()
|
|
|
|
self.cron = self.env['ir.cron'].create({
|
|
'name': 'TestCron',
|
|
'model_id': self.env.ref('base.model_res_partner').id,
|
|
'state': 'code',
|
|
'code': 'model.search([("name", "=", "TestCronRecord")]).write({"name": "You have been CRONWNED"})',
|
|
'interval_number': 1,
|
|
'interval_type': 'days',
|
|
'numbercall': -1,
|
|
'doall': False,
|
|
})
|
|
self.test_partner = self.env['res.partner'].create({
|
|
'name': 'TestCronRecord'
|
|
})
|
|
self.test_partner2 = self.env['res.partner'].create({
|
|
'name': 'NotTestCronRecord'
|
|
})
|
|
|
|
def test_cron_direct_trigger(self):
|
|
self.assertFalse(self.cron.lastcall)
|
|
self.assertEqual(self.test_partner.name, 'TestCronRecord')
|
|
self.assertEqual(self.test_partner2.name, 'NotTestCronRecord')
|
|
|
|
def patched_now(*args, **kwargs):
|
|
return '2020-10-22 08:00:00'
|
|
|
|
with patch('odoo.fields.Datetime.now', patched_now):
|
|
self.cron.method_direct_trigger()
|
|
|
|
self.assertEqual(fields.Datetime.to_string(self.cron.lastcall), '2020-10-22 08:00:00')
|
|
self.assertEqual(self.test_partner.name, 'You have been CRONWNED')
|
|
self.assertEqual(self.test_partner2.name, 'NotTestCronRecord')
|