first commit
This commit is contained in:
root
313
modules/stock_lot/tests/test_module.py
Normal file
313
modules/stock_lot/tests/test_module.py
Normal file
@@ -0,0 +1,313 @@
|
||||
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
||||
# this repository contains the full copyright notices and license terms.
|
||||
|
||||
import datetime as dt
|
||||
from decimal import Decimal
|
||||
|
||||
from trytond.modules.company.tests import (
|
||||
CompanyTestMixin, create_company, set_company)
|
||||
from trytond.pool import Pool
|
||||
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
|
||||
from trytond.transaction import Transaction
|
||||
|
||||
|
||||
class StockLotTestCase(CompanyTestMixin, ModuleTestCase):
|
||||
'Test Stock Lot module'
|
||||
module = 'stock_lot'
|
||||
extras = ['stock_split']
|
||||
|
||||
@with_transaction()
|
||||
def test_products_by_location(self):
|
||||
'Test products_by_location'
|
||||
pool = Pool()
|
||||
Uom = pool.get('product.uom')
|
||||
Template = pool.get('product.template')
|
||||
Product = pool.get('product.product')
|
||||
Location = pool.get('stock.location')
|
||||
Move = pool.get('stock.move')
|
||||
Lot = pool.get('stock.lot')
|
||||
|
||||
kg, = Uom.search([('name', '=', 'Kilogram')])
|
||||
g, = Uom.search([('name', '=', 'Gram')])
|
||||
template, = Template.create([{
|
||||
'name': 'Test products_by_location',
|
||||
'type': 'goods',
|
||||
'default_uom': kg.id,
|
||||
}])
|
||||
product, = Product.create([{
|
||||
'template': template.id,
|
||||
}])
|
||||
supplier, = Location.search([('code', '=', 'SUP')])
|
||||
customer, = Location.search([('code', '=', 'CUS')])
|
||||
storage, = Location.search([('code', '=', 'STO')])
|
||||
company = create_company()
|
||||
currency = company.currency
|
||||
with set_company(company):
|
||||
lot1, lot2 = Lot.create([{
|
||||
'number': '1',
|
||||
'product': product.id,
|
||||
}, {
|
||||
'number': '2',
|
||||
'product': product.id,
|
||||
}])
|
||||
|
||||
moves = Move.create([{
|
||||
'product': product.id,
|
||||
'lot': lot1.id,
|
||||
'unit': kg.id,
|
||||
'quantity': 5,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': lot2.id,
|
||||
'unit': kg.id,
|
||||
'quantity': 10,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': lot2.id,
|
||||
'unit': kg.id,
|
||||
'quantity': 2,
|
||||
'from_location': storage.id,
|
||||
'to_location': customer.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': None,
|
||||
'unit': kg.id,
|
||||
'quantity': 3,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}])
|
||||
Move.do(moves)
|
||||
|
||||
self.assertEqual(Product.products_by_location([storage.id],
|
||||
grouping_filter=([product.id],)), {
|
||||
(storage.id, product.id): 16,
|
||||
})
|
||||
self.assertEqual(Product.products_by_location([storage.id],
|
||||
grouping=('product', 'lot',),
|
||||
grouping_filter=([product.id],)), {
|
||||
(storage.id, product.id, lot1.id): 5,
|
||||
(storage.id, product.id, lot2.id): 8,
|
||||
(storage.id, product.id, None): 3,
|
||||
})
|
||||
with Transaction().set_context(locations=[storage.id]):
|
||||
lot1, lot2 = Lot.browse([lot1, lot2])
|
||||
self.assertEqual(lot1.quantity, 5)
|
||||
self.assertEqual(lot2.quantity, 8)
|
||||
|
||||
@with_transaction()
|
||||
def test_period(self):
|
||||
'Test period'
|
||||
pool = Pool()
|
||||
Uom = pool.get('product.uom')
|
||||
Template = pool.get('product.template')
|
||||
Product = pool.get('product.product')
|
||||
Location = pool.get('stock.location')
|
||||
Move = pool.get('stock.move')
|
||||
Lot = pool.get('stock.lot')
|
||||
Period = pool.get('stock.period')
|
||||
|
||||
unit, = Uom.search([('name', '=', 'Unit')])
|
||||
template, = Template.create([{
|
||||
'name': 'Test period',
|
||||
'type': 'goods',
|
||||
'default_uom': unit.id,
|
||||
}])
|
||||
product, = Product.create([{
|
||||
'template': template.id,
|
||||
}])
|
||||
supplier, = Location.search([('code', '=', 'SUP')])
|
||||
storage, = Location.search([('code', '=', 'STO')])
|
||||
company = create_company()
|
||||
currency = company.currency
|
||||
with set_company(company):
|
||||
lot1, lot2 = Lot.create([{
|
||||
'number': '1',
|
||||
'product': product.id,
|
||||
}, {
|
||||
'number': '2',
|
||||
'product': product.id,
|
||||
}])
|
||||
|
||||
today = dt.date.today()
|
||||
|
||||
moves = Move.create([{
|
||||
'product': product.id,
|
||||
'lot': lot1.id,
|
||||
'unit': unit.id,
|
||||
'quantity': 5,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'planned_date': today - dt.timedelta(days=1),
|
||||
'effective_date': today - dt.timedelta(days=1),
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': lot2.id,
|
||||
'unit': unit.id,
|
||||
'quantity': 10,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'planned_date': today - dt.timedelta(days=1),
|
||||
'effective_date': today - dt.timedelta(days=1),
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': None,
|
||||
'unit': unit.id,
|
||||
'quantity': 3,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'planned_date': today - dt.timedelta(days=1),
|
||||
'effective_date': today - dt.timedelta(days=1),
|
||||
'company': company.id,
|
||||
'unit_price': Decimal('1'),
|
||||
'currency': currency.id,
|
||||
}])
|
||||
Move.do(moves)
|
||||
|
||||
period, = Period.create([{
|
||||
'date': today - dt.timedelta(days=1),
|
||||
'company': company.id,
|
||||
}])
|
||||
Period.close([period])
|
||||
self.assertEqual(period.state, 'closed')
|
||||
|
||||
quantities = {
|
||||
supplier: -18,
|
||||
storage: 18,
|
||||
}
|
||||
for cache in period.caches:
|
||||
self.assertEqual(cache.product, product)
|
||||
self.assertEqual(cache.internal_quantity,
|
||||
quantities[cache.location])
|
||||
|
||||
quantities = {
|
||||
(supplier, lot1): -5,
|
||||
(storage, lot1): 5,
|
||||
(supplier, lot2): -10,
|
||||
(storage, lot2): 10,
|
||||
(supplier, None): -3,
|
||||
(storage, None): 3,
|
||||
}
|
||||
for lot_cache in period.lot_caches:
|
||||
self.assertEqual(lot_cache.product, product)
|
||||
self.assertEqual(lot_cache.internal_quantity,
|
||||
quantities[(lot_cache.location, lot_cache.lot)])
|
||||
|
||||
@with_transaction()
|
||||
def test_assign_try_with_lot(self):
|
||||
"Test Move assign_try with lot"
|
||||
pool = Pool()
|
||||
Template = pool.get('product.template')
|
||||
Product = pool.get('product.product')
|
||||
Uom = pool.get('product.uom')
|
||||
Location = pool.get('stock.location')
|
||||
Move = pool.get('stock.move')
|
||||
Lot = pool.get('stock.lot')
|
||||
|
||||
unit, = Uom.search([('name', '=', 'Meter')])
|
||||
template = Template(
|
||||
name="Product",
|
||||
type='goods',
|
||||
default_uom=unit,
|
||||
)
|
||||
template.save()
|
||||
product = Product(template=template.id)
|
||||
product.save()
|
||||
|
||||
supplier, = Location.search([('code', '=', 'SUP')])
|
||||
storage, = Location.search([('code', '=', 'STO')])
|
||||
customer, = Location.search([('code', '=', 'CUS')])
|
||||
|
||||
company = create_company()
|
||||
with set_company(company):
|
||||
lot1, lot2 = Lot.create([{
|
||||
'number': "1",
|
||||
'product': product.id,
|
||||
}, {
|
||||
'number': "2",
|
||||
'product': product.id,
|
||||
}])
|
||||
moves = Move.create([{
|
||||
'product': product.id,
|
||||
'lot': lot1.id,
|
||||
'unit': unit.id,
|
||||
'quantity': 2,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal(1),
|
||||
'currency': company.currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': lot2.id,
|
||||
'unit': unit.id,
|
||||
'quantity': 3,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal(1),
|
||||
'currency': company.currency.id,
|
||||
}, {
|
||||
'product': product.id,
|
||||
'lot': None,
|
||||
'unit': unit.id,
|
||||
'quantity': 3,
|
||||
'from_location': supplier.id,
|
||||
'to_location': storage.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal(1),
|
||||
'currency': company.currency.id,
|
||||
}])
|
||||
Move.do(moves)
|
||||
|
||||
move, = Move.create([{
|
||||
'product': product.id,
|
||||
'unit': unit.id,
|
||||
'quantity': 10,
|
||||
'from_location': storage.id,
|
||||
'to_location': customer.id,
|
||||
'company': company.id,
|
||||
'unit_price': Decimal(1),
|
||||
'currency': company.currency.id,
|
||||
}])
|
||||
|
||||
self.assertFalse(
|
||||
Move.assign_try([move], grouping=('product', 'lot')))
|
||||
moves = Move.search([
|
||||
('product', '=', product.id),
|
||||
('from_location', '=', storage.id),
|
||||
('to_location', '=', customer.id),
|
||||
('company', '=', company.id),
|
||||
])
|
||||
self.assertEqual(len(moves), 4)
|
||||
self.assertEqual({
|
||||
(m.lot, m.quantity, m.state) for m in moves}, {
|
||||
(lot1, 2, 'assigned'),
|
||||
(lot2, 3, 'assigned'),
|
||||
(None, 3, 'assigned'),
|
||||
(None, 2, 'draft'),
|
||||
})
|
||||
|
||||
|
||||
del ModuleTestCase
|
||||
Reference in New Issue
Block a user