# Part of Flectra See LICENSE file for full copyright and licensing details.

from .discount_common import TestDiscountCommon


class TestSODiscount(TestDiscountCommon):
    def setUp(self):
        super(TestSODiscount, self).setUp()

    def discount_01_set_fixamount(self):
        sale_order = self.SaleOrder.create({
            'partner_id': self.partner_id.id,
            'pricelist_id': self.pricelist_id.id,
        })
        sale_order.onchange_discount_method()
        self.SaleOrderLine.create({
            'name': self.product_1.name,
            'product_id': self.product_1.id,
            'product_uom_qty': 20,
            'product_uom': self.product_1.uom_id.id,
            'price_unit': self.product_1.list_price,
            'order_id': sale_order.id,
        })
        self.SaleOrderLine.create({
            'name': self.product_2.name,
            'product_id': self.product_2.id,
            'product_uom_qty': 20,
            'product_uom': self.product_2.uom_id.id,
            'price_unit': self.product_2.list_price,
            'order_id': sale_order.id,
        })
        sale_order.write({
            'discount_method': 'fixed',
            'discount_amount': 100,
        })
        return sale_order

    def discount_02_set_percentages(self):
        sale_order = self.SaleOrder.create({
            'partner_id': self.partner_id.id,
            'pricelist_id': self.pricelist_id.id,
        })
        sale_order.onchange_discount_method()
        sale_order.write({
            'discount_method': 'per',
            'discount_per': 10,
        })
        self.SaleOrderLine.create({
            'name': self.product_1.name,
            'product_id': self.product_1.id,
            'product_uom_qty': 10,
            'product_uom': self.product_1.uom_id.id,
            'price_unit': self.product_1.list_price,
            'order_id': sale_order.id,
            'discount': 10,
        })
        self.SaleOrderLine.create({
            'name': self.product_2.name,
            'product_id': self.product_2.id,
            'product_uom_qty': 20,
            'product_uom': self.product_2.uom_id.id,
            'price_unit': self.product_2.list_price,
            'order_id': sale_order.id,
            'discount': 10,
        })
        sale_order.write({
            'discount_method': 'per',
            'discount_per': 20.0,
        })
        return sale_order