[ADD] purchase: add test to check new functionalities

addons/purchase/tests/__init__.py

@@ -1,2 +1,4 @@
 # -*- coding: utf-8 -*-
 # Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import test_purchase_order

addons/purchase/tests/test_purchase_order.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from datetime import datetime
+from openerp.tools import DEFAULT_SERVER_DATETIME_FORMAT
+from openerp.tests import common
+
+class TestPurchaseOrder(common.TransactionCase):
+
+    def setUp(self):
+        super(TestPurchaseOrder, self).setUp()
+        # Useful models
+        self.PurchaseOrder = self.env['purchase.order']
+        self.PurchaseOrderLine = self.env['purchase.order.line']
+        self.AccountInvoice = self.env['account.invoice']
+        self.AccountInvoiceLine = self.env['account.invoice.line']
+
+    def test_00_purchase_order_flow(self):
+        self.partner_id = self.env.ref('base.res_partner_1')
+        self.product_id_1 = self.env.ref('product.product_product_8')
+        self.product_id_2 = self.env.ref('product.product_product_11')
+
+        # Ensure product_id_2 doesn't have res_partner_1 as supplier
+        if self.partner_id in self.product_id_2.seller_ids.mapped('name'):
+            id_to_remove = self.product_id_2.seller_ids.filtered(lambda r: r.name == self.partner_id).ids[0] if self.product_id_2.seller_ids.filtered(lambda r: r.name == self.partner_id) else False
+            if id_to_remove:
+                self.product_id_2.write({
+                    'seller_ids': [(2, id_to_remove, False)],
+                })
+        self.assertFalse(self.product_id_2.seller_ids.filtered(lambda r: r.name == self.partner_id), 'Purchase: the partner should not be in the list of the product suppliers')
+
+        po_vals = {
+            'partner_id': self.partner_id.id,
+            'order_line': [
+                (0, 0, {
+                    'name': self.product_id_1.name,
+                    'product_id': self.product_id_1.id,
+                    'product_qty': 5.0,
+                    'product_uom': self.product_id_1.uom_po_id.id,
+                    'price_unit': 500.0,
+                    'date_planned': datetime.today().strftime(DEFAULT_SERVER_DATETIME_FORMAT),
+                }),
+                (0, 0, {
+                    'name': self.product_id_2.name,
+                    'product_id': self.product_id_2.id,
+                    'product_qty': 5.0,
+                    'product_uom': self.product_id_2.uom_po_id.id,
+                    'price_unit': 250.0,
+                    'date_planned': datetime.today().strftime(DEFAULT_SERVER_DATETIME_FORMAT),
+                })],
+        }
+        self.po = self.PurchaseOrder.create(po_vals)
+        self.assertTrue(self.po, 'Purchase: no purchase order created')
+        self.assertEqual(self.po.invoice_status, 'no', 'Purchase: PO invoice_status should be "Not purchased"')
+        self.assertEqual(self.po.order_line.mapped('qty_received'), [0.0, 0.0], 'Purchase: no product should be received"')
+        self.assertEqual(self.po.order_line.mapped('qty_invoiced'), [0.0, 0.0], 'Purchase: no product should be invoiced"')
+
+        self.po.button_confirm()
+        self.assertEqual(self.po.state, 'purchase', 'Purchase: PO state should be "Purchase"')
+        self.assertEqual(self.po.invoice_status, 'to invoice', 'Purchase: PO invoice_status should be "Waiting Invoices"')
+
+        self.assertTrue(self.product_id_2.seller_ids.filtered(lambda r: r.name == self.partner_id), 'Purchase: the partner should be in the list of the product suppliers')
+        res = self.PurchaseOrderLine._get_name_price_quantity_date(self.product_id_2, self.partner_id, self.po.date_planned, 2.0, self.product_id_2.uom_po_id, self.po.currency_id)
+        self.assertEqual(res['price_unit'], 250.0, 'Purchase: the price of the product for the suppleir should be 250.0.')
+
+        self.assertEqual(self.po.picking_count, 1, 'Purchase: one picking should be created"')
+        self.picking = self.po.picking_ids[0]
+        self.picking.force_assign()
+        self.picking.pack_operation_product_ids.write({'qty_done': 5.0})
+        self.picking.do_new_transfer()
+        self.assertEqual(self.po.order_line.mapped('qty_received'), [5.0, 5.0], 'Purchase: all products should be received"')
+
+        self.invoice = self.AccountInvoice.create({
+            'partner_id': self.partner_id.id,
+            'purchase_id': self.po.id,
+            'account_id': self.AccountInvoiceLine.get_invoice_line_account('in_invoice', self.product_id_1, self.po.fiscal_position_id, self.env.user.company_id).id,
+        })
+        self.invoice.purchase_order_change()
+        self.assertEqual(self.po.order_line.mapped('qty_invoiced'), [5.0, 5.0], 'Purchase: all products should be invoiced"')