Propagate test error and try it without localhost

6 files changed, 18 insertions, 16 deletions

diff --git a/test/advanced_features/extras_test.py b/test/advanced_features/extras_test.py
index 8763f8ed..abdd5aa2 100644
--- a/test/advanced_features/extras_test.py
+++ b/test/advanced_features/extras_test.py
@@ -3,7 +3,7 @@ import unittest
 
 class TestExtrasWorking(unittest.TestCase):
     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 = {
             "resize_mode": 0,
             "show_extras_results": True,
diff --git a/test/advanced_features/txt2img_test.py b/test/advanced_features/txt2img_test.py
index 36ed7b9a..6ab5a242 100644
--- a/test/advanced_features/txt2img_test.py
+++ b/test/advanced_features/txt2img_test.py
@@ -4,7 +4,7 @@ import requests
 
 class TestTxt2ImgWorking(unittest.TestCase):
     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 = {
             "enable_hr": False,
             "denoising_strength": 0,
diff --git a/test/basic_features/img2img_test.py b/test/basic_features/img2img_test.py
index 0a9c1e8a..b9a34113 100644
--- a/test/basic_features/img2img_test.py
+++ b/test/basic_features/img2img_test.py
@@ -6,7 +6,7 @@ from PIL import Image
 
 class TestImg2ImgWorking(unittest.TestCase):
     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 = {
             "init_images": [encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png"))],
             "resize_mode": 0,
diff --git a/test/basic_features/txt2img_test.py b/test/basic_features/txt2img_test.py
index fe4af999..0be675d9 100644
--- a/test/basic_features/txt2img_test.py
+++ b/test/basic_features/txt2img_test.py
@@ -4,7 +4,7 @@ import requests
 
 class TestTxt2ImgWorking(unittest.TestCase):
     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 = {
             "enable_hr": False,
             "denoising_strength": 0,
diff --git a/test/basic_features/utils_test.py b/test/basic_features/utils_test.py
index 9706db8b..592f7619 100644
--- a/test/basic_features/utils_test.py
+++ b/test/basic_features/utils_test.py
@@ -3,17 +3,17 @@ import requests
 
 class UtilsTests(unittest.TestCase):
   def setUp(self):
-    self.url_options = "http://localhost:7860/sdapi/v1/options"
-    self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags"
-    self.url_samplers = "http://localhost:7860/sdapi/v1/samplers"
-    self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers"
-    self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models"
-    self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks"
-    self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers"
-    self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models"
-    self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles"
-    self.url_artist_categories = "http://localhost:7860/sdapi/v1/artist-categories"
-    self.url_artists = "http://localhost:7860/sdapi/v1/artists"
+    self.url_options = "http://127.0.0.1:7860/sdapi/v1/options"
+    self.url_cmd_flags = "http://127.0.0.1:7860/sdapi/v1/cmd-flags"
+    self.url_samplers = "http://127.0.0.1:7860/sdapi/v1/samplers"
+    self.url_upscalers = "http://127.0.0.1:7860/sdapi/v1/upscalers"
+    self.url_sd_models = "http://127.0.0.1:7860/sdapi/v1/sd-models"
+    self.url_hypernetworks = "http://127.0.0.1:7860/sdapi/v1/hypernetworks"
+    self.url_face_restorers = "http://127.0.0.1:7860/sdapi/v1/face-restorers"
+    self.url_realesrgan_models = "http://127.0.0.1:7860/sdapi/v1/realesrgan-models"
+    self.url_prompt_styles = "http://127.0.0.1:7860/sdapi/v1/prompt-styles"
+    self.url_artist_categories = "http://127.0.0.1:7860/sdapi/v1/artist-categories"
+    self.url_artists = "http://127.0.0.1:7860/sdapi/v1/artists"
 
   def test_options_get(self):
     self.assertEqual(requests.get(self.url_options).status_code, 200)
diff --git a/test/server_poll.py b/test/server_poll.py
index c71e906a..028fd476 100644
--- a/test/server_poll.py
+++ b/test/server_poll.py
@@ -8,7 +8,7 @@ def run_tests(proc, test_dir):
     start_time = time.time()
     while time.time()-start_time < timeout_threshold:
         try:
-            requests.head("http://localhost:7860/")
+            requests.head("http://127.0.0.1:7860/")
             break
         except requests.exceptions.ConnectionError:
             if proc.poll() is not None:
@@ -18,5 +18,7 @@ def run_tests(proc, test_dir):
             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)
     else:
         print("Launch unsuccessful")
+        return 1