diff --git a/DistTestCore/Helpers/PeerDownloadTestHelpers.cs b/DistTestCore/Helpers/PeerDownloadTestHelpers.cs index b43b7abc..40024e13 100644 --- a/DistTestCore/Helpers/PeerDownloadTestHelpers.cs +++ b/DistTestCore/Helpers/PeerDownloadTestHelpers.cs @@ -48,11 +48,7 @@ namespace DistTestCore.Helpers try { - var downloadedFile = fileManager.CreateEmptyTestFile(expectedFile.Label + "_downloaded"); - using var downloadStream = File.OpenWrite(downloadedFile.Filename); - using var stream = to.Node.DownloadFile(contentId); - stream.CopyTo(downloadStream); - + var downloadedFile = DownloadFile(to.Node, contentId, expectedFile.Label + "_downloaded"); expectedFile.AssertIsEqual(downloadedFile); return PeerConnectionState.Connection; } @@ -70,6 +66,15 @@ namespace DistTestCore.Helpers // Should an exception occur during upload, then this try is inconclusive and we try again next loop. } + private TestFile DownloadFile(CodexAccess node, string contentId, string label) + { + var downloadedFile = fileManager.CreateEmptyTestFile(label); + using var downloadStream = File.OpenWrite(downloadedFile.Filename); + using var stream = node.DownloadFile(contentId); + stream.CopyTo(downloadStream); + return downloadedFile; + } + private TestFile GenerateTestFile(CodexAccess uploader, CodexAccess downloader) { var up = uploader.GetName().Replace("<", "").Replace(">", ""); diff --git a/Tests/PeerDiscoveryTests/PeerDiscoveryTests.cs b/Tests/PeerDiscoveryTests/PeerDiscoveryTests.cs index 170f8504..2ca67e60 100644 --- a/Tests/PeerDiscoveryTests/PeerDiscoveryTests.cs +++ b/Tests/PeerDiscoveryTests/PeerDiscoveryTests.cs @@ -20,7 +20,7 @@ namespace Tests.PeerDiscoveryTests [TestCase(3)] [TestCase(10)] [TestCase(20)] - public void VariableNodesInPods(int number) + public void VariableNodes(int number) { SetupCodexNodes(number);