fixup! test: get tests to pass without setting NONCE_LIMITER_RESET_FR…

…EQUENCY
ArweaveTeam · Dec 24, 2024 · 5713fed · 5713fed
1 parent 0eb8ed8
commit 5713fed
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/apps/arweave/test/ar_test_node.erl b/apps/arweave/test/ar_test_node.erl
@@ -852,16 +852,22 @@ wait_until_height(Node, TargetHeight) ->
 	wait_until_height(Node, TargetHeight, true).
 
 wait_until_height(Node, TargetHeight, Strict) ->
-	BI = case Node of 
+	{BI, Height} = case Node of 
 		main ->
-			wait_until_height(TargetHeight);
+			{
+				wait_until_height(TargetHeight),
+				ar_node:get_height()
+			};
 		_ -> 
-			remote_call(Node, ?MODULE, wait_until_height, [TargetHeight],
-				?WAIT_UNTIL_BLOCK_HEIGHT_TIMEOUT + 500)
+			{
+				remote_call(Node, ?MODULE, wait_until_height, [TargetHeight],
+					?WAIT_UNTIL_BLOCK_HEIGHT_TIMEOUT + 500),
+				remote_call(Node, ar_node, get_height, [])
+			}
 	end,
 	case Strict of
 		true ->
-			?assertEqual(TargetHeight, remote_call(Node, ar_node, get_height, []));
+			?assertEqual(TargetHeight, Height);
 		false ->
 			ok
 	end,