re-structure
This commit is contained in:
@@ -9,7 +9,7 @@ from datetime import datetime, timezone
|
||||
from unittest.mock import MagicMock, patch
|
||||
from uuid import uuid4, UUID
|
||||
|
||||
from inference.data.repositories.base import BaseRepository
|
||||
from backend.data.repositories.base import BaseRepository
|
||||
|
||||
|
||||
class ConcreteRepository(BaseRepository[MagicMock]):
|
||||
@@ -31,7 +31,7 @@ class TestBaseRepository:
|
||||
|
||||
def test_session_yields_session(self, repo):
|
||||
"""Test _session yields a database session."""
|
||||
with patch("inference.data.repositories.base.get_session_context") as mock_ctx:
|
||||
with patch("backend.data.repositories.base.get_session_context") as mock_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
|
||||
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
|
||||
|
||||
Reference in New Issue
Block a user