diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-04 14:40:19 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-04 14:40:19 +0000 |
| commit | da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6 (patch) | |
| tree | a2eec9c47e820e7ab351337f73c99d874b4b904f /test/server_poll.py | |
| parent | cffc240a7327ae60671ff533469fc4ed4bf605de (diff) | |
| parent | 47df0849019abac6722c49512f4dd2285bff5b7d (diff) | |
| download | stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.tar.gz stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.tar.bz2 stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.zip | |
Merge branch 'master' into inpaint_textual_inversion
Diffstat (limited to 'test/server_poll.py')
| -rw-r--r-- | test/server_poll.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/test/server_poll.py b/test/server_poll.py index eeefb7eb..d4df697b 100644 --- a/test/server_poll.py +++ b/test/server_poll.py @@ -3,7 +3,7 @@ import requests import time -def run_tests(): +def run_tests(proc, test_dir): timeout_threshold = 240 start_time = time.time() while time.time()-start_time < timeout_threshold: @@ -11,9 +11,14 @@ def run_tests(): requests.head("http://localhost:7860/") break except requests.exceptions.ConnectionError: - pass - if time.time()-start_time < timeout_threshold: - suite = unittest.TestLoader().discover('', pattern='*_test.py') + if proc.poll() is not None: + break + if proc.poll() is None: + if test_dir is None: + test_dir = "" + 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 |