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)
