From 30545e03640714d415116097568674db27fe1abd Mon Sep 17 00:00:00 2001
From: Tytus Bierwiaczonek <tbierwiaczon@vmware.com>
Date: Mon, 11 Jul 2022 05:26:28 -0700
Subject: [PATCH] State transfer check addition

---
 tests/apollo/test_skvbc_restart_recovery.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/apollo/test_skvbc_restart_recovery.py b/tests/apollo/test_skvbc_restart_recovery.py
index 832e1a8dc0..937cdcff8d 100644
--- a/tests/apollo/test_skvbc_restart_recovery.py
+++ b/tests/apollo/test_skvbc_restart_recovery.py
@@ -174,6 +174,8 @@ async def _restarting_replica_during_system_is_in_view_change(self, bft_network,
 
             old_view = view
 
+            await self._await_replicas_in_state_transfer(log, bft_network, skvbc, current_primary)
+
             # Wait for quorum of replicas to move to a higher view
             with trio.fail_after(seconds=20 + timeouts):
                 while view == old_view: