Files
stock-logistics-warehouse/stock_demand_estimate/tests/test_stock_demand_estimate.py

126 lines
4.7 KiB
Python

# Copyright 2017-19 ForgeFlow S.L. (https://www.forgeflow.com)
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html).
from odoo.tests.common import SavepointCase
from datetime import date, timedelta as td
class TestStockDemandEstimate(SavepointCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.res_users_model = cls.env['res.users']
cls.product_model = cls.env['product.product']
cls.stock_location_model = cls.env['stock.location']
cls.estimate_model = cls.env["stock.demand.estimate"]
cls.g_stock_manager = cls.env.ref('stock.group_stock_manager')
cls.g_stock_user = cls.env.ref('stock.group_stock_user')
cls.company = cls.env.ref('base.main_company')
cls.uom_unit = cls.env.ref('uom.product_uom_unit')
cls.uom_dozen = cls.env.ref('uom.product_uom_dozen')
# Create users:
cls.manager = cls._create_user(
'user_1',
[cls.g_stock_manager],
cls.company,
).id
cls.user = cls._create_user(
'user_2',
[cls.g_stock_user],
cls.company,
).id
# Create a product:
cls.product_1 = cls.product_model.create({
'name': 'Test Product 1',
'type': 'product',
'default_code': 'PROD1',
"uom_id": cls.uom_unit.id,
})
# Create a location:
cls.location = cls.stock_location_model.create({
'name': 'Place',
'usage': 'production',
})
@classmethod
def _create_user(cls, login, groups, company):
group_ids = [group.id for group in groups]
user = cls.res_users_model.create({
'name': login,
'login': login,
'password': 'demo',
'email': 'example@yourcompany.com',
'company_id': company.id,
'company_ids': [(4, company.id)],
'groups_id': [(6, 0, group_ids)],
})
return user
def test_01_create_estimate(self):
"""Crete an estimate entering manually the date from and date to."""
date_from = date.today() + td(days=10)
date_to = date.today() + td(days=19)
estimate = self.estimate_model.create({
"product_id": self.product_1.id,
"location_id": self.location.id,
"manual_date_from": date_from,
"manual_date_to": date_to,
"product_uom_qty": 500.0,
})
self.assertEqual(estimate.date_from, date_from)
self.assertEqual(estimate.date_to, date_to)
self.assertEqual(estimate.duration, 10)
self.assertEqual(estimate.product_qty, 500.0)
self.assertEqual(estimate.daily_qty, 500.0 / 10.0)
def test_02_create_estimate_by_duration_and_different_uom(self):
"""Create an estimate entering manually the date from, duration and
using a different UoM than product's."""
date_from = date.today() + td(days=10)
estimate = self.estimate_model.create({
"product_id": self.product_1.id,
"location_id": self.location.id,
"manual_date_from": date_from,
"manual_duration": 15,
"product_uom_qty": 100.0,
"product_uom": self.uom_dozen.id,
})
self.assertEqual(estimate.date_from, date_from)
expected_date_to = estimate.date_from + td(days=15 - 1)
self.assertEqual(estimate.date_to, expected_date_to)
self.assertEqual(estimate.duration, 15)
expected_qty = 100 * 12.0 # 100 dozens -> units
self.assertEqual(estimate.product_qty, expected_qty)
self.assertEqual(estimate.daily_qty, expected_qty / 15)
def test_03_get_qty_by_range(self):
date_from = date.today() + td(days=10)
date_to = date.today() + td(days=19)
estimate = self.estimate_model.create({
"product_id": self.product_1.id,
"location_id": self.location.id,
"manual_date_from": date_from,
"manual_date_to": date_to,
"product_uom_qty": 100.0,
})
self.assertEqual(estimate.duration, 10.0)
self.assertEqual(estimate.daily_qty, 10.0)
res = estimate.get_quantity_by_date_range(
date_from + td(days=3), date_from + td(days=17))
self.assertEqual(res, 70)
res = estimate.get_quantity_by_date_range(
date_from + td(days=3), date_from + td(days=7))
self.assertEqual(res, 50)
# get full period
res = estimate.get_quantity_by_date_range(
date.today(), date_from + td(days=17))
self.assertEqual(res, 100)
# Get exact period:
res = estimate.get_quantity_by_date_range(
estimate.date_from, estimate.date_to)
self.assertEqual(res, 100)