Added method for get lines of unit loads link to origin.

This commit refs #16136

load.py

@@ -462,8 +462,7 @@ class LoadOrder(metaclass=PoolMeta):
     def get_failed_uls(self, unit_loads):
         failed_uls = []
         for unit_load in unit_loads:
-            lines = [l for l in self.lines if l.origin and
-                unit_load in l.origin.unit_loads]
+            lines = self._get_lines_from_origin_uls(unit_load)
             if not self.check_ul_data_match(lines, unit_load):
                 failed_uls.append(unit_load)
         return failed_uls
@@ -500,8 +499,7 @@ class LoadOrder(metaclass=PoolMeta):
                     self.rec_name))
 
             # check it is linked to origin lines
-            lines = [l for l in self.lines if l.origin and
-                unit_load in getattr(l.origin, 'unit_loads', [])]
+            lines = self._get_lines_from_origin_uls(unit_load)
             if not lines:
                 lines = self.check_origin_restrict(unit_load, origin_restrict,
                     origin_restrict_warn)
@@ -572,6 +570,10 @@ class LoadOrder(metaclass=PoolMeta):
     def valid_origin_restrict(cls):
         return ['sale.line']
 
+    def _get_lines_from_origin_uls(self, unit_load):
+        return [l for l in self.lines if l.origin and
+            unit_load in getattr(l.origin, 'unit_loads', [])]
+
     def check_ul_data_match(self, lines, unit_load):
         valid_lines = []
         for line in lines:

tests/scenario_force_unload.rst

@@ -75,13 +75,6 @@ Create chart of accounts::
     >>> accounts = get_accounts(company)
     >>> revenue = accounts['revenue']
     >>> expense = accounts['expense']
-    >>> cash = accounts['cash']
-
-    >>> Journal = Model.get('account.journal')
-    >>> cash_journal, = Journal.find([('type', '=', 'cash')])
-    >>> cash_journal.credit_account = cash
-    >>> cash_journal.debit_account = cash
-    >>> cash_journal.save()
 
 Create tax::
 

tests/scenario_load_sale.rst

@@ -39,13 +39,6 @@ Create chart of accounts::
     >>> accounts = get_accounts(company)
     >>> revenue = accounts['revenue']
     >>> expense = accounts['expense']
-    >>> cash = accounts['cash']
-
-    >>> Journal = Model.get('account.journal')
-    >>> cash_journal, = Journal.find([('type', '=', 'cash')])
-    >>> cash_journal.credit_account = cash
-    >>> cash_journal.debit_account = cash
-    >>> cash_journal.save()
 
 Create tax::
 

tests/scenario_load_ul.rst

@@ -39,13 +39,6 @@ Create chart of accounts::
     >>> accounts = get_accounts(company)
     >>> revenue = accounts['revenue']
     >>> expense = accounts['expense']
-    >>> cash = accounts['cash']
-
-    >>> Journal = Model.get('account.journal')
-    >>> cash_journal, = Journal.find([('type', '=', 'cash')])
-    >>> cash_journal.credit_account = cash
-    >>> cash_journal.debit_account = cash
-    >>> cash_journal.save()
 
 Create tax::
 

tests/scenario_load_ul_internal.rst

@@ -39,13 +39,6 @@ Create chart of accounts::
     >>> accounts = get_accounts(company)
     >>> revenue = accounts['revenue']
     >>> expense = accounts['expense']
-    >>> cash = accounts['cash']
-
-    >>> Journal = Model.get('account.journal')
-    >>> cash_journal, = Journal.find([('type', '=', 'cash')])
-    >>> cash_journal.credit_account = cash
-    >>> cash_journal.debit_account = cash
-    >>> cash_journal.save()
 
 Create tax::
 
@@ -98,7 +91,6 @@ Create carrier::
     >>> carrier_product = carrier.carrier_product.template
     >>> carrier_product.purchasable = True
     >>> carrier_product.purchase_uom = carrier_product.default_uom
-    >>> carrier_product.account_expense = expense
     >>> carrier_product.save()
 
 Get warehouse and dock::
@@ -170,8 +162,6 @@ Create unit load::
     ...         default_values={'start_date': datetime.datetime.now() + relativedelta(days=-1)})
     >>> template = ul.product.template
     >>> template.salable = True
-    >>> template.account_expense = expense
-    >>> template.account_revenue = revenue
     >>> template.save()
     >>> main_product = ul.product
 

tests/scenario_many_load_ul.rst

@@ -39,13 +39,6 @@ Create chart of accounts::
     >>> accounts = get_accounts(company)
     >>> revenue = accounts['revenue']
     >>> expense = accounts['expense']
-    >>> cash = accounts['cash']
-
-    >>> Journal = Model.get('account.journal')
-    >>> cash_journal, = Journal.find([('type', '=', 'cash')])
-    >>> cash_journal.credit_account = cash
-    >>> cash_journal.debit_account = cash
-    >>> cash_journal.save()
 
 Create tax::