fixed system tests

Author: gkevinzheng

tests/system/data/test_system_async.py

@@ -507,9 +507,8 @@ async def test_mutations_batcher_completed_callback(
         )
         bulk_mutation = RowMutationEntry(row_key, [mutation])
         flush_interval = 0.1
-        async with target.mutations_batcher(
-            flush_interval=flush_interval, _batch_completed_callback=callback
-        ) as batcher:
+        async with target.mutations_batcher(flush_interval=flush_interval) as batcher:
+            batcher._user_batch_completed_callback = callback
             await batcher.append(bulk_mutation)
             await CrossSync.yield_to_event_loop()
             assert len(batcher._staged_entries) == 1

tests/system/data/test_system_autogen.py

@@ -403,9 +403,8 @@ def test_mutations_batcher_completed_callback(self, client, target, temp_rows):
         )
         bulk_mutation = RowMutationEntry(row_key, [mutation])
         flush_interval = 0.1
-        with target.mutations_batcher(
-            flush_interval=flush_interval, _batch_completed_callback=callback
-        ) as batcher:
+        with target.mutations_batcher(flush_interval=flush_interval) as batcher:
+            batcher._user_batch_completed_callback = callback
             batcher.append(bulk_mutation)
             CrossSync._Sync_Impl.yield_to_event_loop()
             assert len(batcher._staged_entries) == 1