ensure update_one is called with correct args

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

tests/unittests/execution_store_tests.py

@@ -236,7 +236,10 @@ def test_operator_execute_with_store(self):
         result = self.operator.execute(self.ctx)
 
         # Verify that the store interactions were made correctly
-        self.mock_collection.update_one.assert_called_once()  # Checking that set operation inserts data
+        self.mock_collection.update_one.assert_called_once_with(
+            {"store_name": "test_store", "key": "my_key"},
+            {"$set": {"value": {"foo": "bar"}}}
+        )
         self.mock_collection.find_one.assert_called_once_with(
             {"store_name": "test_store", "key": "my_key"}
         )  # Checking that get operation retrieves data