bug th qt

modules/purchase_trade/tests/test_module.py

@@ -184,6 +184,30 @@ class PurchaseTradeTestCase(ModuleTestCase):
             self.assertEqual(
                 PurchaseLine.default_pricing_rule(), 'Default pricing rule')
 
+    def test_component_quota_uses_quantity_fallback_when_theoretical_is_missing(self):
+        'component quota does not crash when theoretical quantity is still empty'
+        SaleComponent = Pool().get('pricing.component')
+        PurchaseComponent = Pool().get('pricing.component')
+
+        sale_component = SaleComponent()
+        sale_component.nbdays = None
+        sale_component.sale_line = Mock(
+            quantity=Decimal('12'),
+            quantity_theorical=None,
+        )
+
+        purchase_component = PurchaseComponent()
+        purchase_component.nbdays = None
+        purchase_component.line = Mock(
+            quantity=Decimal('15'),
+            quantity_theorical=None,
+        )
+
+        self.assertEqual(sale_component.get_quota_sale('quota_sale'), Decimal('12.0000'))
+        self.assertEqual(
+            purchase_component.get_quota_purchase('quota'),
+            Decimal('15.00000'))
+
     def test_sale_and_purchase_trader_operator_domains_use_explicit_categories(self):
         'sale and purchase trader/operator fields are filtered by TRADER/OPERATOR categories'
         Sale = Pool().get('sale.sale')