changes for HL7Configuration endpoint

Signed-off-by: Lillie Dae <lillie.dae@answerdigital.com>
Project-MONAI · Nov 16, 2023 · 95e4c7d · 95e4c7d
1 parent 3c9a77a
commit 95e4c7d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Database/EntityFramework/Repositories/Hl7ApplicationConfigRepository.cs b/src/Database/EntityFramework/Repositories/Hl7ApplicationConfigRepository.cs
@@ -21,6 +21,7 @@
 using Microsoft.Extensions.Options;
 using Monai.Deploy.InformaticsGateway.Api;
 using Monai.Deploy.InformaticsGateway.Configuration;
+using Monai.Deploy.InformaticsGateway.Database.Api;
 using Monai.Deploy.InformaticsGateway.Database.Api.Logging;
 using Monai.Deploy.InformaticsGateway.Database.Api.Repositories;
 using Polly;
@@ -63,6 +64,10 @@ public Task<Hl7ApplicationConfigEntity> DeleteAsync(string id, CancellationToken
             return _retryPolicy.ExecuteAsync(async () =>
             {
                 var entity = await GetByIdAsync(id).ConfigureAwait(false);
+                if (entity is null)
+                {
+                    throw new DatabaseException("Failed to delete entity.");
+                }
                 var result = _dataset.Remove(entity);
                 await _informaticsGatewayContext.SaveChangesAsync(cancellationToken).ConfigureAwait(false);
                 return result.Entity;