Reconfigure script relationships.

immense · Jul 26, 2023 · 9bf5319 · 9bf5319
1 parent 6e1c231
commit 9bf5319
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 3 deletions.
diff --git a/Server/Data/AppDb.cs b/Server/Data/AppDb.cs
@@ -161,6 +161,10 @@ protected override void OnModelCreating(ModelBuilder builder)
             .HasMany(x => x.Results)
             .WithOne(x => x.ScriptRun)
             .IsRequired(false);
+        builder.Entity<ScriptRun>()
+            .HasOne(x => x.SavedScript)
+            .WithMany(x => x.ScriptRuns)
+            .IsRequired(false);
 
         builder.Entity<ScriptResult>()
           .Property(x => x.ErrorOutput)
@@ -181,6 +185,10 @@ protected override void OnModelCreating(ModelBuilder builder)
             .HasOne(x => x.ScriptRun)
             .WithMany(x => x.Results)
             .IsRequired(false);
+        builder.Entity<ScriptResult>()
+           .HasOne(x => x.SavedScript)
+           .WithMany(x => x.ScriptResults)
+           .IsRequired(false);
 
         builder.Entity<Alert>()
             .HasOne(x => x.User)

diff --git a/Shared/Models/SavedScript.cs b/Shared/Models/SavedScript.cs
@@ -47,4 +47,10 @@ public class SavedScript
     public string? SendErrorEmailTo { get; set; }
 
     public ScriptingShell Shell { get; set; }
+
+    [JsonIgnore]
+    public List<ScriptRun>? ScriptRuns { get; set; }
+
+    [JsonIgnore]
+    public List<ScriptResult>? ScriptResults { get; set; }
 }
diff --git a/Shared/Models/ScriptResult.cs b/Shared/Models/ScriptResult.cs
@@ -24,6 +24,9 @@ public class ScriptResult : ScriptResultBase
     public Organization? Organization { get; set; }
     public string OrganizationID { get; set; } = string.Empty;
 
+    [JsonIgnore]
+    public SavedScript? SavedScript { get; set; }
+
     [JsonIgnore]
     public ScriptSchedule? Schedule { get; set; }
 

diff --git a/Shared/Models/ScriptRun.cs b/Shared/Models/ScriptRun.cs
@@ -32,6 +32,12 @@ public class ScriptRun
 
     public DateTimeOffset RunAt { get; set; }
     public bool RunOnNextConnect { get; set; }
+
+    [JsonIgnore]
+    public SavedScript? SavedScript { get; set; }
     public Guid? SavedScriptId { get; set; }
+
+    [JsonIgnore]
+    public ScriptSchedule? Schedule { get; set; }
     public int? ScheduleId { get; set; }
 }
diff --git a/Tests/Server.Tests/DataServiceTests.cs b/Tests/Server.Tests/DataServiceTests.cs
@@ -162,18 +162,17 @@ public async Task GetPendingScriptRuns_GivenMultipleRunsQueued_ReturnsOnlyLatest
         Assert.AreEqual(1, pendingRuns.Count());
         Assert.AreEqual(2, pendingRuns.First().Id);
 
-        var scriptResult = new ScriptResult()
+        var dto = new ScriptResultDto()
         {
             DeviceID = _testData.Org1Device1.ID,
             InputType = Shared.Enums.ScriptInputType.ScheduledScript,
-            OrganizationID = _testData.Org1Id,
             SavedScriptId = savedScript.Id,
             ScriptRunId = scriptRun.Id,
             Shell = Shared.Enums.ScriptingShell.PSCore,
             ScriptInput = "echo test"
         };
 
-        _dataService.AddOrUpdateScriptResult(scriptResult);
+        var scriptResult = (await _dataService.AddScriptResult(dto)).Value!;
 
         await _dataService.AddScriptResultToScriptRun(scriptResult.ID, scriptRun.Id);