19.0 vanilla

odoo-bringout-oca-ocb-website_sale_stock/website_sale_stock/tests/common.py

@@ -0,0 +1,40 @@
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo.addons.website_sale.tests.common import WebsiteSaleCommon
+
+
+class WebsiteSaleStockCommon(WebsiteSaleCommon):
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+        cls.warehouse = cls.env['stock.warehouse'].search([('company_id', '=', cls.company.id)])
+
+    @classmethod
+    def _add_product_qty_to_wh(cls, product_id, qty, loc_id):
+        cls.env['stock.quant'].with_context(inventory_mode=True).create({
+            'product_id': product_id,
+            'inventory_quantity': qty,
+            'location_id': loc_id,
+        }).action_apply_inventory()
+
+    @classmethod
+    def _create_product(cls, **create_values):
+        """ Override of `website_sale` to create storable products by default and restrict them from
+        selling when out of stock.
+        """
+        if create_values.get('type', 'consu') == 'consu':  # Only for goods.
+            if 'is_storable' not in create_values:
+                create_values['is_storable'] = True
+            if 'allow_out_of_stock_order' not in create_values:
+                create_values['allow_out_of_stock_order'] = False
+        return super()._create_product(**create_values)
+
+    @classmethod
+    def _create_warehouse(cls, **kwargs):
+        return cls.env['stock.warehouse'].create({
+            'name': 'Test Warehouse',
+            'code': 'TWH',
+            **kwargs,
+        })