update tests

Author: zhangchiqing

module/state_synchronization/indexer/indexer_core_test.go

@@ -144,6 +144,7 @@ func (i *indexCoreTest) setStoreEvents(f func(*testing.T, flow.Identifier, []flo
 	return i
 }
 
+
 func (i *indexCoreTest) setStoreTransactionResults(f func(*testing.T, flow.Identifier, []flow.LightTransactionResult) error) *indexCoreTest {
 	i.results.
 		On("BatchStore", mock.Anything, mock.Anything, mock.AnythingOfType("flow.Identifier"), mock.AnythingOfType("[]flow.LightTransactionResult")).
@@ -257,8 +258,18 @@ func TestExecutionState_IndexBlockData(t *testing.T) {
 	t.Run("Index AllTheThings", func(t *testing.T) {
 		test := newIndexCoreTest(t, g, blocks, tf.ExecutionDataEntity()).initIndexer()
 
-		test.events.On("BatchStore", mock.Anything, []flow.EventsList{tf.ExpectedEvents}, mock.Anything).Return(nil)
-		test.results.On("BatchStore", mock.Anything, mock.Anything, blockID, tf.ExpectedResults).Return(nil)
+		test.events.On("BatchStore", mock.Anything, []flow.EventsList{tf.ExpectedEvents}, mock.Anything).
+			Return(func(blockID flow.Identifier, events []flow.EventsList, batch storage.ReaderBatchWriter) error {
+				require.NotNil(t, batch)
+				// Events BatchStore doesn't require specific locks, but we validate the batch is provided
+				return nil
+			})
+		test.results.On("BatchStore", mock.Anything, mock.Anything, blockID, tf.ExpectedResults).
+			Return(func(lctx lockctx.Proof, batch storage.ReaderBatchWriter, blockID flow.Identifier, results []flow.LightTransactionResult) error {
+				require.True(t, lctx.HoldsLock(storage.LockInsertLightTransactionResult))
+				require.NotNil(t, batch)
+				return nil
+			})
 		test.registers.
 			On("Store", mock.Anything, tf.Block.Height).
 			Run(func(args mock.Arguments) {
@@ -271,7 +282,12 @@ func TestExecutionState_IndexBlockData(t *testing.T) {
 			test.collections.On("StoreAndIndexByTransaction", mock.Anything, collection).Return(&flow.LightCollection{}, nil)
 		}
 		for txID, scheduledTxID := range tf.ExpectedScheduledTransactions {
-			test.scheduledTransactions.On("BatchIndex", mock.Anything, blockID, txID, scheduledTxID, mock.Anything).Return(nil)
+			test.scheduledTransactions.On("BatchIndex", mock.Anything, blockID, txID, scheduledTxID, mock.Anything).
+				Return(func(lctx lockctx.Proof, blockID flow.Identifier, txID flow.Identifier, scheduledTxID uint64, batch storage.ReaderBatchWriter) error {
+					require.True(t, lctx.HoldsLock(storage.LockIndexScheduledTransaction))
+					require.NotNil(t, batch)
+					return nil
+				})
 		}
 
 		err := test.indexer.IndexBlockData(tf.ExecutionDataEntity())