Skip to content
Snippets Groups Projects
Commit aa7dd77a authored by Laurent Smet's avatar Laurent Smet
Browse files

[IMP] account: Allow mocking a datetime in mocked_today

X-original-commit: a9a1d8e5348010eca7de15199faae1f2469d96ff
parent ddf9f0f4
Branches
Tags
No related merge requests found
......@@ -6,6 +6,7 @@ from odoo.tests.common import SavepointCase, HttpCase, tagged, Form
import time
import logging
import datetime
from contextlib import contextmanager
from unittest.mock import patch
......@@ -993,14 +994,22 @@ class AccountTestInvoicingCommon(SavepointCase):
'''
if isinstance(forced_today, str):
forced_today = fields.Date.from_string(forced_today)
forced_today_date = fields.Date.from_string(forced_today)
forced_today_datetime = fields.Datetime.from_string(forced_today)
elif isinstance(forced_today, datetime.datetime):
forced_today_datetime = forced_today
forced_today_date = forced_today_datetime.date()
else:
forced_today_date = forced_today
forced_today_datetime = datetime.datetime.combine(forced_today_date, datetime.time())
def today(*args, **kwargs):
return forced_today
return forced_today_date
with patch.object(fields.Date, 'today', today):
with patch.object(fields.Date, 'context_today', today):
yield
with patch.object(fields.Datetime, 'now', return_value=forced_today_datetime):
yield
@classmethod
def init_invoice(cls, move_type, partner=None, invoice_date=None):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment