| 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 |
|
|
