sale_product_multi_add/tests/test_sale.py

# -*- coding: utf-8 -*-
# © 2016  Cédric Pigeon, Acsone SA/NV (http://www.acsone.eu)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
import openerp.tests.common as common


class TestSale(common.TransactionCase):

    def setUp(self):
        common.TransactionCase.setUp(self)

        self.product_35 = self.env.ref("product.product_product_35")
        self.product_36 = self.env.ref("product.product_product_36")

    def test_import_product(self):
        """ Create SO
            Import products
            Check products are presents
        """

        so = self.env["sale.order"].create(
            {"partner_id": self.env.ref("base.res_partner_2").id,
             })

        wiz_obj = self.env['sale.import.products']
        wizard = wiz_obj.with_context(active_id=so.id,
                                      active_model='sale.order')

        products = [(6, 0, [self.product_35.id, self.product_36.id])]

        wizard_id = wizard.create({'products': products})
        wizard_id.create_items()
        wizard_id.items[0].quantity = 4
        wizard_id.items[1].quantity = 6
        wizard_id.select_products()

        self.assertEqual(len(so.order_line), 2)

        for line in so.order_line:
            if line.product_id.id == self.product_35.id:
                self.assertEqual(line.product_uom_qty, 4)
            else:
                self.assertEqual(line.product_uom_qty, 6)