feat: updated test from list to single model

NHSDigital · Jan 6, 2025 · d80ee86 · d80ee86
1 parent 91151f0
commit d80ee86
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...itTests/CohortDistributionTests/CohortDistributionDataTests/AddCohortDistributionTests.cs b/...itTests/CohortDistributionTests/CohortDistributionDataTests/AddCohortDistributionTests.cs
@@ -194,7 +194,7 @@ public void GetOutstandingCohortRequestAudits_NoOutstandingParticipants_ReturnsE
         var result = _createCohortDistributionData.GetNextCohortRequestAudit(lastRequestId);
 
         // Assert
-        Assert.AreEqual(0, result.Count);
+        Assert.IsNull(result);
     }
 
     [TestMethod]
@@ -216,8 +216,7 @@ public void GetOutstandingCohortRequestAudits_ValidRequestId_ReturnsAuditList()
 
         // Assert
         Assert.IsNotNull(result);
-        Assert.AreEqual(1, result.Count);
-        Assert.AreEqual("moreRecentRequestId", result[0].RequestId);
+        Assert.AreEqual("moreRecentRequestId", result.RequestId);
     }
 
     [TestMethod]