using CodexPlugin;
using NUnit.Framework;
using Utils;

namespace CodexTests.BasicTests
{
    [TestFixture]
    public class BlockExchangeTests : CodexDistTest
    {
        [Test]
        public void EmptyAfterExchange()
        {
            var bootstrap = AddCodex(s => s.WithName("bootstrap"));
            var node = AddCodex(s => s.WithName("node").WithBootstrapNode(bootstrap));

            AssertExchangeIsEmpty(bootstrap, node);

            var file = GenerateTestFile(1.MB());
            var cid = bootstrap.UploadFile(file);
            node.DownloadContent(cid);

            AssertExchangeIsEmpty(bootstrap, node);
        }

        [Test]
        public void EmptyAfterExchangeWithBystander()
        {
            var bootstrap = AddCodex(s => s.WithName("bootstrap"));
            var node = AddCodex(s => s.WithName("node").WithBootstrapNode(bootstrap));
            var bystander = AddCodex(s => s.WithName("bystander").WithBootstrapNode(bootstrap));

            AssertExchangeIsEmpty(bootstrap, node, bystander);

            var file = GenerateTestFile(1.MB());
            var cid = bootstrap.UploadFile(file);
            node.DownloadContent(cid);

            AssertExchangeIsEmpty(bootstrap, node, bystander);
        }

        private void AssertExchangeIsEmpty(params ICodexNode[] nodes)
        {
            foreach (var node in nodes)
            {
                Time.Retry(() => AssertBlockExchangeIsEmpty(node), nameof(AssertExchangeIsEmpty));
            }
        }

        private void AssertBlockExchangeIsEmpty(ICodexNode node)
        {
            var msg = $"BlockExchange for {node.GetName()}: ";
            var response = node.GetDebugBlockExchange();
            foreach (var peer in response.peers)
            {
                var activeWants = peer.wants.Where(w => !w.cancel).ToArray();
                Assert.That(activeWants.Length, Is.EqualTo(0), msg + "thinks a peer has active wants.");
            }
            Assert.That(response.taskQueue, Is.EqualTo(0), msg + "has tasks in queue.");
            Assert.That(response.pendingBlocks, Is.EqualTo(0), msg + "has pending blocks.");
        }
    }
}