fix unit test

ChainSafe · Jun 17, 2021 · 4d4942a · 4d4942a
1 parent d1b23b1
commit 4d4942a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/babe/babe_test.go b/lib/babe/babe_test.go
@@ -206,7 +206,7 @@ func TestService_ProducesBlocks(t *testing.T) {
 	}()
 
 	time.Sleep(babeService.slotDuration * 2)
-	babeService.blockImportHandler.(*mocks.BlockImportHandler).AssertCalled(t, "HandleBlockImport", mock.AnythingOfType("*types.Block"), mock.AnythingOfType("*storage.TrieState"))
+	babeService.blockImportHandler.(*mocks.BlockImportHandler).AssertCalled(t, "HandleBlockProduced", mock.AnythingOfType("*types.Block"), mock.AnythingOfType("*storage.TrieState"))
 }
 
 func TestService_GetAuthorityIndex(t *testing.T) {