Merge pull request #26 from PythonFloripa/feature/#12

Feature/#12

Author: dan5e3s6ares

app/services/database/models/__init__.py

@@ -1,4 +1,5 @@
 from app.services.database.models.communities import Community
 from app.services.database.models.libraries import Library
+from app.services.database.models.subscriptions import Subscription
 
-__all__ = ["Community", "Library"]
+__all__ = ["Community", "Library", "Subscription"]

app/services/database/models/subscriptions.py

@@ -0,0 +1,12 @@
+from typing import Optional
+
+from sqlmodel import SQLModel, Field
+
+
+class Subscription(SQLModel, table=True):
+    __tablename__ = 'subscriptions'
+
+    id: Optional[int] = Field(default=None, primary_key=True)
+    email: str
+    tags: str
+    community_id: Optional[int] = Field(default=None, foreign_key="communities.id")

tests/test_subscriptions.py

@@ -0,0 +1,36 @@
+import pytest
+import pytest_asyncio
+
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+from services.database.models import Community, Subscription
+
+
+@pytest_asyncio.fixture
+async def community(session: AsyncSession):
+    community = Community(username="admin", email="[email protected]", password="123")
+    session.add(community)
+    await session.commit()
+    await session.refresh(community)
+    return community
+
+
+@pytest.mark.asyncio
+async def test_insert_subscription(session: AsyncSession, community: Community):
+    subscription = Subscription(
+        email="[email protected]",
+        tags="teste,Python",
+        community_id=community.id,
+    )
+    session.add(subscription)
+    await session.commit()
+
+    statement = select(Subscription).where(Subscription.email == "[email protected]")
+    result = await session.exec(statement)
+    found = result.first()
+
+    assert found is not None
+    assert found.email == "[email protected]"
+    assert found.tags == "teste,Python"
+    assert found.community_id == community.id