mirror of
https://github.com/wowlikon/LiB.git
synced 2026-02-04 04:31:09 +00:00
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
from typing import Any, List
|
|
|
|
from tests.mocks.mock_storage import mock_storage
|
|
|
|
|
|
class MockSession:
|
|
"""Mock SQLModel Session that works with MockStorage"""
|
|
|
|
def __init__(self):
|
|
self.storage = mock_storage
|
|
|
|
def add(self, obj: Any): ...
|
|
|
|
def commit(self): ...
|
|
|
|
def refresh(self, obj: Any): ...
|
|
|
|
def get(self, model_class, pk: int):
|
|
if hasattr(model_class, "__name__"):
|
|
model_name = model_class.__name__.lower()
|
|
else:
|
|
model_name = str(model_class).lower()
|
|
|
|
if "book" in model_name:
|
|
return self.storage.get_book(pk)
|
|
elif "author" in model_name:
|
|
return self.storage.get_author(pk)
|
|
elif "genre" in model_name:
|
|
return self.storage.get_genre(pk)
|
|
return None
|
|
|
|
def delete(self, obj: Any): ...
|
|
|
|
def exec(self, statement):
|
|
return MockResult([])
|
|
|
|
|
|
class MockResult:
|
|
"""Mock result for query operations"""
|
|
|
|
def __init__(self, data: List):
|
|
self.data = data
|
|
|
|
def all(self):
|
|
return self.data
|
|
|
|
def first(self):
|
|
return self.data[0] if self.data else None
|
|
|
|
|
|
def mock_get_session():
|
|
"""Mock session dependency"""
|
|
return MockSession()
|