addded more code coverage

Signed-off-by: Neil South <[email protected]>
Project-MONAI · May 23, 2024 · e57c58b · e57c58b
1 parent 8b09c91
commit e57c58b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/src/WorkflowManager/Common/Services/PayloadService.cs b/src/WorkflowManager/Common/Services/PayloadService.cs
@@ -252,12 +252,5 @@ public Task<bool> UpdateAsyncWorkflowIds(Payload payload)
 
             return _payloadRepository.UpdateAsyncWorkflowIds(payload);
         }
-
-        public Task<bool> UpdateAsync(Payload payload)
-        {
-            ArgumentNullException.ThrowIfNull(payload, nameof(payload));
-
-            return _payloadRepository.UpdateAsyncWorkflowIds(payload);
-        }
     }
 }
diff --git a/tests/UnitTests/Common.Tests/Services/PayloadServiceTests.cs b/tests/UnitTests/Common.Tests/Services/PayloadServiceTests.cs
@@ -488,5 +488,16 @@ public async Task PayloadServiceCreate_Should_Call_GetExpiry()
             Assert.Equal(99, (int)daysdiff);
         }
 
+
+        [Fact]
+        public async Task UpdateAsync_NullPayload_ThrowsArgumentNullException()
+        {
+            // Arrange
+            Payload payload = null;
+
+            // Act & Assert
+            await Assert.ThrowsAsync<ArgumentNullException>(() => PayloadService.UpdateAsyncWorkflowIds(payload));
+        }
+
     }
 }