From 2263ec40d314760e91a04612c5110ccb30e577bc Mon Sep 17 00:00:00 2001
From: Sam Rooke <sam.rooke@answerdigital.com>
Date: Mon, 8 May 2023 21:00:23 +0100
Subject: [PATCH] fix payload delete code

Signed-off-by: Sam Rooke <sam.rooke@answerdigital.com>
---
 src/WorkflowManager/Database/Repositories/PayloadRepository.cs | 3 +--
 .../InformaticsGateway/InformaticsGatewayServiceTests.cs       | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/WorkflowManager/Database/Repositories/PayloadRepository.cs b/src/WorkflowManager/Database/Repositories/PayloadRepository.cs
index 1b4b7383d..3fc10ce8a 100755
--- a/src/WorkflowManager/Database/Repositories/PayloadRepository.cs
+++ b/src/WorkflowManager/Database/Repositories/PayloadRepository.cs
@@ -107,8 +107,7 @@ public async Task<bool> UpdateAsync(Payload payload)
             try
             {
                 var filter = Builders<Payload>.Filter.Eq(p => p.PayloadId, payload.PayloadId);
-                var update = Builders<Payload>.Update.Set(p => p, payload);
-                await _payloadCollection.UpdateOneAsync(filter, update);
+                await _payloadCollection.ReplaceOneAsync(filter, payload);
 
                 return true;
             }
diff --git a/tests/UnitTests/WorkflowManager.Services.Tests/InformaticsGateway/InformaticsGatewayServiceTests.cs b/tests/UnitTests/WorkflowManager.Services.Tests/InformaticsGateway/InformaticsGatewayServiceTests.cs
index 8df0d17c9..5a69c8926 100644
--- a/tests/UnitTests/WorkflowManager.Services.Tests/InformaticsGateway/InformaticsGatewayServiceTests.cs
+++ b/tests/UnitTests/WorkflowManager.Services.Tests/InformaticsGateway/InformaticsGatewayServiceTests.cs
@@ -67,7 +67,7 @@ public async Task OriginsExist_InvalidSource_ReturnsFalse()
         }
 
         [Fact]
-        public async Task OriginsExist_InvalidSource_ReturnsTrue()
+        public async Task OriginsExist_ValidSource_ReturnsTrue()
         {
             var source = "valid_source";