Skip to content

Commit cacffce

Browse files
committed
test: fix RuntimeWarning in async cluster mode tests
Fix pipeline json mock configuration in AsyncMockRedis and AsyncMockRedisCluster to properly handle method chaining. The json().set() and json().get() methods now return the pipeline object for fluent interface compatibility. Resolves "coroutine 'AsyncMockMixin._execute_mock_call' was never awaited" warning in test_async_checkpoint_saver_aput_ttl_behavior.
1 parent 78222c9 commit cacffce

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/test_async_cluster_mode.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,10 @@ def pipeline(self, transaction=True):
4444
mock_pipeline.delete = MagicMock(return_value=1)
4545
mock_pipeline.execute = AsyncMock(return_value=[])
4646

47-
# Mock json().get() behavior within pipeline
47+
# Mock json().set() behavior within pipeline - set must return pipeline for method chaining
4848
mock_json_pipeline = AsyncMock()
49-
mock_json_pipeline.get = MagicMock()
49+
mock_json_pipeline.get = MagicMock(return_value=mock_pipeline)
50+
mock_json_pipeline.set = MagicMock(return_value=mock_pipeline)
5051
mock_pipeline.json = MagicMock(return_value=mock_json_pipeline)
5152
return mock_pipeline
5253

@@ -124,8 +125,10 @@ def pipeline(self, transaction=True):
124125
mock_pipeline.expire = MagicMock(return_value=True)
125126
mock_pipeline.delete = MagicMock(return_value=1)
126127

127-
mock_json_pipeline = MagicMock()
128-
mock_json_pipeline.get = MagicMock()
128+
# Mock json().set() behavior within pipeline - set must return pipeline for method chaining
129+
mock_json_pipeline = AsyncMock()
130+
mock_json_pipeline.get = MagicMock(return_value=mock_pipeline)
131+
mock_json_pipeline.set = MagicMock(return_value=mock_pipeline)
129132
mock_pipeline.json = MagicMock(return_value=mock_json_pipeline)
130133
return mock_pipeline
131134

0 commit comments

Comments
 (0)