Skip to content

Commit 46d62b1

Browse files
committed
gh-347 Fix unit test
Signed-off-by: Victor Chang <[email protected]>
1 parent f354218 commit 46d62b1

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

src/InformaticsGateway/Test/Services/Connectors/PayloadAssemblerTest.cs

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -120,14 +120,19 @@ public async Task GivenAPayloadThatHasNotCompleteUploads_WhenProcessedByTimedEve
120120
{
121121
var payloadAssembler = new PayloadAssembler(_options, _logger.Object, _serviceScopeFactory.Object);
122122

123-
var file = new TestStorageInfo(Guid.NewGuid().ToString(), Guid.NewGuid().ToString(), "file1", ".txt");
123+
var file1 = new TestStorageInfo(Guid.NewGuid().ToString(), Guid.NewGuid().ToString(), "file1", ".txt");
124+
var file2 = new TestStorageInfo(Guid.NewGuid().ToString(), Guid.NewGuid().ToString(), "file1", ".txt");
124125

125-
await payloadAssembler.Queue("A", file, 1);
126+
await payloadAssembler.Queue("A", file1, 1);
127+
await payloadAssembler.Queue("A", file2, 1);
128+
129+
file1.SetFailed();
130+
file2.SetUploaded();
126131
await Task.Delay(1001);
127132
payloadAssembler.Dispose();
128133

129134
_repository.Verify(p => p.UpdateAsync(It.Is<Payload>(p => p.State == Payload.PayloadState.Move), It.IsAny<CancellationToken>()), Times.Never());
130-
_logger.VerifyLoggingMessageBeginsWith("Payload deleted due to upload failure(s)", LogLevel.Error, Times.Once());
135+
_logger.VerifyLoggingMessageBeginsWith($"Payload (A) with 2 files deleted due to 1 upload failure(s).", LogLevel.Error, Times.Once());
131136
}
132137

133138
[RetryFact(10, 200)]

src/Shared/Test/TestStorageInfo.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616

1717
using Monai.Deploy.InformaticsGateway.Api.Storage;
18+
using Monai.Deploy.Messaging;
1819

1920
namespace Monai.Deploy.InformaticsGateway.SharedTest;
2021

@@ -33,4 +34,9 @@ public TestStorageInfo(string correlationsId, string identifier, string filePath
3334
public override string DataTypeDirectoryName => "dir";
3435

3536
public override StorageObjectMetadata File { get; set; }
37+
38+
public void SetUploaded()
39+
{
40+
File.SetUploaded("test");
41+
}
3642
}

0 commit comments

Comments
 (0)