From 72e5a1995d7c5273bc8b0b3579d1fdac5c92fabe Mon Sep 17 00:00:00 2001
From: Ben <thatbenbierens@gmail.com>
Date: Tue, 2 Sep 2025 10:11:01 +0200
Subject: [PATCH] DHT test WIP. Retry update

---
 Framework/Utils/Retry.cs                     |  1 +
 Tests/CodexReleaseTests/DataTests/DhtTest.cs | 58 ++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 Tests/CodexReleaseTests/DataTests/DhtTest.cs

diff --git a/Framework/Utils/Retry.cs b/Framework/Utils/Retry.cs
index 75f9e06f..240110f9 100644
--- a/Framework/Utils/Retry.cs
+++ b/Framework/Utils/Retry.cs
@@ -98,6 +98,7 @@
             private void CheckMaximums()
             {
                 if (Duration() > maxTimeout) Fail();
+                if (tryNumber > 30) Fail();
 
                 // If we have a few very fast failures, retrying won't help us. There's probably something wrong with our operation.
                 // In this case, don't wait the full duration and fail quickly.
diff --git a/Tests/CodexReleaseTests/DataTests/DhtTest.cs b/Tests/CodexReleaseTests/DataTests/DhtTest.cs
new file mode 100644
index 00000000..8b47a63e
--- /dev/null
+++ b/Tests/CodexReleaseTests/DataTests/DhtTest.cs
@@ -0,0 +1,58 @@
+using CodexClient;
+using CodexTests;
+using Logging;
+using NUnit.Framework;
+using Utils;
+
+namespace CodexReleaseTests.DataTests
+{
+    [Ignore("work in progress")]
+    [TestFixture(10,  10000)]
+    [TestFixture(50, 1000)]
+    [TestFixture(100,  1000)]
+    public class DhtTest : AutoBootstrapDistTest
+    {
+        public DhtTest(int nodes, int files)
+        {
+            numNodes = nodes;
+            numFilesPerNode = files;
+        }
+
+        private readonly int numNodes;
+        private readonly int numFilesPerNode;
+        private readonly int numToFetch = 200;
+        private readonly ByteSize fileSize = 100.KB();
+        private readonly TimeSpan maxDownloadDuration = TimeSpan.FromSeconds(5.0);
+
+        [Test]
+        public void PressureTest()
+        {
+            var cids = new List<ContentId>();
+            var nodes = StartCodex(numNodes);
+
+            for (var i = 0; i < numFilesPerNode; i++)
+            {
+                foreach (var n in nodes)
+                {
+                    cids.Add(n.UploadFile(GenerateTestFile(fileSize)));
+                }
+            }
+
+            // We announce both manifest-cid and tree-cid for each file;
+            var estimate = numNodes * numFilesPerNode * 2;
+            Log($"Estimate of DHT records: {estimate}");
+
+            var node = StartCodex();
+            for (var i = 0; i < numToFetch; i++)
+            {
+                var timing = Stopwatch.Measure(GetTestLog(), nameof(PressureTest) + i, () =>
+                {
+                    node.DownloadContent(cids.PickOneRandom());
+                });
+
+                Log(Time.FormatDuration(timing));
+                Assert.That(timing, Is.LessThan(maxDownloadDuration));
+            }
+        }
+    }
+}