test_int_sync_api_provider.py 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. # This module is a copy paste of test_int_api_provider.py
  2. from http import HTTPStatus
  3. import pytest
  4. from clean_python import ctx
  5. from clean_python import Tenant
  6. from clean_python.api_client import ApiException
  7. from clean_python.api_client import SyncApiProvider
  8. @pytest.fixture
  9. def provider(fastapi_example_app) -> SyncApiProvider:
  10. ctx.tenant = Tenant(id=2, name="")
  11. yield SyncApiProvider(fastapi_example_app + "/", lambda a, b: "token")
  12. ctx.tenant = None
  13. def test_request_params(provider: SyncApiProvider):
  14. response = provider.request("GET", "v1/books", params={"limit": 10, "offset": 2})
  15. assert isinstance(response, dict)
  16. assert response["limit"] == 10
  17. assert response["offset"] == 2
  18. def test_request_json_body(provider: SyncApiProvider):
  19. response = provider.request(
  20. "POST", "v1/books", json={"title": "test_body", "author": {"name": "foo"}}
  21. )
  22. assert isinstance(response, dict)
  23. assert response["title"] == "test_body"
  24. assert response["author"] == {"name": "foo"}
  25. def test_request_form_body(provider: SyncApiProvider):
  26. response = provider.request("POST", "v1/form", fields={"name": "foo"})
  27. assert isinstance(response, dict)
  28. assert response["name"] == "foo"
  29. def test_request_form_file(provider: SyncApiProvider):
  30. response = provider.request("POST", "v1/file", fields={"file": ("x.txt", b"foo")})
  31. assert isinstance(response, dict)
  32. assert response["x.txt"] == "foo"
  33. @pytest.fixture
  34. def book(provider: SyncApiProvider):
  35. return provider.request(
  36. "POST", "v1/books", json={"title": "fixture", "author": {"name": "foo"}}
  37. )
  38. def test_no_content(provider: SyncApiProvider, book):
  39. response = provider.request("DELETE", f"v1/books/{book['id']}")
  40. assert response is None
  41. def test_not_found(provider: SyncApiProvider):
  42. with pytest.raises(ApiException) as e:
  43. provider.request("GET", "v1/book")
  44. assert e.value.status is HTTPStatus.NOT_FOUND
  45. assert e.value.args[0] == {"detail": "Not Found"}
  46. def test_bad_request(provider: SyncApiProvider):
  47. with pytest.raises(ApiException) as e:
  48. provider.request("GET", "v1/books", params={"limit": "foo"})
  49. assert e.value.status is HTTPStatus.BAD_REQUEST
  50. assert e.value.args[0]["detail"][0]["loc"] == ["query", "limit"]
  51. def test_no_json_response(provider: SyncApiProvider):
  52. with pytest.raises(ApiException) as e:
  53. provider.request("GET", "v1/text")
  54. assert e.value.args[0] == "Unexpected content type 'text/plain; charset=utf-8'"
  55. def test_urlencode(provider: SyncApiProvider):
  56. response = provider.request("PUT", "v1/urlencode/x?")
  57. assert isinstance(response, dict)
  58. assert response["name"] == "x?"
  59. def test_request_raw(provider: SyncApiProvider, book):
  60. response = provider.request_raw("GET", f"v1/books/{book['id']}")
  61. assert response.status is HTTPStatus.OK
  62. assert len(response.data) > 0
  63. assert response.content_type == "application/json"