fix test format
This commit is contained in:
@@ -8,6 +8,7 @@ import tests.factories.shipments as shipment_factory
|
||||
|
||||
from fastapi.exceptions import HTTPException
|
||||
|
||||
|
||||
class TestShipments:
|
||||
def test_get_all(self, client, mocker, mock_session, mock_user):
|
||||
mock_results = [
|
||||
@@ -33,6 +34,7 @@ class TestShipments:
|
||||
[],
|
||||
[],
|
||||
)
|
||||
|
||||
def test_get_all_filters(self, client, mocker, mock_session, mock_user):
|
||||
mock_results = [
|
||||
shipment_factory.shipment_public_factory(name="test 2", id=2),
|
||||
@@ -43,7 +45,8 @@ class TestShipments:
|
||||
return_value=mock_results
|
||||
)
|
||||
|
||||
response = client.get('/api/shipments?dates=2025-10-10&names=test 2&forms=contract form 1')
|
||||
response = client.get(
|
||||
'/api/shipments?dates=2025-10-10&names=test 2&forms=contract form 1')
|
||||
response_data = response.json()
|
||||
assert response.status_code == 200
|
||||
assert response_data[0]['id'] == 2
|
||||
@@ -56,10 +59,15 @@ class TestShipments:
|
||||
['contract form 1'],
|
||||
)
|
||||
|
||||
def test_get_all_unauthorized(self, client, mocker, mock_session, mock_user):
|
||||
def test_get_all_unauthorized(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
def unauthorized():
|
||||
raise HTTPException(status_code=401)
|
||||
|
||||
|
||||
app.dependency_overrides[get_current_user] = unauthorized
|
||||
|
||||
mock = mocker.patch('src.shipments.service.get_all')
|
||||
@@ -71,7 +79,8 @@ class TestShipments:
|
||||
app.dependency_overrides.clear()
|
||||
|
||||
def test_get_one(self, client, mocker, mock_session, mock_user):
|
||||
mock_result = shipment_factory.shipment_public_factory(name="test 2", id=2)
|
||||
mock_result = shipment_factory.shipment_public_factory(
|
||||
name="test 2", id=2)
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
@@ -88,7 +97,7 @@ class TestShipments:
|
||||
mock_session,
|
||||
2
|
||||
)
|
||||
|
||||
|
||||
def test_get_one_notfound(self, client, mocker, mock_session, mock_user):
|
||||
mock_result = None
|
||||
mock = mocker.patch.object(
|
||||
@@ -103,11 +112,16 @@ class TestShipments:
|
||||
mock_session,
|
||||
2
|
||||
)
|
||||
|
||||
def test_get_one_unauthorized(self, client, mocker, mock_session, mock_user):
|
||||
|
||||
def test_get_one_unauthorized(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
def unauthorized():
|
||||
raise HTTPException(status_code=401)
|
||||
|
||||
|
||||
app.dependency_overrides[get_current_user] = unauthorized
|
||||
|
||||
mock = mocker.patch('src.shipments.service.get_one')
|
||||
@@ -117,11 +131,14 @@ class TestShipments:
|
||||
mock.assert_not_called()
|
||||
|
||||
app.dependency_overrides.clear()
|
||||
|
||||
|
||||
def test_create_one(self, client, mocker, mock_session, mock_user):
|
||||
shipment_body = shipment_factory.shipment_body_factory(name='test shipment create')
|
||||
shipment_create = shipment_factory.shipment_create_factory(name='test shipment create')
|
||||
shipment_result = shipment_factory.shipment_public_factory(name='test shipment create')
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment create')
|
||||
shipment_create = shipment_factory.shipment_create_factory(
|
||||
name='test shipment create')
|
||||
shipment_result = shipment_factory.shipment_public_factory(
|
||||
name='test shipment create')
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
@@ -139,11 +156,17 @@ class TestShipments:
|
||||
shipment_create
|
||||
)
|
||||
|
||||
def test_create_one_unauthorized(self, client, mocker, mock_session, mock_user):
|
||||
def test_create_one_unauthorized(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
def unauthorized():
|
||||
raise HTTPException(status_code=401)
|
||||
shipment_body = shipment_factory.shipment_body_factory(name='test shipment create')
|
||||
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment create')
|
||||
|
||||
app.dependency_overrides[get_current_user] = unauthorized
|
||||
|
||||
mock = mocker.patch('src.shipments.service.create_one')
|
||||
@@ -155,9 +178,12 @@ class TestShipments:
|
||||
app.dependency_overrides.clear()
|
||||
|
||||
def test_update_one(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 = shipment_factory.shipment_public_factory(name='test shipment update')
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment update')
|
||||
shipment_update = shipment_factory.shipment_update_factory(
|
||||
name='test shipment update')
|
||||
shipment_result = shipment_factory.shipment_public_factory(
|
||||
name='test shipment update')
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
@@ -176,15 +202,20 @@ class TestShipments:
|
||||
shipment_update
|
||||
)
|
||||
|
||||
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')
|
||||
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')
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
'update_one',
|
||||
side_effect=exceptions.ShipmentNotFoundError(messages.Messages.not_found('shipment'))
|
||||
)
|
||||
service, 'update_one', side_effect=exceptions.ShipmentNotFoundError(
|
||||
messages.Messages.not_found('shipment')))
|
||||
|
||||
response = client.put('/api/shipments/2', json=shipment_body)
|
||||
response_data = response.json()
|
||||
@@ -196,11 +227,17 @@ class TestShipments:
|
||||
shipment_update
|
||||
)
|
||||
|
||||
def test_update_one_unauthorized(self, client, mocker, mock_session, mock_user):
|
||||
def test_update_one_unauthorized(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
def unauthorized():
|
||||
raise HTTPException(status_code=401)
|
||||
shipment_body = shipment_factory.shipment_body_factory(name='test shipment update')
|
||||
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment update')
|
||||
|
||||
app.dependency_overrides[get_current_user] = unauthorized
|
||||
|
||||
mock = mocker.patch('src.shipments.service.update_one')
|
||||
@@ -212,7 +249,8 @@ class TestShipments:
|
||||
app.dependency_overrides.clear()
|
||||
|
||||
def test_delete_one(self, client, mocker, mock_session, mock_user):
|
||||
shipment_result = shipment_factory.shipment_public_factory(name='test shipment delete')
|
||||
shipment_result = shipment_factory.shipment_public_factory(
|
||||
name='test shipment delete')
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
@@ -230,14 +268,17 @@ class TestShipments:
|
||||
2,
|
||||
)
|
||||
|
||||
def test_delete_one_notfound(self, client, mocker, mock_session, mock_user):
|
||||
def test_delete_one_notfound(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
shipment_result = None
|
||||
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
'delete_one',
|
||||
side_effect=exceptions.ShipmentNotFoundError(messages.Messages.not_found('shipment'))
|
||||
)
|
||||
service, 'delete_one', side_effect=exceptions.ShipmentNotFoundError(
|
||||
messages.Messages.not_found('shipment')))
|
||||
|
||||
response = client.delete('/api/shipments/2')
|
||||
response_data = response.json()
|
||||
@@ -248,11 +289,17 @@ class TestShipments:
|
||||
2,
|
||||
)
|
||||
|
||||
def test_delete_one_unauthorized(self, client, mocker, mock_session, mock_user):
|
||||
def test_delete_one_unauthorized(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user):
|
||||
def unauthorized():
|
||||
raise HTTPException(status_code=401)
|
||||
shipment_body = shipment_factory.shipment_body_factory(name='test shipment delete')
|
||||
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment delete')
|
||||
|
||||
app.dependency_overrides[get_current_user] = unauthorized
|
||||
|
||||
mock = mocker.patch('src.shipments.service.delete_one')
|
||||
@@ -261,4 +308,4 @@ class TestShipments:
|
||||
assert response.status_code == 401
|
||||
mock.assert_not_called()
|
||||
|
||||
app.dependency_overrides.clear()
|
||||
app.dependency_overrides.clear()
|
||||
|
||||
Reference in New Issue
Block a user