Adapt initDb

equinor · Feb 20, 2025 · 6471e5a · 6471e5a
1 parent 8341505
commit 6471e5a
Showing 1 changed file with 35 additions and 9 deletions.
diff --git a/backend/api/Database/Context/InitDb.cs b/backend/api/Database/Context/InitDb.cs
@@ -1,4 +1,5 @@
-using Api.Database.Models;
+using System;
+using Api.Database.Models;
 using Microsoft.EntityFrameworkCore;
 using TaskStatus = Api.Database.Models.TaskStatus;
 
@@ -13,10 +14,11 @@ public static class InitDb
         private static readonly List<InspectionArea> inspectionAreas = GetInspectionAreas();
         private static readonly List<Area> areas = GetAreas();
         private static readonly List<Source> sources = GetSources();
-        private static readonly List<MissionTask> tasks = GetMissionTasks();
         private static readonly List<MissionDefinition> missionDefinitions =
             GetMissionDefinitions();
         private static readonly List<MissionRun> missionRuns = GetMissionRuns();
+
+        private static readonly List<MissionTask> tasks = GetMissionTasks();
         private static readonly List<AccessRole> accessRoles = GetAccessRoles();
 
         private static List<Inspection> GetInspections()
@@ -453,6 +455,14 @@ private static List<MissionTask> GetMissionTasks()
                 status: TaskStatus.Failed
             );
 
+            missionRuns[0].Tasks = [task1, task2];
+            missionRuns[1].Tasks = [task2, task3];
+            missionRuns[2].Tasks = [task3, task4];
+            missionRuns[3].Tasks = [task4, task5];
+            missionRuns[4].Tasks = [task5, task6];
+            missionRuns[5].Tasks = [task6, task7];
+            missionRuns[6].Tasks = [task7, task1];
+
             return [task1, task2, task3, task4, task5, task6, task7];
         }
 
@@ -504,7 +514,7 @@ private static List<MissionRun> GetMissionRuns()
                 MissionId = missionDefinitions[1].Id,
                 Status = MissionStatus.Failed,
                 DesiredStartTime = DateTime.UtcNow,
-                Tasks = [tasks[0], tasks[1]],
+                Tasks = [],
             };
 
             var missionRun5 = new MissionRun
@@ -516,7 +526,7 @@ private static List<MissionRun> GetMissionRuns()
                 MissionId = missionDefinitions[1].Id,
                 Status = MissionStatus.PartiallySuccessful,
                 DesiredStartTime = DateTime.UtcNow,
-                Tasks = [tasks[0], tasks[2]],
+                Tasks = [],
             };
 
             var missionRun6 = new MissionRun
@@ -528,7 +538,7 @@ private static List<MissionRun> GetMissionRuns()
                 MissionId = missionDefinitions[1].Id,
                 Status = MissionStatus.Cancelled,
                 DesiredStartTime = DateTime.UtcNow,
-                Tasks = [tasks[0], tasks[3]],
+                Tasks = [],
             };
 
             var missionRun7 = new MissionRun
@@ -540,10 +550,24 @@ private static List<MissionRun> GetMissionRuns()
                 MissionId = missionDefinitions[1].Id,
                 Status = MissionStatus.Failed,
                 DesiredStartTime = DateTime.UtcNow,
-                Tasks = [tasks[0], tasks[1], tasks[2], tasks[3], tasks[4], tasks[5], tasks[6]],
+                Tasks = [],
+            };
+
+            var inspectionArea = new InspectionArea
+            {
+                Id = Guid.NewGuid().ToString(),
+                Plant = plants[0],
+                Installation = plants[0].Installation,
+                Name = "TestInspectionArea",
             };
 
-            missionDefinitions[1].LastSuccessfulRun = missionRun3;
+            // missionDefinitions[1].LastSuccessfulRun = missionRun3;
+            // missionDefinitions[1].InspectionFrequency = TimeSpan.FromDays(1);
+            // missionDefinitions[1].StartDate = DateTime.Now.AddDays(-1);
+            // missionDefinitions[1].InspectionArea = inspectionArea;
+            // missionDefinitions[1].Source = new Source { SourceId = "990" };
+
+            // robots[0].CurrentInspectionArea = inspectionArea;
 
             return new List<MissionRun>(
                 [
@@ -598,14 +622,16 @@ public static void PopulateDb(FlotillaDbContext context)
             context.AddRange(areas);
             context.AddRange(sources);
 
+            context.AddRange(missionDefinitions);
+            context.AddRange(missionRuns);
+
             var tasks = GetMissionTasks();
             foreach (var task in tasks)
             {
                 task.Inspection = inspections[0];
             }
             context.AddRange(tasks);
-            context.AddRange(missionDefinitions);
-            context.AddRange(missionRuns);
+
             context.AddRange(accessRoles);
 
             context.SaveChanges();