From 3822d5e2ebba6640309c26d2a925ffa10a853091 Mon Sep 17 00:00:00 2001
From: Ben <thatbenbierens@gmail.com>
Date: Wed, 4 Jun 2025 12:49:10 +0200
Subject: [PATCH] wip

---
 .../Utils/MarketplaceAutoBootstrapDistTest.cs | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/Tests/CodexReleaseTests/Utils/MarketplaceAutoBootstrapDistTest.cs b/Tests/CodexReleaseTests/Utils/MarketplaceAutoBootstrapDistTest.cs
index 72442098..26c47600 100644
--- a/Tests/CodexReleaseTests/Utils/MarketplaceAutoBootstrapDistTest.cs
+++ b/Tests/CodexReleaseTests/Utils/MarketplaceAutoBootstrapDistTest.cs
@@ -103,6 +103,30 @@ namespace CodexReleaseTests.Utils
             return host;
         }
 
+        public void AssertHostAvailabilitiesAreEmpty(IEnumerable<ICodexNode> hosts)
+        {
+            var retry = GetAvailabilitySpaceAssertRetry();
+            retry.Run(() =>
+            {
+                foreach (var host in hosts)
+                {
+                    AssertHostAvailabilitiesAreEmpty(host);
+                }
+            });
+        }
+
+        private void AssertHostAvailabilitiesAreEmpty(ICodexNode host)
+        {
+            var availabilities = host.Marketplace.GetAvailabilities();
+            foreach (var a in availabilities)
+            {
+                if (a.FreeSpace.SizeInBytes != a.TotalSpace.SizeInBytes)
+                {
+                    throw new Exception(nameof(AssertHostAvailabilitiesAreEmpty) + $" free: {a.FreeSpace} total: {a.TotalSpace}");
+                }
+            }
+        }
+
         public void AssertTstBalance(ICodexNode node, TestToken expectedBalance, string message)
         {
             AssertTstBalance(node.EthAddress, expectedBalance, message);
@@ -160,6 +184,15 @@ namespace CodexReleaseTests.Utils
                 failFast: false);
         }
 
+        private Retry GetAvailabilitySpaceAssertRetry()
+        {
+            return new Retry("AssertAvailabilitySpace",
+                maxTimeout: HostBlockTTL * 3,
+                sleepAfterFail: TimeSpan.FromSeconds(10.0),
+                onFail: f => { },
+                failFast: false);
+        }
+
         private TestToken GetTstBalance(ICodexNode node)
         {
             return GetContracts().GetTestTokenBalance(node);