Peer table test

Tests/CodexReleaseTests/NodeTests/PeerTableTests.cs

@@ -1,12 +1,60 @@
-using System;
+using CodexPlugin;
+using CodexTests;
+using NUnit.Framework;
+using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
 using System.Threading.Tasks;
+using Utils;
 
 namespace CodexReleaseTests.NodeTests
 {
-    internal class PeerTableTests
+    [TestFixture]
+    public class PeerTableTests : AutoBootstrapDistTest
     {
+        [Test]
+        public void PeerTableCompleteness()
+        {
+            var nodes = StartCodex(10);
+
+            var retry = new Retry(
+                description: nameof(PeerTableCompleteness),
+                maxTimeout: TimeSpan.FromMinutes(2),
+                sleepAfterFail: TimeSpan.FromSeconds(5),
+                onFail: f => { }
+            );
+
+            retry.Run(() => AssertAllNodesSeeEachOther(nodes));
+        }
+
+        private void AssertAllNodesSeeEachOther(ICodexNodeGroup nodes)
+        {
+            foreach (var a in nodes)
+            {
+                AssertHasSeenAllOtherNodes(a, nodes);
+            }
+        }
+
+        private void AssertHasSeenAllOtherNodes(ICodexNode node, ICodexNodeGroup nodes)
+        {
+            var localNode = node.GetDebugInfo().Table.LocalNode;
+
+            foreach (var other in nodes)
+            {
+                var info = other.GetDebugInfo();
+                if (info.Table.LocalNode.PeerId != localNode.PeerId)
+                {
+                    AssertContainsPeerId(info, localNode.PeerId);
+                }
+            }
+        }
+
+        private void AssertContainsPeerId(DebugInfo info, string peerId)
+        {
+            var entry = info.Table.Nodes.SingleOrDefault(n => n.PeerId == peerId);
+            if (entry == null) throw new Exception("Table entry not found.");
+            if (!entry.Seen) throw new Exception("Peer not seen.");
+        }
     }
 }