server-ux/base_substate/tests/test_base_substate.py
2024-10-01 07:56:39 +02:00

120 lines
4.0 KiB
Python

# Copyright 2020 Akretion Mourad EL HADJ MIMOUNE
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
from flectra.tests import common
from .sale_test import LineTest, SaleTest
@common.tagged("post_install", "-at_install")
class TestBaseSubstate(common.TransactionCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
SaleTest._test_setup_models(cls.env, [SaleTest, LineTest])
LineTest._test_setup_model(cls.env)
cls.substate_test_sale = cls.env["base.substate.test.sale"]
cls.substate_test_sale_line = cls.env["base.substate.test.sale.line"]
cls.base_substate = cls.env["base.substate.mixin"]
cls.substate_type = cls.env["base.substate.type"]
cls.substate_type._fields["model"].selection.append(
("base.substate.test.sale", "Sale Order")
)
cls.substate_type = cls.env["base.substate.type"].create(
{
"name": "Sale",
"model": "base.substate.test.sale",
"target_state_field": "state",
}
)
cls.substate_val_quotation = cls.env["target.state.value"].create(
{
"name": "Quotation",
"base_substate_type_id": cls.substate_type.id,
"target_state_value": "draft",
}
)
cls.substate_val_sale = cls.env["target.state.value"].create(
{
"name": "Sale order",
"base_substate_type_id": cls.substate_type.id,
"target_state_value": "sale",
}
)
cls.substate_under_negotiation = cls.env["base.substate"].create(
{
"name": "Under negotiation",
"sequence": 1,
"target_state_value_id": cls.substate_val_quotation.id,
}
)
cls.substate_won = cls.env["base.substate"].create(
{
"name": "Won",
"sequence": 1,
"target_state_value_id": cls.substate_val_quotation.id,
}
)
cls.substate_wait_docs = cls.env["base.substate"].create(
{
"name": "Waiting for legal documents",
"sequence": 2,
"target_state_value_id": cls.substate_val_sale.id,
}
)
cls.substate_valid_docs = cls.env["base.substate"].create(
{
"name": "To validate legal documents",
"sequence": 3,
"target_state_value_id": cls.substate_val_sale.id,
}
)
cls.substate_in_delivering = cls.env["base.substate"].create(
{
"name": "In delivering",
"sequence": 4,
"target_state_value_id": cls.substate_val_sale.id,
}
)
@classmethod
def tearDownClass(cls):
SaleTest._test_teardown_model(cls.env)
LineTest._test_teardown_model(cls.env)
return super().tearDownClass()
def test_sale_order_substate(self):
partner = self.env.ref("base.res_partner_1")
so_test1 = self.substate_test_sale.create(
{
"name": "Test base substate to basic sale",
"partner_id": partner.id,
"line_ids": [
(0, 0, {"name": "line test", "amount": 120.0, "qty": 1.5})
],
}
)
self.assertTrue(so_test1.state == "draft")
self.assertTrue(so_test1.substate_id == self.substate_under_negotiation)
# Test that validation of sale order change substate_id
so_test1.button_confirm()
self.assertTrue(so_test1.state == "sale")
self.assertTrue(so_test1.substate_id == self.substate_wait_docs)
# Test that substate_id is set to false if
# there is not substate corresponding to state
so_test1.button_cancel()
self.assertTrue(so_test1.state == "cancel")
self.assertTrue(not so_test1.substate_id)