25 lines
758 B
Python
25 lines
758 B
Python
import unittest
|
|
import requests
|
|
import time
|
|
|
|
|
|
def run_tests(proc, test_dir):
|
|
timeout_threshold = 240
|
|
start_time = time.time()
|
|
while time.time()-start_time < timeout_threshold:
|
|
try:
|
|
requests.head("http://localhost:7860/")
|
|
break
|
|
except requests.exceptions.ConnectionError:
|
|
if proc.poll() is not None:
|
|
break
|
|
if proc.poll() is None:
|
|
if test_dir is None:
|
|
test_dir = "test"
|
|
suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir="test")
|
|
result = unittest.TextTestRunner(verbosity=2).run(suite)
|
|
return len(result.failures) + len(result.errors)
|
|
else:
|
|
print("Launch unsuccessful")
|
|
return 1
|