test_superfaktura_api: fix format

This commit is contained in:
Richard Kubíček
2025-03-12 21:49:10 +01:00
parent 765334ed92
commit 133702dc1d

View File

@@ -10,21 +10,27 @@ from superfaktura.superfaktura_api import (
SuperFakturaAPIMissingCredentialsException,
)
@pytest.fixture
def api():
with patch.dict(os.environ, {
"SUPERFAKTURA_API_KEY": "test_key",
"SUPERFAKTURA_API_URL": "https://api.superfaktura.cz",
"SUPERFAKTURA_API_EMAIL": "test_email",
"SUPERFAKTURA_API_COMPANY_ID": "test_company_id"
}):
with patch.dict(
os.environ,
{
"SUPERFAKTURA_API_KEY": "test_key",
"SUPERFAKTURA_API_URL": "https://api.superfaktura.cz",
"SUPERFAKTURA_API_EMAIL": "test_email",
"SUPERFAKTURA_API_COMPANY_ID": "test_company_id",
},
):
return SuperFakturaAPI()
def test_missing_credentials():
with patch.dict(os.environ, {}, clear=True):
with pytest.raises(SuperFakturaAPIMissingCredentialsException):
SuperFakturaAPI()
def test_get(api):
with patch("requests.get") as mock_get:
mock_get.return_value.status_code = 200
@@ -32,12 +38,14 @@ def test_get(api):
response = api.get("test_endpoint")
assert response == {"data": "test"}
def test_get_failure(api):
with patch("requests.get") as mock_get:
mock_get.return_value.status_code = 404
with pytest.raises(SuperFakturaAPIException):
api.get("test_endpoint")
def test_download(api):
with patch("requests.get") as mock_get:
mock_get.return_value.status_code = 200
@@ -47,6 +55,7 @@ def test_download(api):
api.download("test_endpoint", f)
mock_file().write.assert_called_once_with(b"test_content")
def test_download_failure(api):
with patch("requests.get") as mock_get:
mock_get.return_value.status_code = 404
@@ -55,6 +64,7 @@ def test_download_failure(api):
with pytest.raises(SuperFakturaAPIException):
api.download("test_endpoint", f)
def test_post(api):
with patch("requests.post") as mock_post:
mock_post.return_value.status_code = 200
@@ -62,6 +72,7 @@ def test_post(api):
response = api.post("test_endpoint", '{"name": "Example"}')
assert response == {"data": "test"}
def test_post_failure(api):
with patch("requests.post") as mock_post:
mock_post.return_value.status_code = 404
@@ -69,15 +80,21 @@ def test_post_failure(api):
with pytest.raises(SuperFakturaAPIException):
api.post("test_endpoint", '{"name": "Example"}')
def test_get_invalid_json(api):
with patch("requests.get") as mock_get:
mock_response = MagicMock()
mock_response.status_code = 200
mock_response.json.side_effect = requests.exceptions.JSONDecodeError("msg", "doc", 0)
mock_response.json.side_effect = requests.exceptions.JSONDecodeError(
"msg", "doc", 0
)
mock_get.return_value = mock_response
with pytest.raises(SuperFakturaAPIException, match="Unable to decode response as JSON"):
with pytest.raises(
SuperFakturaAPIException, match="Unable to decode response as JSON"
):
api.get("test_endpoint")
def test_download_not_writable_descriptor(api):
with patch("requests.get") as mock_get:
mock_get.return_value.status_code = 200