mirror of
https://github.com/Eledio/superfaktura-client.git
synced 2025-10-31 16:11:20 +01:00
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
import os
|
|
import pytest
|
|
from unittest.mock import patch, mock_open, MagicMock
|
|
|
|
import requests
|
|
|
|
from superfaktura.superfaktura_api import (
|
|
SuperFakturaAPI,
|
|
SuperFakturaAPIException,
|
|
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"
|
|
}):
|
|
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
|
|
mock_get.return_value.json.return_value = {"data": "test"}
|
|
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
|
|
mock_get.return_value.content = b"test_content"
|
|
with patch("builtins.open", mock_open()) as mock_file:
|
|
with open("test_file", "wb") as f:
|
|
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
|
|
with patch("builtins.open", mock_open()):
|
|
with open("test_file", "wb") as f:
|
|
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
|
|
mock_post.return_value.json.return_value = {"data": "test"}
|
|
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
|
|
mock_post.return_value.json.return_value = {"error": "not found"}
|
|
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_get.return_value = mock_response
|
|
with pytest.raises(SuperFakturaAPIException, match="Unable to decode response as JSON"):
|
|
api.get("test_endpoint")
|