making tests properly handle exceptions

This commit is contained in:
Game_Time 2023-08-20 16:23:48 +05:00 committed by GitHub
parent 70c8dec419
commit 13416802cb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -92,7 +92,7 @@ def test_api_moderation() -> dict:
def test_all(): def test_all():
"""Runs all tests.""" """Runs all tests."""
try:
print("Waiting until API Server is started up...") print("Waiting until API Server is started up...")
time.sleep(6) time.sleep(6)
@ -110,6 +110,9 @@ def test_all():
print('[lightblue]Checking the /v1/models endpoint...') print('[lightblue]Checking the /v1/models endpoint...')
print(test_models()) print(test_models())
except Exception as e:
print('[red]Error: ' + e)
exit(500)
if __name__ == '__main__': if __name__ == '__main__':
closedai.api_base = api_endpoint closedai.api_base = api_endpoint