GH-2173 Fix for test

AntelopeIO · Feb 1, 2024 · 4cc5eed · 4cc5eed
1 parent 8b9a0f5
commit 4cc5eed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/trx_finality_status_forked_test.py b/tests/trx_finality_status_forked_test.py
@@ -130,6 +130,8 @@
 
     assert not nonProdNode.verifyAlive(), "Bridge node should have been killed if test was functioning correctly."
 
+    assert prodC.waitForNextBlock(), "Prod node C should continue to advance, even after bridge node is killed"
+
     def getState(status):
         assert status is not None, "ERROR: getTransactionStatus failed to return any status"
         assert "state" in status, \
@@ -149,8 +151,8 @@ def getState(status):
     forkedOutState = "FORKED_OUT"
     unknownState = "UNKNOWN"
 
-    assert state == localState, \
-        f"ERROR: getTransactionStatus didn't return \"{localState}\" state.\n\nstatus: {json.dumps(retStatus, indent=1)}"
+    assert state == localState or state == inBlockState, \
+        f"ERROR: getTransactionStatus didn't return \"{localState}\" or \"{inBlockState}\" state.\n\nstatus: {json.dumps(retStatus, indent=1)}"
 
     assert prodC.waitForNextBlock(), "Production node C should continue to advance, even after bridge node is killed"