From 339cf2b82407678dd40f8fa8303f0632361572e0 Mon Sep 17 00:00:00 2001
From: Ben <thatbenbierens@gmail.com>
Date: Thu, 21 Nov 2024 14:34:13 +0100
Subject: [PATCH] Peer table test

---
 .../NodeTests/PeerTableTests.cs               | 52 ++++++++++++++++++-
 1 file changed, 50 insertions(+), 2 deletions(-)

diff --git a/Tests/CodexReleaseTests/NodeTests/PeerTableTests.cs b/Tests/CodexReleaseTests/NodeTests/PeerTableTests.cs
index d1b4161..f61a421 100644
--- a/Tests/CodexReleaseTests/NodeTests/PeerTableTests.cs
+++ b/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.");
+        }
     }
 }