openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-14 14:45:06 +03:00
Code Issues Packages Projects Releases Wiki Activity

Propagate test error and try it without localhost

Browse Source
This commit is contained in:
Vladimir Repin 2022-11-14 14:36:07 +03:00
parent 3ffc1c6cee
commit 0646040667
7 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
launch.py
View File

@ -234,8 +234,8 @@ def prepare_enviroment():
exit(0) exit(0)
if run_tests: if run_tests:
tests(test_dir) exitcode = tests(test_dir)
exit(0) exit(exitcode)
def tests(test_dir): def tests(test_dir):
@ -251,10 +251,11 @@ def tests(test_dir):
proc = subprocess.Popen([sys.executable, *sys.argv], stdout=stdout, stderr=stderr) proc = subprocess.Popen([sys.executable, *sys.argv], stdout=stdout, stderr=stderr)
import test.server_poll import test.server_poll
test.server_poll.run_tests(proc, test_dir) exitcode = test.server_poll.run_tests(proc, test_dir)
print(f"Stopping Web UI process with id {proc.pid}") print(f"Stopping Web UI process with id {proc.pid}")
proc.kill() proc.kill()
return exitcode
def start(): def start():

2
test/advanced_features/extras_test.py
View File

@ -3,7 +3,7 @@ import unittest
class TestExtrasWorking(unittest.TestCase): class TestExtrasWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_img2img = "http://localhost:7860/sdapi/v1/extra-single-image" self.url_img2img = "http://127.0.0.1:7860/sdapi/v1/extra-single-image"
self.simple_extras = { self.simple_extras = {
"resize_mode": 0, "resize_mode": 0,
"show_extras_results": True, "show_extras_results": True,

2
test/advanced_features/txt2img_test.py
View File

@ -4,7 +4,7 @@ import requests
class TestTxt2ImgWorking(unittest.TestCase): class TestTxt2ImgWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img" self.url_txt2img = "http://127.0.0.1:7860/sdapi/v1/txt2img"
self.simple_txt2img = { self.simple_txt2img = {
"enable_hr": False, "enable_hr": False,
"denoising_strength": 0, "denoising_strength": 0,

2
test/basic_features/img2img_test.py
View File

@ -6,7 +6,7 @@ from PIL import Image
class TestImg2ImgWorking(unittest.TestCase): class TestImg2ImgWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_img2img = "http://localhost:7860/sdapi/v1/img2img" self.url_img2img = "http://127.0.0.1:7860/sdapi/v1/img2img"
self.simple_img2img = { self.simple_img2img = {
"init_images": [encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png"))], "init_images": [encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png"))],
"resize_mode": 0, "resize_mode": 0,

2
test/basic_features/txt2img_test.py
View File

@ -4,7 +4,7 @@ import requests
class TestTxt2ImgWorking(unittest.TestCase): class TestTxt2ImgWorking(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img" self.url_txt2img = "http://127.0.0.1:7860/sdapi/v1/txt2img"
self.simple_txt2img = { self.simple_txt2img = {
"enable_hr": False, "enable_hr": False,
"denoising_strength": 0, "denoising_strength": 0,

22
test/basic_features/utils_test.py
View File

@ -3,17 +3,17 @@ import requests
class UtilsTests(unittest.TestCase): class UtilsTests(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_options = "http://localhost:7860/sdapi/v1/options" self.url_options = "http://127.0.0.1:7860/sdapi/v1/options"
self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags" self.url_cmd_flags = "http://127.0.0.1:7860/sdapi/v1/cmd-flags"
self.url_samplers = "http://localhost:7860/sdapi/v1/samplers" self.url_samplers = "http://127.0.0.1:7860/sdapi/v1/samplers"
self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers" self.url_upscalers = "http://127.0.0.1:7860/sdapi/v1/upscalers"
self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models" self.url_sd_models = "http://127.0.0.1:7860/sdapi/v1/sd-models"
self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks" self.url_hypernetworks = "http://127.0.0.1:7860/sdapi/v1/hypernetworks"
self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers" self.url_face_restorers = "http://127.0.0.1:7860/sdapi/v1/face-restorers"
self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models" self.url_realesrgan_models = "http://127.0.0.1:7860/sdapi/v1/realesrgan-models"
self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles" self.url_prompt_styles = "http://127.0.0.1:7860/sdapi/v1/prompt-styles"
self.url_artist_categories = "http://localhost:7860/sdapi/v1/artist-categories" self.url_artist_categories = "http://127.0.0.1:7860/sdapi/v1/artist-categories"
self.url_artists = "http://localhost:7860/sdapi/v1/artists" self.url_artists = "http://127.0.0.1:7860/sdapi/v1/artists"
def test_options_get(self): def test_options_get(self):
self.assertEqual(requests.get(self.url_options).status_code, 200) self.assertEqual(requests.get(self.url_options).status_code, 200)

4
test/server_poll.py
View File

@ -8,7 +8,7 @@ def run_tests(proc, test_dir):
start_time = time.time() start_time = time.time()
while time.time()-start_time < timeout_threshold: while time.time()-start_time < timeout_threshold:
try: try:
requests.head("http://localhost:7860/") requests.head("http://127.0.0.1:7860/")
break break
except requests.exceptions.ConnectionError: except requests.exceptions.ConnectionError:
if proc.poll() is not None: if proc.poll() is not None:
@ -18,5 +18,7 @@ def run_tests(proc, test_dir):
test_dir = "" test_dir = ""
suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir="test") suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir="test")
result = unittest.TextTestRunner(verbosity=2).run(suite) result = unittest.TextTestRunner(verbosity=2).run(suite)
return len(result.failures)
else: else:
print("Launch unsuccessful") print("Launch unsuccessful")
return 1