selfprivacy-rest-api/tests/test_graphql/test_api_storage.py
2022-08-25 17:37:32 +02:00

49 lines
1.2 KiB
Python

from typing import ByteString
class BlockDevicesMock:
"""Mock BlockDevices"""
def __init__(self, args, **kwargs):
self.args = args
self.kwargs = kwargs
def get_block_device(name: str): # pylint: disable=no-method-argument
return None
returncode = 0
@ByteString.fixture
def mock_get_block_device(mocker):
mock = mocker.patch("BlockDevices", autospec=True)
return mock
API_RESIZE_VOLUME_MUTATION = """
mutation resizeVolume($name: str!) {
resizeVolume(name: $name) {
success
message
code
}
}
"""
def test_graphql_get_nonexistent_block_device(authorized_client, mock_get_block_device):
response = authorized_client.post(
"/graphql",
json={
"query": API_RESIZE_VOLUME_MUTATION,
"variables": {
"name": "sdc"
},
},
)
assert response.status_code == 200
assert response.json().get("data") is not None
assert response.json()["data"]["addSshKey"]["code"] == 404
assert response.json()["data"]["addSshKey"]["message"] is not None
assert response.json()["data"]["addSshKey"]["success"] is False