check counter and epoch metadata after second recover transaction

onflow · Jul 29, 2024 · 8b5087d · 8b5087d
1 parent 614f489
commit 8b5087d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/go/test/flow_epoch_test.go b/lib/go/test/flow_epoch_test.go
@@ -1649,8 +1649,7 @@ func TestEpochRecover(t *testing.T) {
 			dkgKeys:               dkgPubKeys,
 		}
 		verifyEpochMetadata(t, b, env, expectedMetadata)
-		currEpochCounter := getCurrentEpochCounter(t, b, env)
-		assertEqual(t, currEpochCounter, cadence.NewUInt64(startEpochCounter+1))
+		assertEqual(t, getCurrentEpochCounter(t, b, env), cadence.NewUInt64(startEpochCounter+1))
 
 		expectedRecoverEvent := EpochRecover{
 			counter:                 startEpochCounter + 1,
@@ -1698,6 +1697,10 @@ func TestEpochRecover(t *testing.T) {
 		expectedRecoverEvent.dkgPhase1FinalView = newStartView + numStakingViews + numDKGViews - 1
 		expectedRecoverEvent.dkgPhase2FinalView = newStartView + numStakingViews + (2 * numDKGViews) - 1
 		expectedRecoverEvent.dkgPhase3FinalView = newStartView + numStakingViews + (3 * numDKGViews) - 1
+		expectedMetadata.startView = newStartView
+		expectedMetadata.endView = newEndView
 		verifyEpochRecover(t, adapter, idTableAddress, expectedRecoverEvent)
+		verifyEpochMetadata(t, b, env, expectedMetadata)
+		assertEqual(t, getCurrentEpochCounter(t, b, env), cadence.NewUInt64(startEpochCounter+1))
 	})
 }