From 35e9dd58943d87bdf8fa96df0df1df034f5a3d3d Mon Sep 17 00:00:00 2001
From: Ben <thatbenbierens@gmail.com>
Date: Mon, 2 Dec 2024 15:37:50 +0100
Subject: [PATCH] Reduces swarm size for smallswarm test

---
 ProjectPlugins/CodexPlugin/StoragePurchaseContract.cs | 1 +
 Tests/CodexReleaseTests/DataTests/SwarmTest.cs        | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/ProjectPlugins/CodexPlugin/StoragePurchaseContract.cs b/ProjectPlugins/CodexPlugin/StoragePurchaseContract.cs
index e4325cab..24566980 100644
--- a/ProjectPlugins/CodexPlugin/StoragePurchaseContract.cs
+++ b/ProjectPlugins/CodexPlugin/StoragePurchaseContract.cs
@@ -77,6 +77,7 @@ public void WaitForStorageContractFinished(ICodexContracts contracts)
             LogFinishedDuration();
             AssertDuration(SubmittedToFinished, timeout, nameof(SubmittedToFinished));
 
+            contracts.WaitUntilNextPeriod();
             contracts.WaitUntilNextPeriod();
 
             var blocks = 3;
diff --git a/Tests/CodexReleaseTests/DataTests/SwarmTest.cs b/Tests/CodexReleaseTests/DataTests/SwarmTest.cs
index c08d2011..931590f0 100644
--- a/Tests/CodexReleaseTests/DataTests/SwarmTest.cs
+++ b/Tests/CodexReleaseTests/DataTests/SwarmTest.cs
@@ -14,7 +14,7 @@ namespace CodexReleaseTests.DataTests
     [TestFixture]
     public class SwarmTests : AutoBootstrapDistTest
     {
-        private const int NumberOfNodes = 5;
+        private const int NumberOfNodes = 2;
         private const int FileSizeMb = 2;
 
         [Test]