Initial commit of account_invoice_margin for 11.0

account_invoice_margin/tests/test_invoice_margin.py

@@ -0,0 +1,64 @@
+from odoo.addons.sale_margin.tests.test_sale_margin import TestSaleMargin
+
+class TestInvoiceMargin(TestSaleMargin):
+
+    def setUp(self):
+        super(TestInvoiceMargin, self).setUp()
+        self.AccountInvoice = self.env['account.invoice']
+
+    def test_invoice_margin(self):
+        """ Test the sale_margin module in Odoo. """
+        # Create a sales order for product Graphics Card.
+        sale_order_so11 = self.SaleOrder.create({
+            'name': 'Test_SO011',
+            'order_line': [
+                (0, 0, {
+                    'name': '[CARD] Graphics Card',
+                    'purchase_price': 700.0,
+                    'price_unit': 1000.0,
+                    'product_uom': self.product_uom_id,
+                    'product_uom_qty': 10.0,
+                    'state': 'draft',
+                    'product_id': self.product_id}),
+                (0, 0, {
+                    'name': 'Line without product_uom',
+                    'price_unit': 1000.0,
+                    'purchase_price': 700.0,
+                    'product_uom_qty': 10.0,
+                    'state': 'draft',
+                    'product_id': self.product_id})
+                ],
+            'partner_id': self.partner_id,
+            'partner_invoice_id': self.partner_invoice_address_id,
+            'partner_shipping_id': self.partner_invoice_address_id,
+            'pricelist_id': self.pricelist_id})
+        # Confirm the sales order.
+        sale_order_so11.action_confirm()
+        # Verify that margin field gets bind with the value.
+        self.assertEqual(sale_order_so11.margin, 6000.00, "Sales order margin should be 6000.00")
+
+        # Invoice the sales order.
+        inv_id = sale_order_so11.action_invoice_create()
+        inv = self.AccountInvoice.browse(inv_id)
+        self.assertEqual(inv.margin, sale_order_so11.margin)
+
+        account = self.env['account.account'].search([('internal_type', '=', 'other')], limit=1)
+        inv = self.AccountInvoice.create({
+            'partner_id': self.partner_id,
+            'invoice_line_ids': [
+                (0, 0, {
+                    'account_id': account.id,
+                    'name': '[CARD] Graphics Card',
+                    'purchase_price': 600.0,
+                    'price_unit': 1000.0,
+                    'quantity': 10.0,
+                    'product_id': self.product_id}),
+                (0, 0, {
+                    'account_id': account.id,
+                    'name': 'Line without product_uom',
+                    'price_unit': 1000.0,
+                    'purchase_price': 800.0,
+                    'quantity': 10.0,})
+                ],
+        })
+        self.assertEqual(inv.margin, 6000.0)