changeset fc5176c72a23 in modules/product_price_list_dates:6.4
details:
https://hg.tryton.org/modules/product_price_list_dates?cmd=changeset&node=fc5176c72a23
description:
Use context date as default pattern
issue11359
review417411003
(grafted from 3ca1623dd74c4c1a9aff760d2b43d5dabf82a1ee)
diffstat:
product.py | 9 +++++++++
tests/test_module.py | 21 +++++++++++++++++++++
2 files changed, 30 insertions(+), 0 deletions(-)
diffs (63 lines):
diff -r 9b01092ec750 -r fc5176c72a23 product.py
--- a/product.py Mon May 02 17:35:04 2022 +0200
+++ b/product.py Tue Jun 21 10:13:53 2022 +0200
@@ -3,6 +3,7 @@
from trytond.model import ModelView, fields
from trytond.pool import Pool, PoolMeta
from trytond.pyson import Eval, If
+from trytond.transaction import Transaction
class PriceList(metaclass=PoolMeta):
@@ -21,6 +22,14 @@
def open_lines(cls, price_lists):
pass
+ def compute(
+ self, party, product, unit_price, quantity, uom, pattern=None):
+ context = Transaction().context
+ pattern = pattern.copy() if pattern is not None else {}
+ pattern.setdefault('date', context.get('date'))
+ return super().compute(
+ party, product, unit_price, quantity, uom, pattern=pattern)
+
class PriceListLine(metaclass=PoolMeta):
__name__ = 'product.price_list.line'
diff -r 9b01092ec750 -r fc5176c72a23 tests/test_module.py
--- a/tests/test_module.py Mon May 02 17:35:04 2022 +0200
+++ b/tests/test_module.py Tue Jun 21 10:13:53 2022 +0200
@@ -8,6 +8,7 @@
CompanyTestMixin, create_company, set_company)
from trytond.pool import Pool
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
+from trytond.transaction import Transaction
class ProductPriceListDatesTestCase(CompanyTestMixin, ModuleTestCase):
@@ -72,5 +73,25 @@
None, None, Decimal(10), 1, None, pattern={'date': yesterday}),
Decimal(9))
+ @with_transaction()
+ def test_price_list_with_context_date(self):
+ "Test price list with context date"
+ pool = Pool()
+ Date = pool.get('ir.date')
+
+ today = Date.today()
+ tomorrow = today + datetime.timedelta(days=1)
+
+ price_list = self.create_price_list('start_date', tomorrow)
+
+ with Transaction().set_context(date=today):
+ self.assertEqual(
+ price_list.compute(None, None, Decimal(10), 1, None),
+ Decimal(10))
+ with Transaction().set_context(date=tomorrow):
+ self.assertEqual(
+ price_list.compute(None, None, Decimal(10), 1, None),
+ Decimal(9))
+
del ModuleTestCase
