feat: enhance mock database store with context manager support and improve cursor behavior
CI / lint-test-build (push) Failing after 52s
CI / lint-test-build (push) Failing after 52s
This commit is contained in:
@@ -20,8 +20,17 @@ def mock_settings():
|
||||
|
||||
@pytest.fixture
|
||||
def mock_store():
|
||||
"""Create a mock database store."""
|
||||
"""Create a mock database store with context manager support."""
|
||||
store = Mock(spec=DuckDBStore)
|
||||
cursor = Mock()
|
||||
cursor.fetchone.return_value = None
|
||||
cursor.fetchall.return_value = []
|
||||
cursor.execute.return_value = cursor
|
||||
# Set up context manager via mock_connect property
|
||||
store.connect.side_effect = None # disable side effect
|
||||
context_mock = Mock()
|
||||
context_mock.__enter__.return_value = cursor
|
||||
store.connect.return_value = context_mock
|
||||
return store
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user