97 lines
2.9 KiB
C#
97 lines
2.9 KiB
C#
using CodexPlugin;
|
|
using FileUtils;
|
|
using Logging;
|
|
using NUnit.Framework;
|
|
using Utils;
|
|
|
|
namespace ContinuousTests.Tests
|
|
{
|
|
public class TwoClientTest : ContinuousTest
|
|
{
|
|
private const string BytesStoredMetric = "codex_repostore_bytes_used";
|
|
|
|
public override int RequiredNumberOfNodes => 2;
|
|
public override TimeSpan RunTestEvery => TimeSpan.FromMinutes(2);
|
|
public override TestFailMode TestFailMode => TestFailMode.StopAfterFirstFailure;
|
|
|
|
private ContentId? cid;
|
|
private TrackedFile file = null!;
|
|
private readonly ByteSize size = 80.MB();
|
|
|
|
[TestMoment(t: Zero)]
|
|
public void UploadTestFile()
|
|
{
|
|
LogBlockExchangeStatus(Nodes[0], "Before upload");
|
|
LogBlockExchangeStatus(Nodes[1], "Before upload");
|
|
|
|
file = FileManager.GenerateFile(size);
|
|
|
|
LogStoredBytes(Nodes[0]);
|
|
|
|
LogBytesPerMillisecond(() => cid = Nodes[0].UploadFile(file));
|
|
Assert.That(cid, Is.Not.Null);
|
|
}
|
|
|
|
[TestMoment(t: 10)]
|
|
public void DownloadTestFile()
|
|
{
|
|
TrackedFile? dl = null;
|
|
|
|
try
|
|
{
|
|
LogBytesPerMillisecond(() => dl = Nodes[1].DownloadContent(cid!));
|
|
|
|
file.AssertIsEqual(dl);
|
|
}
|
|
catch
|
|
{
|
|
LogRepoStore(Nodes[0]);
|
|
LogRepoStore(Nodes[1]);
|
|
throw;
|
|
}
|
|
|
|
LogBlockExchangeStatus(Nodes[0], "After download");
|
|
LogBlockExchangeStatus(Nodes[1], "After download");
|
|
}
|
|
|
|
private void LogRepoStore(ICodexNode codexNode)
|
|
{
|
|
//var response = codexNode.GetDebugRepoStore();
|
|
//Log.Log($"{codexNode.GetName()} has {string.Join(",", response.Select(r => r.cid))}");
|
|
}
|
|
|
|
private void LogStoredBytes(ICodexNode node)
|
|
{
|
|
var metrics = CreateMetricsAccess(node);
|
|
var metric = metrics.GetMetric(BytesStoredMetric);
|
|
if (metric == null)
|
|
{
|
|
Log.Log($"Unabled to fetch metric '{BytesStoredMetric}' for node '{node.GetName()}'");
|
|
return;
|
|
}
|
|
|
|
var bytes = new ByteSize(Convert.ToInt64(metric.Values.Single().Value));
|
|
|
|
Log.Log($"{node.GetName()}: {bytes}");
|
|
}
|
|
|
|
private void LogBytesPerMillisecond(Action action)
|
|
{
|
|
var sw = Stopwatch.Begin(Log);
|
|
action();
|
|
var duration = sw.End();
|
|
double totalMs = duration.TotalMilliseconds;
|
|
double totalBytes = size.SizeInBytes;
|
|
|
|
var bytesPerMs = totalBytes / totalMs;
|
|
Log.Log($"Bytes per millisecond: {bytesPerMs}");
|
|
}
|
|
|
|
private void LogBlockExchangeStatus(ICodexNode codexNode, string msg)
|
|
{
|
|
//var response = codexNode.GetDebugBlockExchange();
|
|
//Log.Log($"{codexNode.GetName()} {msg}: {JsonConvert.SerializeObject(response)}");
|
|
}
|
|
}
|
|
}
|