Add a default False to the user model email_confirmed field to avoid violation errors if the value is not explicitly set

This helps avoid issues where model classes are not evaluated. If the models
are not properly imported then Sqlalchemy can explode during database initialization
since the mapped columns have not yet been registered.

Later, having the models all imported into __init__.py will help with setting
up a test db since it will ensure that the registrations for all models is happening
as the classes are evaluated during the imports in the package init file.

pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+testpaths = src
+python_files = *_test.py
+addopts = -ra -q

setup.py

@@ -6,5 +6,8 @@ setup(
     packages=find_packages(),
     include_package_data=True,
     package_data={'src': ['templates/**/*.html', 'templates/**/*.txt']},
+    exclude_package_data={
+        '': ['*_test.py'],
+    },
     scripts=['./src/main.py'],
 )

shell.nix

@@ -20,6 +20,7 @@ mkShell {
     python313Packages.ruff
     python313Packages.python-lsp-server
     python313Packages.jedi-language-server
+    python313Packages.pytest
     ty
   ];
 }

src/infra/__init__.py

@@ -0,0 +1,27 @@
+from .db import Base, CRUD, CRUDRepo, UpdateMissingEntryError, get_database
+from .users import UserModel, UserRepoImpl
+from .publication import (
+    PublicationEntryModel,
+    PublicationSequenceModel,
+    PublicationOrderModel,
+    PublicationModel,
+    PublicationRepoImpl,
+)
+from .subscription import SubscriptionModel, SubscriptionRepoImpl
+
+__all__ = [
+    'Base',
+    'CRUD',
+    'CRUDRepo',
+    'UpdateMissingEntryError',
+    'get_database',
+    'UserModel',
+    'UserRepoImpl',
+    'PublicationEntryModel',
+    'PublicationSequenceModel',
+    'PublicationOrderModel',
+    'PublicationModel',
+    'PublicationRepoImpl',
+    'SubscriptionModel',
+    'SubscriptionRepoImpl',
+]

src/infra/db_test.py

@@ -0,0 +1,8 @@
+from src.infra import Base
+from sqlalchemy import Engine, create_engine
+
+
+def mock_db() -> Engine:
+    engine = create_engine('sqlite:///:memory:', echo=True)
+    Base.metadata.create_all(engine)
+    return engine

src/infra/users.py

@@ -18,7 +18,7 @@ class UserModel(Base):
     __tablename__ = 'user'
     id: Mapped[int] = mapped_column(primary_key=True)
     email: Mapped[str] = mapped_column(String(254), unique=True)
-    email_confirmed: Mapped[bool] = mapped_column()
+    email_confirmed: Mapped[bool] = mapped_column(default=False)
     password_hash: Mapped[str] = mapped_column(VARCHAR(255))
 
     subscriptions: Mapped[list['SubscriptionModel']] = relationship(back_populates='user')

src/infra/users_test.py

@@ -0,0 +1,31 @@
+from src.utils.secret import SecretBox
+from src.services.users.data import UserDTO, User
+from src.services.users.repo import UserRepo
+from src.infra.users import UserRepoImpl
+from src.infra.db_test import mock_db
+import pytest
+
+DB = mock_db()
+
+
+@pytest.fixture
+def db():
+    return DB
+
+
+@pytest.fixture
+def repo(db):
+    return UserRepoImpl(db)
+
+
+@pytest.fixture
+def users(repo: UserRepo):
+    return [
+        repo.create(UserDTO('example@example.com', SecretBox('hunter1')), None),
+        repo.create(UserDTO('example2@example.com', SecretBox('hunter2')), None),
+    ]
+
+
+def test_getting_a_user_by_email(users: list[User], repo: UserRepo):
+    assert repo.get_user_by_email('example@example.com') == users[0]
+    assert repo.get_user_by_email('fred@example.com') is None

src/main.py

@@ -6,8 +6,7 @@ from src.services.email import EmailService, get_email_service
 from src.services.auth import AuthService
 from src.services.users.service import UserService
 from dataclasses import dataclass
-from src.infra.db import get_database
+from src.infra import get_database, UserRepoImpl
-from src.infra.users import UserRepoImpl
 from waitress import serve
 from flask import Flask
 import argparse

src/utils/secret.py

@@ -7,6 +7,12 @@ class SecretBox[T]:
     def expose_secret(self) -> T:
         return self._item
 
+    def __eq__(self, other):
+        if not isinstance(other, SecretBox):
+            return False
+
+        return self._item == other._item
+
     def __str__(self):
         return '<SECRET>'