logos-storage-nim-cs-dist-tests/Tests/CodexReleaseTests/MarketTests/RepairTest.cs

using CodexClient;
using Nethereum.Hex.HexConvertors.Extensions;
using NUnit.Framework;
using Utils;

namespace CodexReleaseTests.MarketTests
{
    [TestFixture]
    public class RepairTest : MarketplaceAutoBootstrapDistTest
    {
        #region Setup

        private readonly ByteSize Filesize;
        private readonly uint Slots;
        private readonly uint Tolerance;
        private readonly ByteSize EncodedFilesize;
        private readonly ByteSize SlotSize;

        public RepairTest()
        {
            Filesize = 32.MB();
            Slots = 4;
            Tolerance = 2;

            EncodedFilesize = new ByteSize(Filesize.SizeInBytes * (Slots / Tolerance));
            SlotSize = new ByteSize(EncodedFilesize.SizeInBytes / Slots);
            Assert.That(IsPowerOfTwo(SlotSize));
            Assert.That(Slots, Is.LessThan(NumberOfHosts));
        }

        protected override int NumberOfHosts => 5;
        protected override int NumberOfClients => 1;
        protected override ByteSize HostAvailabilitySize => SlotSize.Multiply(1.1); // Each host can hold 1 slot.
        protected override TimeSpan HostAvailabilityMaxDuration => TimeSpan.FromDays(5.0);

        private static bool IsPowerOfTwo(ByteSize size)
        {
            var x = size.SizeInBytes;
            return (x != 0) && ((x & (x - 1)) == 0);
        }

        #endregion

        [Ignore("Test is ready. Waiting for repair implementation.")]
        [Test]
        [Combinatorial]
        public void RollingRepairSingleFailure(
            [Values(10)] int numFailures)
        {
            var hosts = StartHosts().ToList();
            var client = StartClients().Single();

            var contract = CreateStorageRequest(client);
            contract.WaitForStorageContractStarted();
            // All slots are filled.

            for (var i = 0; i < numFailures; i++)
            {
                Log($"Failure step: {i}");

                // Start a new host. Add it to the back of the list:
                hosts.Add(StartOneHost());

                var fill = GetSlotFillByOldestHost(hosts);

                Log($"Causing failure for host: {fill.Host.GetName()} slotIndex: {fill.SlotFilledEvent.SlotIndex}");
                hosts.Remove(fill.Host);
                fill.Host.Stop(waitTillStopped: true);

                // The slot should become free.
                WaitForSlotFreedEvent(contract, fill.SlotFilledEvent.SlotIndex);

                // One of the other hosts should pick up the free slot.
                WaitForNewSlotFilledEvent(contract, fill.SlotFilledEvent.SlotIndex);
            }
        }

        private void WaitForSlotFreedEvent(IStoragePurchaseContract contract, ulong slotIndex)
        {
            Log(nameof(WaitForSlotFreedEvent));
            var start = DateTime.UtcNow;
            var timeout = CalculateContractFailTimespan();

            while (DateTime.UtcNow < start + timeout)
            {
                var events = GetContracts().GetEvents(GetTestRunTimeRange());
                var slotsFreed = events.GetSlotFreedEvents();
                Log($"Slots freed this period: {slotsFreed.Length}");

                foreach (var free in slotsFreed)
                {
                    if (free.RequestId.ToHex().ToLowerInvariant() == contract.PurchaseId.ToLowerInvariant())
                    {
                        if (free.SlotIndex == slotIndex)
                        {
                            Log("Found the correct slotFree event");
                            return;
                        }
                    }
                }

                GetContracts().WaitUntilNextPeriod();
            }
            Assert.Fail($"{nameof(WaitForSlotFreedEvent)} for contract {contract.PurchaseId} and slotIndex {slotIndex} failed after {Time.FormatDuration(timeout)}");
        }

        private void WaitForNewSlotFilledEvent(IStoragePurchaseContract contract, ulong slotIndex)
        {
            Log(nameof(WaitForNewSlotFilledEvent));
            var start = DateTime.UtcNow;
            var timeout = contract.Purchase.Expiry;

            while (DateTime.UtcNow < start + timeout)
            {
                var newTimeRange = new TimeRange(start, DateTime.UtcNow); // We only want to see new fill events.
                var events = GetContracts().GetEvents(newTimeRange);
                var slotFillEvents = events.GetSlotFilledEvents();

                var matches = slotFillEvents.Where(f =>
                {
                    return
                        f.RequestId.ToHex().ToLowerInvariant() == contract.PurchaseId.ToLowerInvariant() &&
                        f.SlotIndex == slotIndex;
                }).ToArray();

                if (matches.Length > 1)
                {
                    var msg = string.Join(",", matches.Select(f => f.ToString()));
                    Assert.Fail($"Somehow, the slot got filled multiple times: {msg}");
                }
                if (matches.Length == 1)
                {
                    Log($"Found the correct new slotFilled event: {matches[0].ToString()}");
                }

                Thread.Sleep(TimeSpan.FromSeconds(15));
            }
            Assert.Fail($"{nameof(WaitForSlotFreedEvent)} for contract {contract.PurchaseId} and slotIndex {slotIndex} failed after {Time.FormatDuration(timeout)}");
        }

        private SlotFill GetSlotFillByOldestHost(List<ICodexNode> hosts)
        {
            var fills = GetOnChainSlotFills(hosts);
            var copy = hosts.ToArray();
            foreach (var host in copy)
            {
                var fill = GetFillByHost(host, fills);
                if (fill == null)
                {
                    // This host didn't fill anything.
                    // Move this one to the back of the list.
                    hosts.Remove(host);
                    hosts.Add(host);
                }
                else
                {
                    return fill;
                }
            }
            throw new Exception("None of the hosts seem to have filled a slot.");
        }

        private SlotFill? GetFillByHost(ICodexNode host, SlotFill[] fills)
        {
            // If these is more than 1 fill by this host, the test is misconfigured.
            // The availability size of the host should guarantee it can fill 1 slot maximum.
            return fills.SingleOrDefault(f => f.Host.EthAddress == host.EthAddress);
        }

        private IStoragePurchaseContract CreateStorageRequest(ICodexNode client)
        {
            var cid = client.UploadFile(GenerateTestFile(Filesize));
            var config = GetContracts().Deployment.Config;
            return client.Marketplace.RequestStorage(new StoragePurchaseRequest(cid)
            {
                Duration = HostAvailabilityMaxDuration / 2,
                Expiry = TimeSpan.FromMinutes(10.0),
                MinRequiredNumberOfNodes = Slots,
                NodeFailureTolerance = Tolerance,
                PricePerBytePerSecond = 10.TstWei(),
                ProofProbability = 20,
                CollateralPerByte = 1.TstWei()
            });
        }
    }
}