Fix create user

industry_game/utils/users/base.py

@@ -1,6 +1,6 @@
 from enum import StrEnum
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 from industry_game.utils.msgspec import CustomStruct
 
@@ -11,8 +11,10 @@ class UserType(StrEnum):
 
 
 class RegisterPlayerModel(BaseModel):
-    username: str
-    password: str
+    username: str = Field(min_length=8)
+    password: str = Field(min_length=8)
+    telegram: str = Field(min_length=2)
+    name: str = Field(min_length=5)
 
 
 class AuthUserModel(BaseModel):

industry_game/utils/users/models.py

@@ -29,11 +29,15 @@ class FullUser(CustomStruct, frozen=True):
     id: int
     type: UserType
     username: str
+    telegram: str
+    name: str
 
     @classmethod
     def from_model(self, obj: UserDb) -> "FullUser":
         return FullUser(
             id=obj.id,
             type=obj.type,
             username=obj.username,
+            telegram=obj.properties.get("telegram", ""),
+            name=obj.properties.get("name", ""),
         )

industry_game/utils/users/processor.py

@@ -28,6 +28,10 @@ async def register(self, player: RegisterPlayerModel) -> AuthToken:
         user = await self.player_storage.create(
             username=player.username,
             password_hash=self.passgen.hashpw(player.password),
+            properties={
+                "telegram": player.telegram,
+                "name": player.name,
+            },
         )
         token = self.authorization_provider.generate_token(user=user)
         return AuthToken(token=token)

industry_game/utils/users/storage.py

@@ -1,4 +1,5 @@
 import asyncio
+from collections.abc import Mapping
 
 from sqlalchemy import func, insert, select
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -18,13 +19,15 @@ async def create(
         *,
         username: str,
         password_hash: str,
+        properties: Mapping[str, str],
         commit: bool = True,
     ) -> FullUser:
         stmt = (
             insert(UserDb)
             .values(
                 username=username,
                 password_hash=password_hash,
+                properties=properties,
             )
             .returning(UserDb)
         )

tests/api/players/test_player_register.py

@@ -0,0 +1,45 @@
+from aiohttp.test_utils import TestClient
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+from yarl import URL
+
+from industry_game.db.models import User
+from industry_game.utils.users.base import UserType
+
+API_URL = URL("/api/v1/players/register/")
+
+
+async def test_player_register_successful_status_created(
+    api_client: TestClient,
+):
+    response = await api_client.post(
+        API_URL,
+        json={
+            "username": "username",
+            "password": "password",
+            "telegram": "telegram",
+            "name": "your name",
+        },
+    )
+    assert response.status == 201
+
+
+async def test_player_register_successful_check_db(
+    api_client: TestClient,
+    session: AsyncSession,
+):
+    await api_client.post(
+        API_URL,
+        json={
+            "username": "username",
+            "password": "password",
+            "telegram": "telegram",
+            "name": "your name",
+        },
+    )
+    user = (
+        await session.scalars(select(User).where(User.username == "username"))
+    ).one()
+
+    assert user.username == "username"
+    assert user.type == UserType.PLAYER