cs-codex-dist-tests/Tests/CodexTests/BasicTests/OneClientTests.cs

using CodexPlugin;
using FileUtils;
using NUnit.Framework;
using System.Diagnostics;
using Utils;

namespace CodexTests.BasicTests
{
    [TestFixture]
    public class OneClientTests : CodexDistTest
    {
        [Test]
        public void OneClientTest()
        {
            var node = StartCodex();

            PerformOneClientTest(node);

            LogNodeStatus(node);
        }

        [Test]
        public void InterruptUploadTest()
        {
            var nodes = StartCodex(10);

            var tasks = nodes.Select(n => Task<bool>.Run(() => RunInterruptUploadTest(n)));
            Task.WaitAll(tasks.ToArray());

            Assert.That(tasks.Select(t => t.Result).All(r => r == true));
        }

        private bool RunInterruptUploadTest(ICodexNode node)
        {
            var file = GenerateTestFile(300.MB());

            var process = StartCurlUploadProcess(node, file);

            Thread.Sleep(500);
            process.Kill();
            Thread.Sleep(1000);

            var log = Ci.DownloadLog(node);
            return !log.GetLinesContaining("Unhandled exception in async proc, aborting").Any();
        }

        private Process StartCurlUploadProcess(ICodexNode node, TrackedFile file)
        {
            var apiAddress = node.Container.GetAddress(CodexContainerRecipe.ApiPortTag);
            var codexUrl = $"{apiAddress}/api/codex/v1/data";
            var filePath = file.Filename;
            return Process.Start("curl", $"-X POST {codexUrl} -H \"Content-Type: application/octet-stream\" -T {filePath}");
        }

        private void PerformOneClientTest(ICodexNode primary)
        {
            var testFile = GenerateTestFile(1.MB());

            var contentId = primary.UploadFile(testFile);

            var downloadedFile = primary.DownloadContent(contentId);

            testFile.AssertIsEqual(downloadedFile);
        }
    }
}