76 lines
3.0 KiB
C#
76 lines
3.0 KiB
C#
using DistTestCore.Codex;
|
|
using NUnit.Framework;
|
|
|
|
namespace DistTestCore.Helpers
|
|
{
|
|
public class PeerDownloadTestHelpers
|
|
{
|
|
private readonly DistTest test;
|
|
|
|
public PeerDownloadTestHelpers(DistTest test)
|
|
{
|
|
this.test = test;
|
|
}
|
|
|
|
public void AssertFullDownloadInterconnectivity(IEnumerable<IOnlineCodexNode> nodes, ByteSize testFileSize)
|
|
{
|
|
test.Log($"Asserting full download interconnectivity for nodes: '{string.Join(",", nodes.Select(n => n.GetName()))}'...");
|
|
var start = DateTime.UtcNow;
|
|
|
|
foreach (var node in nodes)
|
|
{
|
|
var uploader = node;
|
|
var downloaders = nodes.Where(n => n != uploader).ToArray();
|
|
|
|
test.ScopedTestFiles(() =>
|
|
{
|
|
PerformTest(uploader, downloaders, testFileSize);
|
|
});
|
|
}
|
|
|
|
test.Log($"Success! Full download interconnectivity for nodes: {string.Join(",", nodes.Select(n => n.GetName()))}");
|
|
var timeTaken = DateTime.UtcNow - start;
|
|
|
|
AssertTimePerMB(timeTaken, nodes.Count(), testFileSize);
|
|
}
|
|
|
|
private void AssertTimePerMB(TimeSpan timeTaken, int numberOfNodes, ByteSize size)
|
|
{
|
|
var numberOfDownloads = numberOfNodes * (numberOfNodes - 1);
|
|
var timePerDownload = timeTaken / numberOfDownloads;
|
|
float sizeInMB = size.ToMB();
|
|
var timePerMB = timePerDownload / sizeInMB;
|
|
|
|
test.Log($"Performed {numberOfDownloads} downloads of {size} in {timeTaken.TotalSeconds} seconds, for an average of {timePerMB} seconds per MB.");
|
|
|
|
Assert.That(timePerMB, Is.LessThan(CodexContainerRecipe.MaxDownloadTimePerMegabyte), "MaxDownloadTimePerMegabyte performance threshold breached.");
|
|
}
|
|
|
|
private void PerformTest(IOnlineCodexNode uploader, IOnlineCodexNode[] downloaders, ByteSize testFileSize)
|
|
{
|
|
// Generate 1 test file per downloader.
|
|
var files = downloaders.Select(d => GenerateTestFile(uploader, d, testFileSize)).ToArray();
|
|
|
|
// Upload all the test files to the uploader.
|
|
var contentIds = files.Select(uploader.UploadFile).ToArray();
|
|
|
|
// Each downloader should retrieve its own test file.
|
|
for (var i = 0; i < downloaders.Length; i++)
|
|
{
|
|
var expectedFile = files[i];
|
|
var downloadedFile = downloaders[i].DownloadContent(contentIds[i], $"{expectedFile.Label}DOWNLOADED");
|
|
|
|
expectedFile.AssertIsEqual(downloadedFile);
|
|
}
|
|
}
|
|
|
|
private TestFile GenerateTestFile(IOnlineCodexNode uploader, IOnlineCodexNode downloader, ByteSize testFileSize)
|
|
{
|
|
var up = uploader.GetName().Replace("<", "").Replace(">", "");
|
|
var down = downloader.GetName().Replace("<", "").Replace(">", "");
|
|
var label = $"FROM{up}TO{down}";
|
|
return test.GenerateTestFile(testFileSize, label);
|
|
}
|
|
}
|
|
}
|