Initial commit: OCA Technical packages (595 packages)

odoo-bringout-oca-ddmrp-ddmrp_adjustment/ddmrp_adjustment/tests/test_common.py

@@ -0,0 +1,56 @@
+# Copyright 2018 Camptocamp SA
+# Copyright 2020 ForgeFlow S.L. (https://www.forgeflow.com)
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+import datetime
+from calendar import monthrange
+
+from dateutil.relativedelta import relativedelta
+
+from odoo.tests import TransactionCase
+from odoo.tools import DEFAULT_SERVER_DATE_FORMAT
+
+
+class TestDDMRPAdjustmentCommon(TransactionCase):
+    def _create_period(self, year, month, date_range_type):
+        return self.env["date.range"].create(
+            {
+                "name": "%i-%i" % (year, month),
+                "type_id": date_range_type.id,
+                "date_start": datetime.datetime(year=year, month=month, day=1).strftime(
+                    DEFAULT_SERVER_DATE_FORMAT
+                ),
+                "date_end": datetime.datetime(
+                    year=year, month=month, day=monthrange(year, month)[1]
+                ).strftime(DEFAULT_SERVER_DATE_FORMAT),
+            }
+        )
+
+    def _create_adjustment_wizard(self, number_of_periods):
+        date_start = datetime.datetime(year=self.now.year, month=self.now.month, day=1)
+        date_end = (
+            date_start + relativedelta(months=number_of_periods) - relativedelta(days=1)
+        )
+        wiz = self.env["ddmrp.adjustment.sheet"].create(
+            {
+                "date_start": date_start.strftime(DEFAULT_SERVER_DATE_FORMAT),
+                "date_end": date_end.strftime(DEFAULT_SERVER_DATE_FORMAT),
+                "date_range_type_id": self.month_date_range_type.id,
+            }
+        )
+        wiz.buffer_ids = [(4, self.buffer.id, False)]
+        wiz._onchange_sheet()
+        return wiz
+
+    def setUp(self):
+        super().setUp()
+        self.now = datetime.datetime.now()
+        self.month_date_range_type = self.env["date.range.type"].create(
+            {"name": "Month", "allow_overlap": False}
+        )
+        # create date ranges for each month in actual and next years
+        for y in (self.now.year, self.now.year + 1):
+            for m in range(1, 13):
+                date_range = self._create_period(y, m, self.month_date_range_type)
+                setattr(self, "month_%i_%i" % (y, m), date_range)
+        self.buffer = self.env.ref("ddmrp.stock_buffer_fp01")