Homework 1

pyproject.toml

@@ -66,6 +66,9 @@ dennis = "^1.1"
 dump-env = "^1.3"
 ipython = "^8.15"
 import-linter = "^1.11"
+faker = "^19.6.1"
+factory-boy = "^3.3.0"
+pytest-factoryboy = "^2.5.1"
 
 [tool.poetry.group.docs]
 optional = true

tests/conftest.py

@@ -8,7 +8,7 @@
 
 pytest_plugins = [
     # Should be the first custom one:
-    'plugins.django_settings',
-
+    "plugins.django_settings",
+    "plugins.identity.user"
     # TODO: add your own plugins here!
 ]

tests/plugins/identity/user.py

@@ -0,0 +1,89 @@
+import factory
+import pytest
+from django.contrib.auth.hashers import make_password
+from factory.base import FactoryMetaClass
+from pytest_factoryboy import register
+
+from server.apps.identity.models import User
+
+locale = "RU_ru"
+
+
+@register
+class UserDataFactory(factory.DictFactory):
+    email = factory.Faker("email")
+    first_name = factory.Faker("first_name", locale=locale)
+    last_name = factory.Faker("last_name", locale=locale)
+    date_of_birth = factory.Faker("date_object")
+    address = factory.Faker("address", locale=locale)
+    job_title = factory.Faker("job", locale=locale)
+    phone = factory.Faker("phone_number")
+    password = make_password("pass")
+
+
+@register
+class UserUpdateDataFactory(factory.DictFactory):
+    first_name = factory.Faker("first_name", locale=locale)
+    last_name = factory.Faker("last_name", locale=locale)
+    date_of_birth = factory.Faker("date_object")
+    address = factory.Faker("address", locale=locale)
+    job_title = factory.Faker("job", locale=locale)
+    phone = factory.Faker("phone_number")
+
+
+@pytest.fixture
+def registration_data(user_data_factory: FactoryMetaClass) -> dict:
+    data = user_data_factory()
+    data["password1"] = data["password"]
+    data["password2"] = data["password"]
+
+    return data
+
+
+@pytest.fixture()
+def expected_user_data() -> dict:
+    def factory(data: dict) -> dict:
+        return {
+            key_name: value_part
+            for key_name, value_part in data.items()
+            if not key_name.startswith("password")
+        }
+
+    return factory
+
+
+@pytest.fixture
+def user(user_data_factory: FactoryMetaClass) -> User:
+    return User.objects.create_user(**user_data_factory())
+
+
+@pytest.fixture(scope="session")
+def assert_correct_user() -> None:
+    def factory(email: str, expected: dict) -> None:
+        user = User.objects.get(email=email)
+        # Special fields:
+        assert user.id
+        assert user.is_active
+        assert not user.is_superuser
+        assert not user.is_staff
+        # All other fields:
+        for field_name, data_value in expected.items():
+            assert getattr(user, field_name) == data_value
+
+    return factory
+
+
+@pytest.fixture(scope="session")
+def assert_correct_superuser() -> None:
+    def factory(email: str, expected: dict) -> None:
+        user = User.objects.get(email=email)
+        # Special fields:
+        assert user.id
+        assert user.is_active
+        assert user.is_superuser
+        assert user.is_staff
+        # All other fields:
+        for field_name, data_value in expected.items():
+            assert getattr(user, field_name) == data_value
+
+    return factory

tests/test_apps/conftest.py

@@ -0,0 +1,10 @@
+import pytest
+from django.test import Client
+
+from server.apps.identity.models import User
+
+
+@pytest.fixture()
+def authenticated_client(client: Client, user: User) -> Client:
+    client.force_login(user)
+    return client

tests/test_apps/test_identity/test_models.py

@@ -0,0 +1,39 @@
+import pytest
+from django.db.utils import IntegrityError
+from factory.base import FactoryMetaClass
+
+from server.apps.identity.models import User
+
+pytestmark = pytest.mark.django_db
+
+
+def test_create_user(
+    user_data_factory: FactoryMetaClass, expected_user_data, assert_correct_user
+) -> None:
+    user_data = user_data_factory()
+
+    User.objects.create_user(**user_data)
+
+    assert_correct_user(user_data["email"], expected_user_data(user_data))
+
+
+def test_create_superuser(
+    user_data_factory: FactoryMetaClass, expected_user_data, assert_correct_superuser
+) -> None:
+    user_data = user_data_factory()
+
+    User.objects.create_superuser(**user_data)
+
+    assert_correct_superuser(user_data["email"], expected_user_data(user_data))
+
+
+def test_create_user_missing_email(
+    user_data_factory: FactoryMetaClass,
+) -> None:
+    user_data = user_data_factory(email="")
+
+    with pytest.raises(ValueError) as excinfo:
+        User.objects.create_user(**user_data)
+
+    assert "Users must have an email address" in str(excinfo.value)
+    assert not User.objects.filter(email=user_data["email"]).exists()