From e57c58bb9867273e189bf744123868047a56cc48 Mon Sep 17 00:00:00 2001
From: Neil South <neil.south@answerdigital.com>
Date: Thu, 23 May 2024 11:00:05 +0100
Subject: [PATCH] addded more code coverage

Signed-off-by: Neil South <neil.south@answerdigital.com>
---
 src/WorkflowManager/Common/Services/PayloadService.cs |  7 -------
 .../Common.Tests/Services/PayloadServiceTests.cs      | 11 +++++++++++
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/WorkflowManager/Common/Services/PayloadService.cs b/src/WorkflowManager/Common/Services/PayloadService.cs
index f2c7517bb..173854d39 100644
--- 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
index 179b63df7..7cec3977a 100644
--- 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));
+        }
+
     }
 }