public async Task DicomSendFilesTest()
{
var dataSender = new DicomDataSender();
var applicationEntity = new GatewayApplicationEntity("RListenerTest", 131, "127.0.0.1");
var dicomFiles = new DirectoryInfo(@"Images\1ValidSmall\").GetFiles().Select(x => DicomFile.Open(x.FullName)).ToArray();
var rtFile = await DicomFile.OpenAsync(@"Images\LargeSeriesWithContour\rtstruct.dcm").ConfigureAwait(false);
var resultsDirectory = CreateTemporaryDirectory();
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultsDirectory.FullName)))
{
StartDicomDataReceiver(dicomDataReceiver, applicationEntity.Port);
var results = await dataSender.SendFilesAsync(
"Hello",
applicationEntity.Title,
applicationEntity.Port,
applicationEntity.IpAddress,
dicomFiles).ConfigureAwait(false);
foreach (var result in results)
{
// Check we can send non-RT files
Assert.AreEqual(DicomOperationResult.Success, result.Item2);
}
var rtResult = await dataSender.SendFilesAsync(
"Hello",
applicationEntity.Title,
applicationEntity.Port,
applicationEntity.IpAddress,
rtFile).ConfigureAwait(false);
Assert.AreEqual(1, rtResult.Count());
foreach (var result in rtResult)
{
// Check we can send RT files
Assert.AreEqual(DicomOperationResult.Success, result.Item2);
}
dicomDataReceiver.StopServer();
}
}
public async Task DicomDataReceiverServerStarts()
{
var applicationEntity = new GatewayApplicationEntity("RListenerTest", 191, "127.0.0.1");
var resultsDirectory = CreateTemporaryDirectory();
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultsDirectory.FullName)))
{
StartDicomDataReceiver(dicomDataReceiver, applicationEntity.Port);
var eventCount = 0;
var folderPath = string.Empty;
dicomDataReceiver.DataReceived += (sender, e) =>
{
folderPath = e.FolderPath;
Interlocked.Increment(ref eventCount);
};
Assert.ThrowsException <DicomNetworkException>(() => StartDicomDataReceiver(dicomDataReceiver, applicationEntity.Port));
var dataSender = new DicomDataSender();
var echoResult = await dataSender.DicomEchoAsync(
"RListener",
applicationEntity.Title,
applicationEntity.Port,
applicationEntity.IpAddress);
// Check echo
Assert.IsTrue(echoResult == DicomOperationResult.Success);
DcmtkHelpers.SendFileUsingDCMTK(
@"Images\1ValidSmall\1.dcm",
applicationEntity.Port,
ScuProfile.LEExplicitCT,
TestContext);
// Wait for all events to finish on the data received
SpinWait.SpinUntil(() => eventCount >= 3, TimeSpan.FromSeconds(10));
// Check the file exists
Assert.IsTrue(File.Exists(Path.Combine(folderPath, @"1.2.840.113619.2.81.290.1.36662.3.1.20151027.220159.dcm")));
dicomDataReceiver.StopServer();
}
}
public async Task ReceiveServiceEchoTest()
{
var testAETConfigModel = GetTestAETConfigModel();
var receivePort = 180;
using (var receiveService = CreateReceiveService(receivePort))
using (var uploadQueue = receiveService.UploadQueue)
{
uploadQueue.Clear();
receiveService.Start();
var sender = new DicomDataSender();
await sender.DicomEchoAsync(
"Hello",
testAETConfigModel.CalledAET,
receivePort,
"127.0.0.1").ConfigureAwait(false);
// Check nothing is added to the message queue
Assert.ThrowsException <MessageQueueReadException>(() => TransactionalDequeue <UploadQueueItem>(uploadQueue, timeoutMs: 1000));
// Now check when we send a file a message is added
DcmtkHelpers.SendFileUsingDCMTK(
@"Images\1ValidSmall\1.dcm",
receivePort,
ScuProfile.LEExplicitCT,
TestContext,
applicationEntityTitle: testAETConfigModel.CallingAET,
calledAETitle: testAETConfigModel.CalledAET);
Assert.IsNotNull(TransactionalDequeue <UploadQueueItem>(uploadQueue, timeoutMs: 1000));
// Now try another Dicom echo
await sender.DicomEchoAsync(
"Hello",
testAETConfigModel.CalledAET,
receivePort,
"127.0.0.1").ConfigureAwait(false);
// Check nothing is added to the message queue
Assert.ThrowsException <MessageQueueReadException>(() => TransactionalDequeue <UploadQueueItem>(uploadQueue, timeoutMs: 1000));
}
}
public async Task DicomEchoTest()
{
var dataSender = new DicomDataSender();
var applicationEntity = new GatewayApplicationEntity(
title: "RListenerTest",
port: new Random().Next(130, ApplicationEntityValidationHelpers.MaximumPortNumber),
ipAddress: "127.0.0.1");
var resultsDirectory = CreateTemporaryDirectory();
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultsDirectory.FullName)))
{
StartDicomDataReceiver(dicomDataReceiver, applicationEntity.Port);
var result1 = await dataSender.DicomEchoAsync(
"Hello",
applicationEntity.Title,
applicationEntity.Port,
applicationEntity.IpAddress).ConfigureAwait(false);
Assert.AreEqual(DicomOperationResult.Success, result1);
dicomDataReceiver.StopServer();
}
var result2 = await dataSender.DicomEchoAsync(
"Hello",
applicationEntity.Title,
applicationEntity.Port,
applicationEntity.IpAddress).ConfigureAwait(false);
Assert.AreEqual(DicomOperationResult.NoResponse, result2);
// Try ping with IPv6 Address
var result3 = await dataSender.DicomEchoAsync("Hello", "RListenerTest", 105, "2a00:1450:4009:800::200e").ConfigureAwait(false);
Assert.AreEqual(DicomOperationResult.NoResponse, result3);
}
public async Task GatewayLiveEndToEndTest()
{
var testAETConfigModel = GetTestAETConfigModel();
var resultDirectory = CreateTemporaryDirectory();
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultDirectory.FullName)))
{
var eventCount = 0;
var folderPath = string.Empty;
dicomDataReceiver.DataReceived += (sender, e) =>
{
folderPath = e.FolderPath;
Interlocked.Increment(ref eventCount);
};
StartDicomDataReceiver(dicomDataReceiver, testAETConfigModel.AETConfig.Destination.Port);
var receivePort = 141;
using (var segmentationClient = GetMockInnerEyeSegmentationClient())
using (var deleteService = CreateDeleteService())
using (var pushService = CreatePushService())
using (var downloadService = CreateDownloadService(segmentationClient))
using (var uploadService = CreateUploadService(segmentationClient))
using (var receiveService = CreateReceiveService(receivePort))
{
deleteService.Start();
pushService.Start();
downloadService.Start();
uploadService.Start();
receiveService.Start();
var dicomDataSender = new DicomDataSender();
var echoResult = await dicomDataSender.DicomEchoAsync(
testAETConfigModel.CallingAET,
testAETConfigModel.CalledAET,
receivePort,
"127.0.0.1").ConfigureAwait(false);
Assert.IsTrue(echoResult == DicomOperationResult.Success);
DcmtkHelpers.SendFolderUsingDCMTK(
@"Images\1ValidSmall",
receivePort,
ScuProfile.LEExplicitCT,
TestContext,
applicationEntityTitle: testAETConfigModel.CallingAET,
calledAETitle: testAETConfigModel.CalledAET);
// Wait for all events to finish on the data received
SpinWait.SpinUntil(() => eventCount >= 3, TimeSpan.FromMinutes(3));
#pragma warning disable CA1508 // Avoid dead conditional code
Assert.IsFalse(string.IsNullOrWhiteSpace(folderPath));
#pragma warning restore CA1508 // Avoid dead conditional code
var dicomFile = await DicomFile.OpenAsync(new DirectoryInfo(folderPath).GetFiles()[0].FullName).ConfigureAwait(false);
Assert.IsNotNull(dicomFile);
}
}
}
public async Task PushServiceTest()
{
var tempFolder = CreateTemporaryDirectory();
// Copy all files in the P4_Prostate directory to the temporary directory
Directory.EnumerateFiles(@"Images\1ValidSmall\")
.Select(x => new FileInfo(x))
.ToList()
.ForEach(x => x.CopyTo(Path.Combine(tempFolder.FullName, x.Name)));
var applicationEntity = new GatewayApplicationEntity("RListenerTest", 108, "127.0.0.1");
var resultDirectory = CreateTemporaryDirectory();
// Create a Data receiver to receive the RT struct result
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultDirectory.FullName)))
{
var eventCount = 0;
dicomDataReceiver.DataReceived += (sender, e) =>
{
Interlocked.Increment(ref eventCount);
};
StartDicomDataReceiver(dicomDataReceiver, applicationEntity.Port);
var dataSender = new DicomDataSender();
var echoResult = await dataSender.DicomEchoAsync("RListener", applicationEntity.Title, applicationEntity.Port, applicationEntity.IpAddress);
// Check echo
Assert.IsTrue(echoResult == DicomOperationResult.Success);
var testAETConfigModel = GetTestAETConfigModel();
using (var deleteService = CreateDeleteService())
using (var pushService = CreatePushService())
using (var pushQueue = pushService.PushQueue)
{
deleteService.Start();
pushService.Start();
TransactionalEnqueue(
pushQueue,
new PushQueueItem(
destinationApplicationEntity: applicationEntity,
calledApplicationEntityTitle: testAETConfigModel.CalledAET,
callingApplicationEntityTitle: applicationEntity.Title,
associationGuid: Guid.NewGuid(),
associationDateTime: DateTime.UtcNow,
filePaths: tempFolder.GetFiles().Select(x => x.FullName).ToArray()));
// Wait for all events to finish on the data received
SpinWait.SpinUntil(() => eventCount >= 3, TimeSpan.FromMinutes(3));
SpinWait.SpinUntil(() => new DirectoryInfo(tempFolder.FullName).Exists == false, TimeSpan.FromSeconds(30));
Assert.IsFalse(new DirectoryInfo(tempFolder.FullName).Exists);
Assert.AreEqual(20, resultDirectory.GetDirectories()[0].GetFiles().Length);
}
}
}
public async Task PushServiceBadAetTest()
{
var testAETConfigModel = GetTestAETConfigModel();
var destination = testAETConfigModel.AETConfig.Destination;
var tempFolder = CreateTemporaryDirectory();
// Grab a structure set file
var file = new FileInfo(@"Images\LargeSeriesWithContour\rtstruct.dcm");
file.CopyTo(Path.Combine(tempFolder.FullName, file.Name));
var resultDirectory = CreateTemporaryDirectory();
// Create a Data receiver to receive the RT struct result
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultDirectory.FullName)))
{
var eventCount = 0;
var folderPath = string.Empty;
dicomDataReceiver.DataReceived += (sender, e) =>
{
folderPath = e.FolderPath;
Interlocked.Increment(ref eventCount);
};
StartDicomDataReceiver(dicomDataReceiver, destination.Port);
var dataSender = new DicomDataSender();
var echoResult = await dataSender.DicomEchoAsync(
"RListener",
destination.Title,
destination.Port,
destination.Ip);
// Check echo
Assert.IsTrue(echoResult == DicomOperationResult.Success);
using (var deleteService = CreateDeleteService())
using (var pushService = CreatePushService())
using (var pushQueue = pushService.PushQueue)
{
deleteService.Start();
pushService.Start();
TransactionalEnqueue(
pushQueue,
new PushQueueItem(
destinationApplicationEntity: new GatewayApplicationEntity("", -1, "ababa"),
calledApplicationEntityTitle: testAETConfigModel.CalledAET,
callingApplicationEntityTitle: testAETConfigModel.CallingAET,
associationGuid: Guid.NewGuid(),
associationDateTime: DateTime.UtcNow,
filePaths: tempFolder.GetFiles().Select(x => x.FullName).ToArray()));
// Wait for all events to finish on the data received
SpinWait.SpinUntil(() => eventCount >= 3, TimeSpan.FromMinutes(3));
SpinWait.SpinUntil(() => new DirectoryInfo(tempFolder.FullName).Exists == false, TimeSpan.FromSeconds(30));
Assert.IsFalse(new DirectoryInfo(tempFolder.FullName).Exists);
}
}
}
public async Task IntegrationTestEndToEnd()
{
var sourceDirectory = CreateTemporaryDirectory();
var resultDirectory = CreateTemporaryDirectory();
var random = new Random();
// Get file names for all in directory
var sourceDicomFileNames = new DirectoryInfo(@"Images\HN").GetFiles().ToArray();
// Load all DICOM files
var sourceDicomFiles = sourceDicomFileNames.Select(f => DicomFile.Open(f.FullName, FileReadOption.ReadAll)).ToArray();
// Add/Update random tags for the source DICOM files.
DicomAnonymisationTests.AddRandomTags(random, sourceDicomFiles);
// Save them all to the sourceDirectory.
var sourcePairs = sourceDicomFileNames.Zip(sourceDicomFiles, (f, d) => Tuple.Create(f, d)).ToArray();
foreach (var sourcePair in sourcePairs)
{
var sourceImageFilePath = Path.Combine(sourceDirectory.FullName, sourcePair.Item1.Name);
sourcePair.Item2.Save(sourceImageFilePath);
}
// Keep the first as a reference for deanonymization later.
var originalSlice = sourceDicomFiles.First();
var testAETConfigModel = GetTestAETConfigModel();
var receivePort = 160;
using (var dicomDataReceiver = new ListenerDataReceiver(new ListenerDicomSaver(resultDirectory.FullName)))
using (var deleteService = CreateDeleteService())
using (var pushService = CreatePushService())
using (var downloadService = CreateDownloadService())
using (var uploadService = CreateUploadService())
using (var receiveService = CreateReceiveService(receivePort))
{
// Start a DICOM receiver for the final DICOM-RT file
var eventCount = new ConcurrentDictionary <DicomReceiveProgressCode, int>();
var folderPath = string.Empty;
dicomDataReceiver.DataReceived += (sender, e) =>
{
folderPath = e.FolderPath;
eventCount.AddOrUpdate(e.ProgressCode, 1, (k, v) => v + 1);
};
StartDicomDataReceiver(dicomDataReceiver, testAETConfigModel.AETConfig.Destination.Port);
// Start the services.
deleteService.Start();
pushService.Start();
downloadService.Start();
uploadService.Start();
receiveService.Start();
// Try a DICOM C-ECHO
var dicomDataSender = new DicomDataSender();
var echoResult = await dicomDataSender.DicomEchoAsync(
testAETConfigModel.CallingAET,
testAETConfigModel.CalledAET,
receivePort,
"127.0.0.1").ConfigureAwait(false);
Assert.IsTrue(echoResult == DicomOperationResult.Success);
// Send the image stack
DcmtkHelpers.SendFolderUsingDCMTK(
sourceDirectory.FullName,
receivePort,
ScuProfile.LEExplicitCT,
TestContext,
applicationEntityTitle: testAETConfigModel.CallingAET,
calledAETitle: testAETConfigModel.CalledAET);
// Wait for DICOM-RT file to be received.
Func <DicomReceiveProgressCode, int, bool> TestEventCount = (progressCode, count) =>
eventCount.ContainsKey(progressCode) && eventCount[progressCode] == count;
SpinWait.SpinUntil(() => TestEventCount(DicomReceiveProgressCode.AssociationEstablished, 1));
SpinWait.SpinUntil(() => TestEventCount(DicomReceiveProgressCode.FileReceived, 1));
SpinWait.SpinUntil(() => TestEventCount(DicomReceiveProgressCode.AssociationReleased, 1));
SpinWait.SpinUntil(() => TestEventCount(DicomReceiveProgressCode.ConnectionClosed, 1));
Assert.IsTrue(eventCount[DicomReceiveProgressCode.AssociationEstablished] == 1);
Assert.IsTrue(eventCount[DicomReceiveProgressCode.FileReceived] == 1);
Assert.IsTrue(eventCount[DicomReceiveProgressCode.AssociationReleased] == 1);
Assert.IsTrue(eventCount[DicomReceiveProgressCode.ConnectionClosed] == 1);
var receivedFiles = new DirectoryInfo(folderPath).GetFiles();
Assert.AreEqual(1, receivedFiles.Length);
var receivedFilePath = receivedFiles.First().FullName;
var dicomFile = await DicomFile.OpenAsync(receivedFilePath, FileReadOption.ReadAll).ConfigureAwait(false);
Assert.IsNotNull(dicomFile);
var matchedModel = ApplyAETModelConfigProvider.ApplyAETModelConfig(testAETConfigModel.AETConfig.Config.ModelsConfig, sourceDicomFiles);
var segmentationClient = TestGatewayProcessorConfigProvider.CreateInnerEyeSegmentationClient().Invoke();
DicomAnonymisationTests.AssertDeanonymizedFile(
originalSlice,
dicomFile,
segmentationClient.TopLevelReplacements,
matchedModel.Result.TagReplacements,
false);
AssertIsDicomRtFile(DateTime.UtcNow, dicomFile, matchedModel.Result.ModelId);
}
}
