26 lines
835 B
Python
26 lines
835 B
Python
import unittest
|
|
import requests
|
|
import time
|
|
import os
|
|
from modules.paths import script_path
|
|
|
|
|
|
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 = os.path.join(script_path, "test")
|
|
suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir=test_dir)
|
|
result = unittest.TextTestRunner(verbosity=2).run(suite)
|
|
return len(result.failures) + len(result.errors)
|
|
else:
|
|
print("Launch unsuccessful")
|
|
return 1
|