64 lines
1.8 KiB
C#
64 lines
1.8 KiB
C#
using NUnit.Framework;
|
|
|
|
namespace CodexDistTestCore
|
|
{
|
|
public interface IOnlineCodexNode
|
|
{
|
|
CodexDebugResponse GetDebugInfo();
|
|
ContentId UploadFile(TestFile file);
|
|
TestFile? DownloadContent(ContentId contentId);
|
|
}
|
|
|
|
public class OnlineCodexNode : IOnlineCodexNode
|
|
{
|
|
private readonly IFileManager fileManager;
|
|
private readonly CodexNodeContainer environment;
|
|
|
|
public OnlineCodexNode(IFileManager fileManager, CodexNodeContainer environment)
|
|
{
|
|
this.fileManager = fileManager;
|
|
this.environment = environment;
|
|
}
|
|
|
|
public CodexDebugResponse GetDebugInfo()
|
|
{
|
|
return Http().HttpGetJson<CodexDebugResponse>("debug/info");
|
|
}
|
|
|
|
public ContentId UploadFile(TestFile file)
|
|
{
|
|
using var fileStream = File.OpenRead(file.Filename);
|
|
var response = Http().HttpPostStream("upload", fileStream);
|
|
if (response.StartsWith("Unable to store block"))
|
|
{
|
|
Assert.Fail("Node failed to store block.");
|
|
}
|
|
return new ContentId(response);
|
|
}
|
|
|
|
public TestFile? DownloadContent(ContentId contentId)
|
|
{
|
|
var file = fileManager.CreateEmptyTestFile();
|
|
using var fileStream = File.OpenWrite(file.Filename);
|
|
using var downloadStream = Http().HttpGetStream("download/" + contentId.Id);
|
|
downloadStream.CopyTo(fileStream);
|
|
return file;
|
|
}
|
|
|
|
private Http Http()
|
|
{
|
|
return new Http(ip: "127.0.0.1", port: environment.ServicePort, baseUrl: "/api/codex/v1");
|
|
}
|
|
}
|
|
|
|
public class ContentId
|
|
{
|
|
public ContentId(string id)
|
|
{
|
|
Id = id;
|
|
}
|
|
|
|
public string Id { get; }
|
|
}
|
|
}
|