[IMP] fieldservice_route: Update active dayroutes and orders on FSM r…

…oute person change

fieldservice_route/models/fsm_route.py

@@ -26,3 +26,26 @@ def run_on(self, date):
             day_index = date.weekday()
             day = self.env.ref("fieldservice_route.fsm_route_day_" + str(day_index))
             return day in self.day_ids
+
+    def write(self, vals):
+        if "fsm_person_id" in vals:
+            new_person_id = vals["fsm_person_id"]
+            today = fields.Date.today()
+
+            for route in self:
+                dayroutes = self.env["fsm.route.dayroute"].search(
+                    [
+                        ("route_id", "=", route.id),
+                        ("date", ">=", today),
+                    ]
+                )
+
+                dayroutes.write({"person_id": new_person_id})
+
+                for dayroute in dayroutes:
+                    orders_to_update = dayroute.order_ids.filtered(
+                        lambda o: o.fsm_route_id == route and not o.is_closed
+                    )
+                    orders_to_update.write({"person_id": new_person_id})
+
+        return super().write(vals)

fieldservice_route/tests/test_fsm_order.py

@@ -3,7 +3,7 @@
 # Copyright 2022 Tecnativa - Víctor Martínez
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
 
-from datetime import datetime
+from datetime import datetime, timedelta
 
 from odoo.tests import Form, common
 
@@ -14,8 +14,11 @@ def setUp(self):
         self.fsm_stage_obj = self.env["fsm.stage"]
         self.fsm_order_obj = self.env["fsm.order"]
         self.fsm_route_obj = self.env["fsm.route"]
+        self.fsm_dayroute_obj = self.env["fsm.route.dayroute"]
         self.test_person = self.env.ref("fieldservice.test_person")
+        self.person_1 = self.env.ref("fieldservice.person_1")
         self.test_location = self.env.ref("fieldservice.test_location")
+        self.stage_completed = self.env.ref("fieldservice.fsm_stage_completed")
         date = datetime.now()
         self.date = date.replace(microsecond=0)
         self.days = [
@@ -37,6 +40,51 @@ def setUp(self):
         )
         self.test_location.fsm_route_id = self.fsm_route_id.id
 
+        self.dayroute_today = self.fsm_dayroute_obj.create(
+            {
+                "name": "Today Day Route",
+                "route_id": self.fsm_route_id.id,
+                "date": self.date.date(),
+                "person_id": self.test_person.id,
+            }
+        )
+        self.dayroute_future = self.fsm_dayroute_obj.create(
+            {
+                "name": "Future Day Route",
+                "route_id": self.fsm_route_id.id,
+                "date": (self.date + timedelta(days=1)).date(),
+                "person_id": self.test_person.id,
+            }
+        )
+        self.dayroute_past = self.fsm_dayroute_obj.create(
+            {
+                "name": "Past Day Route",
+                "route_id": self.fsm_route_id.id,
+                "date": (self.date - timedelta(days=1)).date(),
+                "person_id": self.test_person.id,
+            }
+        )
+
+        self.active_order = self.fsm_order_obj.create(
+            {
+                "location_id": self.test_location.id,
+                "fsm_route_id": self.fsm_route_id.id,
+                "dayroute_id": self.dayroute_today.id,
+                "person_id": self.test_person.id,
+                "scheduled_date_start": self.date,
+            }
+        )
+        self.closed_order = self.fsm_order_obj.create(
+            {
+                "location_id": self.test_location.id,
+                "fsm_route_id": self.fsm_route_id.id,
+                "dayroute_id": self.dayroute_today.id,
+                "person_id": self.test_person.id,
+                "stage_id": self.stage_completed.id,
+                "scheduled_date_start": self.date,
+            }
+        )
+
     def test_create_day_route(self):
         order_form = Form(self.fsm_order_obj)
         order_form.location_id = self.test_location
@@ -47,3 +95,20 @@ def test_create_day_route(self):
         self.assertEqual(order.dayroute_id.person_id, order.person_id)
         self.assertEqual(order.dayroute_id.date, order.scheduled_date_start.date())
         self.assertEqual(order.dayroute_id.route_id, order.fsm_route_id)
+
+    def test_write_updates_dayroutes_and_orders(self):
+        # Update FSM person on the route
+        self.fsm_route_id.write({"fsm_person_id": self.person_1.id})
+
+        # Assert active day routes are updated
+        self.assertEqual(self.dayroute_today.person_id, self.person_1)
+        self.assertEqual(self.dayroute_future.person_id, self.person_1)
+
+        # Assert past day routes are not updated
+        self.assertEqual(self.dayroute_past.person_id, self.test_person)
+
+        # Assert active orders are updated
+        self.assertEqual(self.active_order.person_id, self.person_1)
+
+        # Assert closed orders are not updated
+        self.assertEqual(self.closed_order.person_id, self.test_person)