Files
suite/sale_sourced_by_line/tests/test_sale_sources.py
2021-10-06 15:54:50 +00:00

64 lines
2.7 KiB
Python

from odoo.tests import common
class TestSaleSources(common.TransactionCase):
def setUp(self):
super(TestSaleSources, self).setUp()
self.partner = self.env.ref('base.res_partner_2')
self.product_1 = self.env['product.product'].create({
'type': 'consu',
'name': 'Test Product 1',
})
self.product_2 = self.env['product.product'].create({
'type': 'consu',
'name': 'Test Product 2',
})
self.wh_1 = self.env.ref('stock.warehouse0')
self.wh_2 = self.env['stock.warehouse'].create({
'name': 'Test WH2',
'code': 'TWH2',
})
def test_plan_one_warehouse(self):
so = self.env['sale.order'].create({
'warehouse_id': self.wh_1.id,
'partner_id': self.partner.id,
'date_planned': '2018-01-01',
'order_line': [(0, 0, {
'product_id': self.product_1.id,
'product_uom_qty': 1.0,
'product_uom': self.product_1.uom_id.id,
'price_unit': 10.0,
}),
(0, 0, {
'product_id': self.product_2.id,
'product_uom_qty': 1.0,
'product_uom': self.product_2.uom_id.id,
'price_unit': 10.0,
})]
})
so.action_confirm()
self.assertTrue(so.state in ('sale', 'done'))
self.assertEqual(len(so.picking_ids), 1)
self.assertEqual(len(so.picking_ids.filtered(lambda p: p.picking_type_id.warehouse_id == self.wh_1)), 1)
self.assertEqual(len(so.picking_ids.filtered(lambda p: p.picking_type_id.warehouse_id == self.wh_2)), 0)
def test_plan_two_warehouses(self):
so = self.env['sale.order'].create({
'warehouse_id': self.wh_1.id,
'partner_id': self.partner.id,
'date_planned': '2018-01-01',
'order_line': [(0, 0, {'product_id': self.product_1.id}),
(0, 0, {'product_id': self.product_2.id,
'date_planned': '2018-02-01', 'warehouse_id': self.wh_2.id})]
})
# in 13 default computation, this would result in a failure
self.assertTrue(so.order_line.filtered(lambda l: l.warehouse_id))
so.action_confirm()
self.assertTrue(so.state in ('sale', 'done'))
self.assertEqual(len(so.picking_ids), 2)
self.assertEqual(len(so.picking_ids.filtered(lambda p: p.picking_type_id.warehouse_id == self.wh_1)), 1)
self.assertEqual(len(so.picking_ids.filtered(lambda p: p.picking_type_id.warehouse_id == self.wh_2)), 1)