Fix call counts with motor test

AikidoSec · Nov 18, 2024 · 9133c1c · 9133c1c
1 parent 7ee0ff7
commit 9133c1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aikido_zen/sinks/tests/motor_test.py b/aikido_zen/sinks/tests/motor_test.py
@@ -334,7 +334,7 @@ async def test_aggregate_raw_batches_key(db):
         async for doc in dogs.aggregate_raw_batches(pipeline=pipeline):
             pass
 
-        called_with = mock_run_vulnerability_scan.call_args[1]
+        called_with = mock_run_vulnerability_scan.call_args_list[0][1]
         assert called_with["args"][0] == pipeline
         assert (
             called_with["op"] == "pymongo.collection.Collection.aggregate_raw_batches"