Include stock moves with same location in upward/downward traces

Closes #12637

modules/stock_lot/stock.py

@@ -273,7 +273,13 @@ class LotTrace(ModelSQL, ModelView):
                     key=self._trace_move_order_key)))
 
     def _get_upward_traces(self):
-        return set()
+        pool = Pool()
+        Move = pool.get('stock.move')
+        return set(Move.search([
+                    ('lot', '=', self.lot.id),
+                    ('from_location', '=', self.to_location),
+                    ('effective_date', '>=', self.date),
+                    ]))
 
     def get_downward_traces(self, name):
         return list(map(int, sorted(filter(
@@ -281,7 +287,13 @@ class LotTrace(ModelSQL, ModelView):
                     key=self._trace_move_order_key, reverse=True)))
 
     def _get_downward_traces(self):
-        return set()
+        pool = Pool()
+        Move = pool.get('stock.move')
+        return set(Move.search([
+                    ('lot', '=', self.lot.id),
+                    ('to_location', '=', self.from_location),
+                    ('effective_date', '<=', self.date),
+                    ]))
 
 
 class LotByLocationContext(ModelView):

modules/stock_lot/tests/scenario_stock_lot_trace.rst

@@ -0,0 +1,84 @@
+========================
+Stock Lot Trace Scenario
+========================
+
+Imports::
+
+    >>> from decimal import Decimal
+
+    >>> from proteus import Model
+    >>> from trytond.modules.company.tests.tools import create_company, get_company
+    >>> from trytond.tests.tools import activate_modules
+
+Activate modules::
+
+    >>> config = activate_modules('stock_lot')
+
+    >>> Location = Model.get('stock.location')
+    >>> Lot = Model.get('stock.lot')
+    >>> LotTrace = Model.get('stock.lot.trace')
+    >>> Move = Model.get('stock.move')
+    >>> ProductTemplate = Model.get('product.template')
+    >>> UoM = Model.get('product.uom')
+
+Create company::
+
+    >>> _ = create_company()
+    >>> company = get_company()
+
+Create product::
+
+    >>> unit, = UoM.find([('name', '=', "Unit")])
+
+    >>> template = ProductTemplate(name="Product")
+    >>> template.type = 'goods'
+    >>> template.default_uom = unit
+    >>> template.save()
+    >>> product, = template.products
+
+Create lot::
+
+    >>> lot = Lot(product=product, number="1")
+    >>> lot.save()
+
+Get locations::
+
+    >>> supplier_loc, = Location.find([('code', '=', 'SUP')])
+    >>> storage_loc, = Location.find([('code', '=', 'STO')])
+    >>> customer_loc, = Location.find([('code', '=', 'CUS')])
+
+Make some moves::
+
+    >>> move_in = Move(product=product, lot=lot)
+    >>> move_in.quantity = 10
+    >>> move_in.from_location = supplier_loc
+    >>> move_in.to_location = storage_loc
+    >>> move_in.currency = company.currency
+    >>> move_in.unit_price = Decimal('0')
+    >>> move_in.click('do')
+    >>> move_in.state
+    'done'
+
+    >>> move_out = Move(product=product, lot=lot)
+    >>> move_out.quantity = 2
+    >>> move_out.from_location = storage_loc
+    >>> move_out.to_location = customer_loc
+    >>> move_out.currency = company.currency
+    >>> move_out.unit_price = Decimal('0')
+    >>> move_out.click('do')
+    >>> move_out.state
+    'done'
+
+Check lot traces::
+
+    >>> trace = LotTrace(move_in.id)
+    >>> trace.upward_traces == [LotTrace(move_out.id)]
+    True
+    >>> trace.downward_traces
+    []
+
+    >>> trace = LotTrace(move_out.id)
+    >>> trace.upward_traces
+    []
+    >>> trace.downward_traces == [LotTrace(move_in.id)]
+    True