fix tests
This commit is contained in:
@@ -92,6 +92,7 @@ class TestShipments:
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
mock_result = shipment_factory.shipment_public_factory(
|
||||
name="test 2", id=2)
|
||||
@@ -101,6 +102,11 @@ class TestShipments:
|
||||
'get_one',
|
||||
return_value=mock_result
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.get('/api/shipments/2')
|
||||
response_data = response.json()
|
||||
@@ -111,12 +117,18 @@ class TestShipments:
|
||||
mock_session,
|
||||
2
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_get_one_notfound(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
mock_result = None
|
||||
mock = mocker.patch.object(
|
||||
@@ -124,12 +136,22 @@ class TestShipments:
|
||||
'get_one',
|
||||
return_value=mock_result
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
response = client.get('/api/shipments/2')
|
||||
assert response.status_code == 404
|
||||
mock.assert_called_once_with(
|
||||
mock_session,
|
||||
2
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_get_one_unauthorized(
|
||||
self,
|
||||
@@ -154,7 +176,8 @@ class TestShipments:
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
):
|
||||
mock_user,
|
||||
):
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment create'
|
||||
)
|
||||
@@ -170,6 +193,11 @@ class TestShipments:
|
||||
'create_one',
|
||||
return_value=shipment_result
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.post('/api/shipments', json=shipment_body)
|
||||
response_data = response.json()
|
||||
@@ -180,6 +208,11 @@ class TestShipments:
|
||||
mock_session,
|
||||
shipment_create
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
shipment=shipment_create
|
||||
)
|
||||
|
||||
def test_create_one_unauthorized(
|
||||
self,
|
||||
@@ -207,6 +240,7 @@ class TestShipments:
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment update'
|
||||
@@ -223,6 +257,11 @@ class TestShipments:
|
||||
'update_one',
|
||||
return_value=shipment_result
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.put('/api/shipments/2', json=shipment_body)
|
||||
response_data = response.json()
|
||||
@@ -234,12 +273,18 @@ class TestShipments:
|
||||
2,
|
||||
shipment_update
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_update_one_notfound(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
shipment_body = shipment_factory.shipment_body_factory(
|
||||
name='test shipment update'
|
||||
@@ -249,8 +294,17 @@ class TestShipments:
|
||||
)
|
||||
|
||||
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')
|
||||
)
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.put('/api/shipments/2', json=shipment_body)
|
||||
|
||||
@@ -260,6 +314,11 @@ class TestShipments:
|
||||
2,
|
||||
shipment_update
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_update_one_unauthorized(
|
||||
self,
|
||||
@@ -287,6 +346,7 @@ class TestShipments:
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
shipment_result = shipment_factory.shipment_public_factory(
|
||||
name='test shipment delete'
|
||||
@@ -297,6 +357,11 @@ class TestShipments:
|
||||
'delete_one',
|
||||
return_value=shipment_result
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.delete('/api/shipments/2')
|
||||
response_data = response.json()
|
||||
@@ -307,12 +372,18 @@ class TestShipments:
|
||||
mock_session,
|
||||
2,
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_delete_one_notfound(
|
||||
self,
|
||||
client,
|
||||
mocker,
|
||||
mock_session,
|
||||
mock_user,
|
||||
):
|
||||
mock = mocker.patch.object(
|
||||
service,
|
||||
@@ -321,6 +392,11 @@ class TestShipments:
|
||||
messages.Messages.not_found('shipment')
|
||||
)
|
||||
)
|
||||
mock_is_allowed = mocker.patch.object(
|
||||
service,
|
||||
'is_allowed',
|
||||
return_value=True
|
||||
)
|
||||
|
||||
response = client.delete('/api/shipments/2')
|
||||
|
||||
@@ -329,6 +405,11 @@ class TestShipments:
|
||||
mock_session,
|
||||
2,
|
||||
)
|
||||
mock_is_allowed.assert_called_once_with(
|
||||
mock_session,
|
||||
mock_user,
|
||||
_id=2
|
||||
)
|
||||
|
||||
def test_delete_one_unauthorized(
|
||||
self,
|
||||
|
||||
Reference in New Issue
Block a user