change back to fix tests

Signed-off-by: Neil South <[email protected]>

src/InformaticsGateway/Services/Connectors/PayloadAssembler.cs

@@ -51,16 +51,14 @@ internal sealed partial class PayloadAssembler : IPayloadAssembler, IDisposable,
         private readonly Task _intializedTask;
         private readonly BlockingCollection<Payload> _workItems;
         private readonly System.Timers.Timer _timer;
-        private readonly IPayloadRepository _repository;
+        //private readonly IPayloadRepository _repository;
 
         public PayloadAssembler(
             ILogger<PayloadAssembler> logger,
             IServiceScopeFactory serviceScopeFactory)
         {
             _logger = logger ?? throw new ArgumentNullException(nameof(logger));
             _serviceScopeFactory = serviceScopeFactory ?? throw new ArgumentNullException(nameof(serviceScopeFactory));
-            var scope = _serviceScopeFactory.CreateScope();
-            _repository = scope.ServiceProvider.GetRequiredService<IPayloadRepository>();
 
             _workItems = [];
             _tokenSource = new CancellationTokenSource();
@@ -85,7 +83,9 @@ public PayloadAssembler(
         private async Task RemovePendingPayloads()
         {
             _logger.RemovingPendingPayloads();
-            var removed = await _repository.RemovePendingPayloadsAsync().ConfigureAwait(false);
+            var scope = _serviceScopeFactory.CreateScope();
+            var repository = scope.ServiceProvider.GetRequiredService<IPayloadRepository>();
+            var removed = await repository.RemovePendingPayloadsAsync().ConfigureAwait(false);
 
             _logger.TotalNumberOfPayloadsRemoved(removed);
         }
@@ -200,7 +200,9 @@ private async Task QueueBucketForNotification(string key, Payload payload)
             {
                 payload.State = Payload.PayloadState.Move;
                 var scope = _serviceScopeFactory.CreateScope();
-                await _repository.UpdateAsync(payload).ConfigureAwait(false);
+
+                var repository = scope.ServiceProvider.GetRequiredService<IPayloadRepository>();
+                await repository.UpdateAsync(payload).ConfigureAwait(false);
                 _logger.PayloadSaved(payload.PayloadId);
                 _workItems.Add(payload);
                 _logger.BucketReady(key, payload.Count);
@@ -227,7 +229,9 @@ private async Task<Payload> CreateOrGetPayload(string key, string correlationId,
         private async Task<Payload> PayloadFactory(string key, string correlationId, string? workflowInstanceId, string? taskId, Messaging.Events.DataOrigin dataOrigin, uint timeout, CancellationToken cancellationToken)
         {
             var newPayload = new Payload(key, correlationId, workflowInstanceId, taskId, dataOrigin, timeout, null);
-            await _repository.AddAsync(newPayload, cancellationToken).ConfigureAwait(false);
+            var scope = _serviceScopeFactory.CreateScope();
+            var repository = scope.ServiceProvider.GetRequiredService<IPayloadRepository>();
+            await repository.AddAsync(newPayload, cancellationToken).ConfigureAwait(false);
             _logger.BucketCreated(key, timeout);
             return newPayload;
         }