43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from creyPY.fastapi.testing import GenericClient
|
|
from .main import app
|
|
|
|
|
|
class TestAPI:
|
|
def setup_class(self):
|
|
self.c = GenericClient(app)
|
|
|
|
def test_swagger_gen(self):
|
|
re = self.c.get("/openapi.json")
|
|
assert re["info"]["title"] == "ServerCrow Pong API"
|
|
|
|
def test_health_check(self):
|
|
self.c.get("/", parse_json=False)
|
|
|
|
def test_get_pong(self):
|
|
re = self.c.get("/pong/?code=200&response_text=OK", parse_json=False)
|
|
assert re == b"OK"
|
|
|
|
def test_get_pong_404(self):
|
|
re = self.c.get("/pong/?code=404&response_text=Not Found", parse_json=False, r_code=404)
|
|
assert re == b"Not Found"
|
|
|
|
def test_get_pong_503(self):
|
|
re = self.c.get(
|
|
"/pong/?code=503&response_text=Service Unavailable", parse_json=False, r_code=503
|
|
)
|
|
assert re == b"Service Unavailable"
|
|
|
|
def test_get_pong_500(self):
|
|
re = self.c.get(
|
|
"/pong/?code=500&response_text=Internal Server Error", parse_json=False, r_code=500
|
|
)
|
|
assert re == b"Internal Server Error"
|
|
|
|
def test_get_pong_400(self):
|
|
re = self.c.get("/pong/?code=400&response_text=Bad Request", parse_json=False, r_code=400)
|
|
assert re == b"Bad Request"
|
|
|
|
def test_get_pong_401(self):
|
|
re = self.c.get("/pong/?code=401&response_text=Unauthorized", parse_json=False, r_code=401)
|
|
assert re == b"Unauthorized"
|