Added Base testing

app/core/db.py

@@ -26,7 +26,7 @@ def __tablename__(cls):
     id: Mapped[int] = mapped_column(primary_key=True)
 
     def __repr__(self) -> str:
-        return (f'\nid: {self.id}\n')
+        return f'\nid: {self.id}\n'
 
 
 engine = create_async_engine(settings.database_url)

tests/unit_tests/test_models.py

@@ -1,17 +1,28 @@
 import pytest
 
-from tests.conftest import Post
+from tests.conftest import Base, Post
 
+BASE_FIELDS = ('id',)
 POST_MODEL_FIELDS = ('id', 'title', 'content', 'created', 'updated', 'likes', 'dislikes', 'author_id', 'author')
 POST_SAVE_DATA = {'title': 'Another New post title.', 'content': 'POST New post content.', 'author_id': 1}
 
+class BaseTest(Base):
+    pass
 
-def test_model_fields() -> None:
-    for field in POST_MODEL_FIELDS:
-        assert hasattr(Post, field)
 
+@pytest.mark.parametrize('class_, class_fields', (
+    (Base, BASE_FIELDS),
+    (Post, POST_MODEL_FIELDS),
+))
+def test_model_fields(class_, class_fields) -> None:
+    for field in class_fields:
+        assert hasattr(class_, field)
 
-def test_model_repr() -> None:
-    representation = str(Post(**POST_SAVE_DATA))
-    for attr_name in POST_MODEL_FIELDS[1:-1]:
-        assert representation.find(attr_name) != -1
+
+@pytest.mark.parametrize('instance, instance_fields', (
+    (BaseTest(), BASE_FIELDS),
+    (Post(**POST_SAVE_DATA), POST_MODEL_FIELDS[1:-1]),
+))
+def test_model_repr(instance, instance_fields) -> None:
+    for attr_name in instance_fields:
+        assert instance.__repr__().find(attr_name) != -1