feat: add test setup and fix related errors

src/database/init/schema.sql

@@ -1,6 +1,6 @@
 CREATE TABLE users (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
-    person_name TEXT NOT NULL,
+    name TEXT NOT NULL,
     age INTEGER,
     height REAL
 );

src/main/routes/user_route.py

@@ -6,5 +6,4 @@
 # Route to register a new user
 @user_routes_bp.route("/user", methods=["POST"])
 def registry_user():
-    # TODO: Implement registration logic and components
     return jsonify({"route": "User registration route"}), 200

src/models/connection/base.py

@@ -0,0 +1,3 @@
+from sqlalchemy.orm import declarative_base
+Base = declarative_base()
+

src/models/connection/db_connection_handle.py → src/models/connection/db_connection_handler.py

@@ -1,9 +1,9 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
-class DbCoonectionHandle:
+class DbConnectionHandler:
     def __init__(self):
-        self.__connection_string = "sqlite:///schema.db"
+        self.__connection_string = "sqlite:///src/database/schema.db"
         self.__engine = self.__create_database_engine()
         self.session = None
 
@@ -12,7 +12,7 @@ def __create_database_engine(self):
         return engine
 
     def __enter__(self):
-        session_make = sessionmaker()
+        session_make = sessionmaker(bind=self.__engine)
         self.session = session_make()
         return self
 

src/models/connection/db_connection_handler_test.py

@@ -0,0 +1,12 @@
+import pytest
+from .db_connection_handler import DbConnectionHandler
+
+#@pytest.mark.skip(reason="Integration with DB")
+def test_db_connection_handler():
+    db_conn_handler = DbConnectionHandler()
+
+    assert db_conn_handler.session is None
+
+    with db_conn_handler:
+        print(db_conn_handler.session)
+        assert db_conn_handler.session is not None

src/models/entities/users.py

@@ -0,0 +1,16 @@
+# pylint: disable=R0903
+
+from sqlalchemy import Column, String, Integer, Float
+from src.models.connection.base import Base
+
+class Users(Base):
+    __tablename__="users"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    name = Column(String, nullable=False)
+    age = Column(Integer)
+    height = Column(Float)
+
+    def __repr__(self):
+        return f"Users [id={self.id}, name{self.name}]"
+

src/models/repositories/users_repository.py

@@ -0,0 +1,29 @@
+from src.models.connection.db_connection_handler import DbConnectionHandler
+from src.models.entities.users import Users
+class UsersRepository:
+    def __init__(self, db_conn_handler: DbConnectionHandler):
+        self.__db_conn_handler = db_conn_handler
+
+    def insert_user(self, name: str, age: int, height: float) -> None:
+        with self.__db_conn_handler as database:
+            try:
+                new_user = Users(
+                    name=name,
+                    age=age,
+                    height=height        
+                )
+                database.session.add(new_user)
+                database.session.commit()
+            except Exception as exception:
+                database.session.rollback()
+                raise exception
+
+    def select_user(self, name: str) -> list[Users]:
+        with self.__db_conn_handler as database:
+            users = (
+                database.session
+                    .query(Users)
+                    .filter(Users.name == name)
+                    .all()
+            )
+            return users

src/models/repositories/users_repository_test.py

@@ -0,0 +1,21 @@
+import pytest
+from src.models.connection.db_connection_handler import DbConnectionHandler
+from .users_repository import UsersRepository
+
+@pytest.mark.skip(reason="Insert in DB")
+def test_users_repository():
+    db_conn = DbConnectionHandler()
+    users_repo = UsersRepository(db_conn)
+
+    name = "Test name"
+    age = 100
+    height = 1.85
+
+    users_repo.insert_user(name, age, height)
+    users = users_repo.select_user(name)
+
+    assert isinstance(users, list)
+    assert len(users) >= 1
+    assert users[0].name == name
+    assert users[0].age == age
+    assert users[0].height == height