cs-codex-dist-tests/Tests/PeerDiscoveryTests/PeerTestHelpers.cs

using DistTestCore.Codex;
using DistTestCore;
using NUnit.Framework;
using Utils;
using Logging;

namespace Tests.PeerDiscoveryTests
{
    public static class PeerTestHelpers
    {
        public static void AssertFullyConnected(IEnumerable<IOnlineCodexNode> nodes, BaseLog? log = null)
        {
            AssertFullyConnected(log, nodes.ToArray());
        }

        public static void AssertFullyConnected(BaseLog? log = null, params IOnlineCodexNode[] nodes)
        {
            Time.Retry(() =>
            {
                var entries = nodes.Select(n => new Entry(n)).ToArray();

                var failureMessags = new List<string>();
                for (var x = 0; x < entries.Length; x++)
                {
                    for (var y = x + 1; y < entries.Length; y++)
                    {
                        AssertKnowEachother(failureMessags, entries[x], entries[y], log);
                    }
                }

                CollectionAssert.IsEmpty(failureMessags);
            });
        }

        private static void AssertKnowEachother(List<string> failureMessags, Entry a, Entry b, BaseLog? log)
        {
            AssertKnows(failureMessags, a, b, log);
            AssertKnows(failureMessags, b, a, log);
        }

        private static void AssertKnows(List<string> failureMessags, Entry a, Entry b, BaseLog? log)
        {
            var peerId = b.Response.id;

            try
            {
                var response = a.Node.GetDebugPeer(peerId);
                if (string.IsNullOrEmpty(response.peerId) || !response.addresses.Any())
                {
                    failureMessags.Add($"{a.Response.id} did not know peer {peerId}");
                }
                else if (log != null)
                {
                    log.Log($"{a.Response.id} knows {peerId}.");
                }
            }
            catch (Exception e)
            {
                failureMessags.Add($"{a.Response.id} was unable to get 'debug/peer/{peerId}'. {e}");
            }

            ////var enginePeers = string.Join(",", a.enginePeers.Select(p => p.peerId));
            ////var switchPeers = string.Join(",", a.switchPeers.Select(p => p.peerId));
            //var tableNodes = string.Join(",", a.table.nodes.Select(n => n.nodeId));

            //var success = a.table.nodes.Any(n => n.nodeId == b.table.localNode.nodeId);

            //if (log != null)
            //{
            //    var msg = success ? "PASS" : "FAIL";
            //    log.Log($"{msg} {a.table.localNode.nodeId} is looking for {b.table.localNode.nodeId} in table-nodes [{tableNodes}]");
            //}

            ////Assert.That(a.enginePeers.Any(p => p.peerId == b.id), $"{a.id} was looking for '{b.id}' in engine-peers [{enginePeers}] but it was not found.");
            ////Assert.That(a.switchPeers.Any(p => p.peerId == b.id), $"{a.id} was looking for '{b.id}' in switch-peers [{switchPeers}] but it was not found.");
            //if (!success)
            //{
            //    failureMessags.Add($"{a.table.localNode.nodeId} was looking for '{b.table.localNode.nodeId}' in table-nodes [{tableNodes}] but it was not found.");
            //}
        }

        public class Entry
        {
            public Entry(IOnlineCodexNode node)
            {
                Node = node;
                Response = node.GetDebugInfo();
            }

            public IOnlineCodexNode Node { get ; }
            public CodexDebugResponse Response { get; }
        }
    }
}