initial commit 2

2 files changed, 204 insertions, 0 deletions

diff --git a/addons/sale_stock_margin/tests/__init__.py b/addons/sale_stock_margin/tests/__init__.py
new file mode 100644
index 00000000..c4661d84
--- /dev/null
+++ b/addons/sale_stock_margin/tests/__init__.py
@@ -0,0 +1,3 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+from . import test_sale_stock_margin
diff --git a/addons/sale_stock_margin/tests/test_sale_stock_margin.py b/addons/sale_stock_margin/tests/test_sale_stock_margin.py
new file mode 100644
index 00000000..f96816e5
--- /dev/null
+++ b/addons/sale_stock_margin/tests/test_sale_stock_margin.py
@@ -0,0 +1,201 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields
+
+from odoo.tests.common import Form
+from odoo.addons.stock_account.tests.test_stockvaluationlayer import TestStockValuationCommon
+
+
+class TestSaleStockMargin(TestStockValuationCommon):
+
+    @classmethod
+    def setUpClass(cls):
+        super(TestSaleStockMargin, cls).setUpClass()
+        cls.pricelist = cls.env['product.pricelist'].create({'name': 'Simple Pricelist'})
+
+    #########
+    # UTILS #
+    #########
+
+    def _create_sale_order(self):
+        return self.env['sale.order'].create({
+            'name': 'Sale order',
+            'partner_id': self.env.ref('base.partner_admin').id,
+            'partner_invoice_id': self.env.ref('base.partner_admin').id,
+            'pricelist_id': self.pricelist.id,
+        })
+
+    def _create_sale_order_line(self, sale_order, product, quantity, price_unit=0):
+        return self.env['sale.order.line'].create({
+            'name': 'Sale order',
+            'order_id': sale_order.id,
+            'price_unit': price_unit,
+            'product_id': product.id,
+            'product_uom_qty': quantity,
+            'product_uom': self.env.ref('uom.product_uom_unit').id,
+        })
+
+    def _create_product(self):
+        product_template = self.env['product.template'].create({
+            'name': 'Super product',
+            'type': 'product',
+        })
+        product_template.categ_id.property_cost_method = 'fifo'
+        return product_template.product_variant_ids
+
+    #########
+    # TESTS #
+    #########
+
+    def test_sale_stock_margin_1(self):
+        sale_order = self._create_sale_order()
+        product = self._create_product()
+
+        self._make_in_move(product, 2, 35)
+        self._make_out_move(product, 1)
+
+        order_line = self._create_sale_order_line(sale_order, product, 1, 50)
+        sale_order.action_confirm()
+
+        self.assertEqual(order_line.purchase_price, 35)
+        self.assertEqual(sale_order.margin, 15)
+
+        sale_order.picking_ids.move_lines.quantity_done = 1
+        sale_order.picking_ids.button_validate()
+
+        self.assertEqual(order_line.purchase_price, 35)
+        self.assertEqual(order_line.margin, 15)
+        self.assertEqual(sale_order.margin, 15)
+
+    def test_sale_stock_margin_2(self):
+        sale_order = self._create_sale_order()
+        product = self._create_product()
+
+        self._make_in_move(product, 2, 32)
+        self._make_in_move(product, 5, 17)
+        self._make_out_move(product, 1)
+
+        order_line = self._create_sale_order_line(sale_order, product, 2, 50)
+        sale_order.action_confirm()
+
+        self.assertEqual(order_line.purchase_price, 32)
+        self.assertAlmostEqual(sale_order.margin, 36)
+
+        sale_order.picking_ids.move_lines.quantity_done = 2
+        sale_order.picking_ids.button_validate()
+
+        self.assertAlmostEqual(order_line.purchase_price, 24.5)
+        self.assertAlmostEqual(order_line.margin, 51)
+        self.assertAlmostEqual(sale_order.margin, 51)
+
+    def test_sale_stock_margin_3(self):
+        sale_order = self._create_sale_order()
+        product = self._create_product()
+
+        self._make_in_move(product, 2, 10)
+        self._make_out_move(product, 1)
+
+        order_line = self._create_sale_order_line(sale_order, product, 2, 20)
+        sale_order.action_confirm()
+
+        self.assertEqual(order_line.purchase_price, 10)
+        self.assertAlmostEqual(sale_order.margin, 20)
+
+        sale_order.picking_ids.move_lines.quantity_done = 1
+        sale_order.picking_ids.button_validate()
+
+        self.assertAlmostEqual(order_line.purchase_price, 10)
+        self.assertAlmostEqual(order_line.margin, 20)
+        self.assertAlmostEqual(sale_order.margin, 20)
+
+    def test_sale_stock_margin_4(self):
+        sale_order = self._create_sale_order()
+        product = self._create_product()
+
+        self._make_in_move(product, 2, 10)
+        self._make_in_move(product, 1, 20)
+        self._make_out_move(product, 1)
+
+        order_line = self._create_sale_order_line(sale_order, product, 2, 20)
+        sale_order.action_confirm()
+
+        self.assertEqual(order_line.purchase_price, 10)
+        self.assertAlmostEqual(sale_order.margin, 20)
+
+        sale_order.picking_ids.move_lines.quantity_done = 1
+        res = sale_order.picking_ids.button_validate()
+        Form(self.env[res['res_model']].with_context(res['context'])).save().process()
+
+        self.assertAlmostEqual(order_line.purchase_price, 15)
+        self.assertAlmostEqual(order_line.margin, 10)
+        self.assertAlmostEqual(sale_order.margin, 10)
+
+    def test_sale_stock_margin_5(self):
+        sale_order = self._create_sale_order()
+        product_1 = self._create_product()
+        product_2 = self._create_product()
+
+        self._make_in_move(product_1, 2, 35)
+        self._make_in_move(product_1, 1, 51)
+        self._make_out_move(product_1, 1)
+
+        self._make_in_move(product_2, 2, 17)
+        self._make_in_move(product_2, 1, 11)
+        self._make_out_move(product_2, 1)
+
+        order_line_1 = self._create_sale_order_line(sale_order, product_1, 2, 60)
+        order_line_2 = self._create_sale_order_line(sale_order, product_2, 4, 20)
+        sale_order.action_confirm()
+
+        self.assertAlmostEqual(order_line_1.purchase_price, 35)
+        self.assertAlmostEqual(order_line_2.purchase_price, 17)
+        self.assertAlmostEqual(order_line_1.margin, 25 * 2)
+        self.assertAlmostEqual(order_line_2.margin, 3 * 4)
+        self.assertAlmostEqual(sale_order.margin, 62)
+
+        sale_order.picking_ids.move_lines[0].quantity_done = 2
+        sale_order.picking_ids.move_lines[1].quantity_done = 3
+
+        res = sale_order.picking_ids.button_validate()
+        Form(self.env[res['res_model']].with_context(res['context'])).save().process()
+
+        self.assertAlmostEqual(order_line_1.purchase_price, 43)       # (35 + 51) / 2
+        self.assertAlmostEqual(order_line_2.purchase_price, 12.5)     # (17 + 11 + 11 + 11) / 4
+        self.assertAlmostEqual(order_line_1.margin, 34)               # (60 - 43) * 2
+        self.assertAlmostEqual(order_line_2.margin, 30)               # (20 - 12.5) * 4
+        self.assertAlmostEqual(sale_order.margin, 64)
+
+    def test_so_and_multicurrency(self):
+        ResCurrencyRate = self.env['res.currency.rate']
+        company_currency = self.env.company.currency_id
+        other_currency = self.env.ref('base.EUR') if company_currency == self.env.ref('base.USD') else self.env.ref('base.USD')
+
+        ResCurrencyRate.create({'currency_id': company_currency.id, 'rate': 1})
+        other_currency_rate = ResCurrencyRate.search([('name', '=', fields.Date.today()), ('currency_id', '=', other_currency.id)])
+        if other_currency_rate:
+            other_currency_rate.rate = 2
+        else:
+            ResCurrencyRate.create({'currency_id': other_currency.id, 'rate': 2})
+
+        so = self._create_sale_order()
+        so.pricelist_id = self.env['product.pricelist'].create({
+            'name': 'Super Pricelist',
+            'currency_id': other_currency.id,
+        })
+
+        product = self._create_product()
+        product.standard_price = 100
+        product.list_price = 200
+
+        so_form = Form(so)
+        with so_form.order_line.new() as line:
+            line.product_id = product
+        so = so_form.save()
+        so_line = so.order_line
+
+        self.assertEqual(so_line.purchase_price, 200)
+        self.assertEqual(so_line.price_unit, 400)
+        so.action_confirm()
+        self.assertEqual(so_line.purchase_price, 200)
+        self.assertEqual(so_line.price_unit, 400)