02.04.26

modules/purchase_trade/tests/test_module.py

@@ -156,6 +156,34 @@ class PurchaseTradeTestCase(ModuleTestCase):
         sale.crop.name = 'Main Crop'
         self.assertEqual(sale.report_crop_name, 'Main Crop')
 
+    def test_sale_line_default_pricing_rule_comes_from_configuration(self):
+        'sale line pricing_rule defaults to the purchase_trade singleton value'
+        SaleLine = Pool().get('sale.line')
+        config = Mock(pricing_rule='Default pricing rule')
+        configuration_model = Mock()
+        configuration_model.search.return_value = [config]
+
+        with patch(
+                'trytond.modules.purchase_trade.sale.Pool'
+                ) as PoolMock:
+            PoolMock.return_value.get.return_value = configuration_model
+            self.assertEqual(
+                SaleLine.default_pricing_rule(), 'Default pricing rule')
+
+    def test_purchase_line_default_pricing_rule_comes_from_configuration(self):
+        'purchase line pricing_rule defaults to the purchase_trade singleton value'
+        PurchaseLine = Pool().get('purchase.line')
+        config = Mock(pricing_rule='Default pricing rule')
+        configuration_model = Mock()
+        configuration_model.search.return_value = [config]
+
+        with patch(
+                'trytond.modules.purchase_trade.purchase.Pool'
+                ) as PoolMock:
+            PoolMock.return_value.get.return_value = configuration_model
+            self.assertEqual(
+                PurchaseLine.default_pricing_rule(), 'Default pricing rule')
+
     def test_sale_report_multi_line_helpers_aggregate_all_lines(self):
         'sale report helpers aggregate quantity, price lines and shipment periods'
         Sale = Pool().get('sale.sale')