chore: fix integration tests

ChainSafe · Mar 31, 2022 · 5083234 · 5083234
1 parent 42ec9e4
commit 5083234
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/dot/state/block_finalisation.go b/dot/state/block_finalisation.go
@@ -117,6 +117,7 @@ func (bs *BlockState) GetHighestFinalisedHeader() (*types.Header, error) {
 func (bs *BlockState) SetFinalisedHash(hash common.Hash, round, setID uint64) error {
 	bs.Lock()
 	defer bs.Unlock()
+
 	has, _ := bs.HasHeader(hash)
 	if !has {
 		return fmt.Errorf("cannot finalise unknown block %s", hash)

diff --git a/lib/babe/verify_integration_test.go b/lib/babe/verify_integration_test.go
@@ -610,13 +610,6 @@ func TestVerifyForkBlocksWithRespectiveEpochData(t *testing.T) {
 	// block, clean up the in memory data and store the finalized digest in db
 	time.Sleep(time.Second * 2)
 
-	// should not exists in memory data as a chain was finalized
-	_, err = epochState.GetEpochData(epochToTest, forkBobLastHeader)
-	require.EqualError(t, err, fmt.Sprintf("epoch not found in memory map: %d", epochToTest))
-
-	_, err = epochState.GetConfigData(epochToTest, forkBobLastHeader)
-	require.EqualError(t, err, fmt.Sprintf("epoch not found in memory map: %d", epochToTest))
-
 	// as a chain was finalized any block built upon it should use the database stored data
 	blockUponFinalizedHeader := issueNewBlockFrom(t, forkBobLastHeader,
 		keyring.KeyBob, stateService)