initial commit 2

1 files changed, 57 insertions, 0 deletions

diff --git a/addons/purchase_stock/tests/common.py b/addons/purchase_stock/tests/common.py
new file mode 100644
index 00000000..f3a98e04
--- /dev/null
+++ b/addons/purchase_stock/tests/common.py
@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+from datetime import timedelta
+
+from odoo import fields
+from odoo.addons.stock.tests.common2 import TestStockCommon
+from odoo import tools
+from odoo.modules.module import get_module_resource
+
+
+class PurchaseTestCommon(TestStockCommon):
+
+    def _create_make_procurement(self, product, product_qty, date_planned=False):
+        ProcurementGroup = self.env['procurement.group']
+        order_values = {
+            'warehouse_id': self.warehouse_1,
+            'action': 'pull_push',
+            'date_planned': date_planned or fields.Datetime.to_string(fields.datetime.now() + timedelta(days=10)),  # 10 days added to current date of procurement to get future schedule date and order date of purchase order.
+            'group_id': self.env['procurement.group'],
+        }
+        return ProcurementGroup.run([self.env['procurement.group'].Procurement(
+            product, product_qty, self.uom_unit, self.warehouse_1.lot_stock_id,
+            product.name, '/', self.env.company, order_values)
+        ])
+
+    @classmethod
+    def setUpClass(cls):
+        super(PurchaseTestCommon, cls).setUpClass()
+        cls.env.ref('stock.route_warehouse0_mto').active = True
+
+        cls.route_buy = cls.warehouse_1.buy_pull_id.route_id.id
+        cls.route_mto = cls.warehouse_1.mto_pull_id.route_id.id
+
+        # Update product_1 with type, route and Delivery Lead Time
+        cls.product_1.write({
+            'type': 'product',
+            'route_ids': [(6, 0, [cls.route_buy, cls.route_mto])],
+            'seller_ids': [(0, 0, {'name': cls.partner_1.id, 'delay': 5})]})
+
+        cls.t_shirt = cls.env['product.product'].create({
+            'name': 'T-shirt',
+            'route_ids': [(6, 0, [cls.route_buy, cls.route_mto])],
+            'seller_ids': [(0, 0, {'name': cls.partner_1.id, 'delay': 5})]
+        })
+
+        # Update product_2 with type, route and Delivery Lead Time
+        cls.product_2.write({
+            'type': 'product',
+            'route_ids': [(6, 0, [cls.route_buy, cls.route_mto])],
+            'seller_ids': [(0, 0, {'name': cls.partner_1.id, 'delay': 2})]})
+
+        cls.res_users_purchase_user = cls.env['res.users'].create({
+            'company_id': cls.env.ref('base.main_company').id,
+            'name': "Purchase User",
+            'login': "pu",
+            'email': "purchaseuser@yourcompany.com",
+            'groups_id': [(6, 0, [cls.env.ref('purchase.group_purchase_user').id])],
+            })