added unit tests

api/app/db/dao/model_dao.py

@@ -41,7 +41,6 @@ def update(self, uuid: UUID, model: Model):
             )
             return session.execute(query).rowcount
 
-
     def delete(self, uuid: UUID) -> int:
         with self.db.begin_session() as session:
             deleted_at = datetime.datetime.now(tz=datetime.UTC)

api/app/routes/model_route.py

@@ -1,9 +1,8 @@
 import logging
 from typing import Annotated, List, Optional
 from uuid import UUID
-from requests import Response
 
-from fastapi import APIRouter
+from fastapi import APIRouter, Response
 from fastapi.params import Query
 from fastapi_pagination import Page, Params
 
@@ -56,7 +55,6 @@ def delete_model(model_uuid: UUID):
         def update_model_by_uuid(model_uuid: UUID, model_in: ModelIn):
             if model_service.update_model_by_uuid(model_uuid, model_in):
                 return Response(status_code=200)
-            else:
-                return Response(status_code=404)
+            return Response(status_code=404)
 
         return router

api/tests/dao/model_dao_test.py

@@ -27,6 +27,16 @@ def test_get_by_uuid_empty(self):
         retrieved = self.model_dao.get_by_uuid(uuid.uuid4())
         assert retrieved is None
 
+    def test_update(self):
+        model = db_mock.get_sample_model()
+        self.model_dao.insert(model)
+        model_to_update = model
+        model_to_update.name = 'updated_name'
+        rows = self.model_dao.update(model.uuid, model_to_update)
+        retrieved = self.model_dao.get_by_uuid(model.uuid)
+        assert rows == 1
+        assert retrieved.name == 'updated_name'
+
     def test_delete(self):
         model = db_mock.get_sample_model()
         self.model_dao.insert(model)

api/tests/routes/model_route_test.py

@@ -50,6 +50,34 @@ def test_create_model(self):
         assert jsonable_encoder(model_out) == res.json()
         self.model_service.create_model.assert_called_once_with(model_in)
 
+    def test_update_model_ok(self):
+        model_in = db_mock.get_sample_model_in()
+        self.model_service.update_model_by_uuid = MagicMock(return_value=True)
+
+        res = self.client.post(
+            f'{self.prefix}/{db_mock.MODEL_UUID}',
+            json=jsonable_encoder(model_in),
+        )
+
+        assert res.status_code == 200
+        self.model_service.update_model_by_uuid.assert_called_once_with(
+            db_mock.MODEL_UUID, model_in
+        )
+
+    def test_update_model_ko(self):
+        model_in = db_mock.get_sample_model_in()
+        self.model_service.update_model_by_uuid = MagicMock(return_value=False)
+
+        res = self.client.post(
+            f'{self.prefix}/{db_mock.MODEL_UUID}',
+            json=jsonable_encoder(model_in),
+        )
+
+        assert res.status_code == 404
+        self.model_service.update_model_by_uuid.assert_called_once_with(
+            db_mock.MODEL_UUID, model_in
+        )
+
     def test_get_model_by_uuid(self):
         model = db_mock.get_sample_model()
         model_out = ModelOut.from_model(model)

api/tests/services/model_service_test.py

@@ -66,6 +66,20 @@ def test_get_model_by_uuid_not_found(self):
         )
         self.model_dao.get_by_uuid.assert_called_once()
 
+    def test_update_model_ok(self):
+        model_in = db_mock.get_sample_model_in()
+        self.model_dao.update = MagicMock(return_value=1)
+        res = self.model_service.update_model_by_uuid(model_uuid, model_in)
+
+        assert res is True
+
+    def test_update_model_ko(self):
+        model_in = db_mock.get_sample_model_in()
+        self.model_dao.update = MagicMock(return_value=0)
+        res = self.model_service.update_model_by_uuid(model_uuid, model_in)
+
+        assert res is False
+
     def test_delete_model_ok(self):
         model = db_mock.get_sample_model()
         self.model_dao.get_by_uuid = MagicMock(return_value=model)