test: introduce basic tests

This commit is contained in:
Richard Kubíček
2025-03-05 12:07:46 +01:00
parent e668f76b4c
commit 886004a20a
3 changed files with 71 additions and 0 deletions

0
test/__init__.py Normal file
View File

View File

@@ -0,0 +1,68 @@
import os
import pytest
import requests
from unittest.mock import patch, mock_open
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()) as mock_file:
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"}')