cs-codex-dist-tests/DistTestCore/OnlineCodexNode.cs

138 lines
4.5 KiB
C#
Raw Normal View History

2023-04-13 07:33:10 +00:00
using DistTestCore.Codex;
2023-04-14 07:54:07 +00:00
using DistTestCore.Logs;
2023-04-14 08:51:35 +00:00
using DistTestCore.Marketplace;
2023-04-13 12:36:17 +00:00
using DistTestCore.Metrics;
2023-04-13 07:33:10 +00:00
using NUnit.Framework;
namespace DistTestCore
{
public interface IOnlineCodexNode
{
string GetName();
2023-04-13 07:33:10 +00:00
CodexDebugResponse GetDebugInfo();
ContentId UploadFile(TestFile file);
TestFile? DownloadContent(ContentId contentId);
void ConnectToPeer(IOnlineCodexNode node);
2023-04-13 09:30:19 +00:00
ICodexNodeLog DownloadLog();
2023-04-13 12:36:17 +00:00
IMetricsAccess Metrics { get; }
2023-04-14 08:51:35 +00:00
IMarketplaceAccess Marketplace { get; }
2023-04-13 07:33:10 +00:00
}
public class OnlineCodexNode : IOnlineCodexNode
{
private const string SuccessfullyConnectedMessage = "Successfully connected to peer";
private const string UploadFailedMessage = "Unable to store block";
private readonly TestLifecycle lifecycle;
2023-04-14 08:51:35 +00:00
public OnlineCodexNode(TestLifecycle lifecycle, CodexAccess codexAccess, CodexNodeGroup group, IMetricsAccess metricsAccess, IMarketplaceAccess marketplaceAccess)
2023-04-13 07:33:10 +00:00
{
this.lifecycle = lifecycle;
CodexAccess = codexAccess;
Group = group;
2023-04-13 12:36:17 +00:00
Metrics = metricsAccess;
2023-04-14 08:51:35 +00:00
Marketplace = marketplaceAccess;
2023-04-13 07:33:10 +00:00
}
public CodexAccess CodexAccess { get; }
public CodexNodeGroup Group { get; }
2023-04-13 12:36:17 +00:00
public IMetricsAccess Metrics { get; }
2023-04-14 08:51:35 +00:00
public IMarketplaceAccess Marketplace { get; }
2023-04-13 07:33:10 +00:00
public string GetName()
{
2023-04-19 07:57:37 +00:00
return CodexAccess.Container.GetName();
2023-04-13 07:33:10 +00:00
}
public CodexDebugResponse GetDebugInfo()
{
var debugInfo = CodexAccess.GetDebugInfo();
2023-04-19 12:57:00 +00:00
Log($"Got DebugInfo with id: '{debugInfo.id}'.");
return debugInfo;
2023-04-13 07:33:10 +00:00
}
public ContentId UploadFile(TestFile file)
{
Log($"Uploading file of size {file.GetFileSize()}...");
using var fileStream = File.OpenRead(file.Filename);
var response = CodexAccess.UploadFile(fileStream);
if (response.StartsWith(UploadFailedMessage))
{
Assert.Fail("Node failed to store block.");
}
Log($"Uploaded file. Received contentId: '{response}'.");
return new ContentId(response);
}
public TestFile? DownloadContent(ContentId contentId)
{
Log($"Downloading for contentId: '{contentId.Id}'...");
var file = lifecycle.FileManager.CreateEmptyTestFile();
DownloadToFile(contentId.Id, file);
Log($"Downloaded file of size {file.GetFileSize()} to '{file.Filename}'.");
return file;
}
public void ConnectToPeer(IOnlineCodexNode node)
{
var peer = (OnlineCodexNode)node;
Log($"Connecting to peer {peer.GetName()}...");
var peerInfo = node.GetDebugInfo();
var response = CodexAccess.ConnectToPeer(peerInfo.id, GetPeerMultiAddress(peer, peerInfo));
Assert.That(response, Is.EqualTo(SuccessfullyConnectedMessage), "Unable to connect codex nodes.");
Log($"Successfully connected to peer {peer.GetName()}.");
}
2023-04-13 09:30:19 +00:00
public ICodexNodeLog DownloadLog()
{
return lifecycle.DownloadLog(this);
}
2023-04-13 07:33:10 +00:00
private string GetPeerMultiAddress(OnlineCodexNode peer, CodexDebugResponse peerInfo)
{
var multiAddress = peerInfo.addrs.First();
// Todo: Is there a case where First address in list is not the way?
if (Group == peer.Group)
{
return multiAddress;
}
// The peer we want to connect is in a different pod.
// We must replace the default IP with the pod IP in the multiAddress.
return multiAddress.Replace("0.0.0.0", peer.Group.Containers.RunningPod.Ip);
}
private void DownloadToFile(string contentId, TestFile file)
{
using var fileStream = File.OpenWrite(file.Filename);
2023-04-19 12:57:00 +00:00
try
{
using var downloadStream = CodexAccess.DownloadFile(contentId);
downloadStream.CopyTo(fileStream);
}
catch
{
Log($"Failed to download file '{contentId}'.");
throw;
}
2023-04-13 07:33:10 +00:00
}
private void Log(string msg)
{
lifecycle.Log.Log($"{GetName()}: {msg}");
}
}
public class ContentId
{
public ContentId(string id)
{
Id = id;
}
public string Id { get; }
}
}