add forms, shipments tests

backend/tests/routers/test_contracts.py

@@ -122,6 +122,9 @@ class TestContracts:
 
         app.dependency_overrides.clear()
 
+    def test_create_one(self, client, mocker, mock_session, mock_user):
+        pass
+
     def test_delete_one(self, client, mocker, mock_session, mock_user):
         contract_result = contract_factory.contract_public_factory(name='test contract delete')
 

backend/tests/routers/test_forms.py

@@ -5,6 +5,7 @@ from src.main import app
 from src.auth.auth import get_current_user
 import tests.factories.forms as form_factory
 from fastapi.exceptions import HTTPException
+import src.messages as messages
 
 class TestForms:
     def test_get_all(self, client, mocker, mock_session, mock_user):
@@ -123,6 +124,44 @@ class TestForms:
             mock_session,
             form_create
         )
+    
+    def test_create_one_referer_notfound(self, client, mocker, mock_session, mock_user):
+        form_body = form_factory.form_body_factory(name='test form create', referer_id=12312)
+        form_create = form_factory.form_create_factory(name='test form create', referer_id=12312)
+
+        mock = mocker.patch.object(
+            service,
+            'create_one',
+            side_effect=forms_exceptions.UserNotFoundError(messages.Messages.not_found('referer'))
+        )
+
+        response = client.post('/api/forms', json=form_body)
+        response_data = response.json()
+
+        assert response.status_code == 404
+        mock.assert_called_once_with(
+            mock_session,
+            form_create
+        )
+
+    def test_create_one_productor_notfound(self, client, mocker, mock_session, mock_user):
+        form_body = form_factory.form_body_factory(name='test form create', productor_id=1231)
+        form_create = form_factory.form_create_factory(name='test form create', productor_id=1231)
+
+        mock = mocker.patch.object(
+            service,
+            'create_one',
+            side_effect=forms_exceptions.ProductorNotFoundError(messages.Messages.not_found('productor'))
+        )
+
+        response = client.post('/api/forms', json=form_body)
+        response_data = response.json()
+
+        assert response.status_code == 404
+        mock.assert_called_once_with(
+            mock_session,
+            form_create
+        )
 
     def test_create_one_unauthorized(self, client, mocker, mock_session, mock_user):
         def unauthorized():
@@ -164,12 +203,51 @@ class TestForms:
     def test_update_one_notfound(self, client, mocker, mock_session, mock_user):
         form_body = form_factory.form_body_factory(name='test form update')
         form_update = form_factory.form_update_factory(name='test form update')
-        form_result = None
 
         mock = mocker.patch.object(
             service,
             'update_one',
-            side_effect=forms_exceptions.FormNotFoundError('Form 1 not found')
+            side_effect=forms_exceptions.FormNotFoundError(messages.Messages.not_found('form'))
+        )
+
+        response = client.put('/api/forms/2', json=form_body)
+        response_data = response.json()
+
+        assert response.status_code == 404
+        mock.assert_called_once_with(
+            mock_session,
+            2,
+            form_update
+        )
+    
+    def test_update_one_referer_notfound(self, client, mocker, mock_session, mock_user):
+        form_body = form_factory.form_body_factory(name='test form update')
+        form_update = form_factory.form_update_factory(name='test form update')
+
+        mock = mocker.patch.object(
+            service,
+            'update_one',
+            side_effect=forms_exceptions.UserNotFoundError(messages.Messages.not_found('referer'))
+        )
+
+        response = client.put('/api/forms/2', json=form_body)
+        response_data = response.json()
+
+        assert response.status_code == 404
+        mock.assert_called_once_with(
+            mock_session,
+            2,
+            form_update
+        )
+
+    def test_update_one_productor_notfound(self, client, mocker, mock_session, mock_user):
+        form_body = form_factory.form_body_factory(name='test form update')
+        form_update = form_factory.form_update_factory(name='test form update')
+
+        mock = mocker.patch.object(
+            service,
+            'update_one',
+            side_effect=forms_exceptions.ProductorNotFoundError(messages.Messages.not_found('productor'))
         )
 
         response = client.put('/api/forms/2', json=form_body)
@@ -222,7 +300,7 @@ class TestForms:
         mock = mocker.patch.object(
             service,
             'delete_one',
-            side_effect=forms_exceptions.FormNotFoundError('Form 2 not found')
+            side_effect=forms_exceptions.FormNotFoundError(messages.Messages.not_found('form'))
         )
 
         response = client.delete('/api/forms/2')

backend/tests/routers/test_productors.py

@@ -1,11 +1,14 @@
-import src.productors.service as service
-import src.models as models
+from fastapi.exceptions import HTTPException
+
 from src.main import app
+import src.models as models
+import src.messages as messages
 from src.auth.auth import get_current_user
-import tests.factories.productors as productor_factory
+
+import src.productors.service as service
 import src.productors.exceptions as exceptions
 
-from fastapi.exceptions import HTTPException
+import tests.factories.productors as productor_factory
 
 class TestProductors:
     def test_get_all(self, client, mocker, mock_session, mock_user):
@@ -181,7 +184,7 @@ class TestProductors:
         mock = mocker.patch.object(
             service,
             'update_one',
-            side_effect=exceptions.ProductorNotFoundError('Productor 1 not found')
+            side_effect=exceptions.ProductorNotFoundError(messages.Messages.not_found('productor'))
         )
 
         response = client.put('/api/productors/2', json=productor_body)
@@ -234,7 +237,7 @@ class TestProductors:
         mock = mocker.patch.object(
             service,
             'delete_one',
-            side_effect=exceptions.ProductorNotFoundError('Productor 1 not found')
+            side_effect=exceptions.ProductorNotFoundError(messages.Messages.not_found('productor'))
         )
 
         response = client.delete('/api/productors/2')

backend/tests/routers/test_shipments.py

@@ -1,6 +1,8 @@
 import src.shipments.service as service
 import src.models as models
 from src.main import app
+import src.messages as messages
+import src.shipments.exceptions as exceptions
 from src.auth.auth import get_current_user
 import tests.factories.shipments as shipment_factory
 
@@ -177,12 +179,11 @@ class TestShipments:
     def test_update_one_notfound(self, client, mocker, mock_session, mock_user):
         shipment_body = shipment_factory.shipment_body_factory(name='test shipment update')
         shipment_update = shipment_factory.shipment_update_factory(name='test shipment update')
-        shipment_result = None
 
         mock = mocker.patch.object(
             service,
             'update_one',
-            return_value=shipment_result
+            side_effect=exceptions.ShipmentNotFoundError(messages.Messages.not_found('shipment'))
         )
 
         response = client.put('/api/shipments/2', json=shipment_body)
@@ -235,7 +236,7 @@ class TestShipments:
         mock = mocker.patch.object(
             service,
             'delete_one',
-            return_value=shipment_result
+            side_effect=exceptions.ShipmentNotFoundError(messages.Messages.not_found('shipment'))
         )
 
         response = client.delete('/api/shipments/2')

backend/tests/routers/test_users.py

@@ -3,6 +3,7 @@ import src.models as models
 from src.main import app
 from src.auth.auth import get_current_user
 import tests.factories.users as user_factory
+import src.users.exceptions as exceptions
 
 from fastapi.exceptions import HTTPException
 
@@ -178,7 +179,7 @@ class TestUsers:
         mock = mocker.patch.object(
             service,
             'update_one',
-            return_value=user_result
+            side_effect=exceptions.UserNotFoundError('User 2 not found')
         )
 
         response = client.put('/api/users/2', json=user_body)
@@ -231,7 +232,7 @@ class TestUsers:
         mock = mocker.patch.object(
             service,
             'delete_one',
-            return_value=user_result
+            side_effect=exceptions.UserNotFoundError('User 2 not found')
         )
 
         response = client.delete('/api/users/2')