From 15a93099a8a17e5def0830347714d29a3b6e5e11 Mon Sep 17 00:00:00 2001 From: Simone Orsi Date: Mon, 18 Nov 2024 17:56:58 +0100 Subject: [PATCH] edi_sale_oca: improve test mixin --- edi_sale_oca/tests/common.py | 5 +++-- edi_sale_oca/tests/test_order.py | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/edi_sale_oca/tests/common.py b/edi_sale_oca/tests/common.py index 2fc1f15406b..523642abb62 100644 --- a/edi_sale_oca/tests/common.py +++ b/edi_sale_oca/tests/common.py @@ -46,5 +46,6 @@ def _setup_order(cls, **kw): if line_defaults: for line in vals["order_line"]: line.update(line_defaults) - cls.sale = cls._create_sale_order(**vals) - cls.sale.action_confirm() + sale = cls._create_sale_order(**vals) + sale.action_confirm() + return sale diff --git a/edi_sale_oca/tests/test_order.py b/edi_sale_oca/tests/test_order.py index 8a182380758..da5591c512a 100644 --- a/edi_sale_oca/tests/test_order.py +++ b/edi_sale_oca/tests/test_order.py @@ -20,7 +20,7 @@ def setUpClass(cls): cls.exc_record_in = cls.backend.create_record( cls.exchange_type_in.code, {"edi_exchange_state": "input_received"} ) - cls._setup_order( + cls.order = cls._setup_order( origin_exchange_record_id=cls.exc_record_in.id, ) @@ -29,7 +29,7 @@ def _get_backend(cls): return cls.env.ref("edi_sale_oca.demo_edi_backend") def test_line_origin(self): - order = self.sale + order = self.order self.assertEqual(order.origin_exchange_record_id, self.exc_record_in) lines = order.order_line self.env["sale.order.line"].create(