stable-diffusion-webui/test/server_poll.py

27 lines
835 B
Python
Raw Normal View History

2022-10-29 09:26:28 -06:00
import unittest
import requests
import time
import os
from modules.paths import script_path
2022-10-29 09:26:28 -06:00
2022-11-14 03:39:22 -07:00
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:
2022-11-12 05:12:15 -07:00
if proc.poll() is not None:
break
if proc.poll() is None:
2022-11-14 03:39:22 -07:00
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