aboutsummaryrefslogtreecommitdiff
path: root/test/basic_features/utils_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/basic_features/utils_test.py')
-rw-r--r--test/basic_features/utils_test.py25
1 files changed, 17 insertions, 8 deletions
diff --git a/test/basic_features/utils_test.py b/test/basic_features/utils_test.py
index 765470c9..0bfc28a0 100644
--- a/test/basic_features/utils_test.py
+++ b/test/basic_features/utils_test.py
@@ -12,12 +12,25 @@ class UtilsTests(unittest.TestCase):
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_embeddings = "http://localhost:7860/sdapi/v1/embeddings"
def test_options_get(self):
self.assertEqual(requests.get(self.url_options).status_code, 200)
+ def test_options_write(self):
+ response = requests.get(self.url_options)
+ self.assertEqual(response.status_code, 200)
+
+ pre_value = response.json()["send_seed"]
+
+ self.assertEqual(requests.post(self.url_options, json={"send_seed":not pre_value}).status_code, 200)
+
+ response = requests.get(self.url_options)
+ self.assertEqual(response.status_code, 200)
+ self.assertEqual(response.json()["send_seed"], not pre_value)
+
+ requests.post(self.url_options, json={"send_seed": pre_value})
+
def test_cmd_flags(self):
self.assertEqual(requests.get(self.url_cmd_flags).status_code, 200)
@@ -41,13 +54,9 @@ class UtilsTests(unittest.TestCase):
def test_prompt_styles(self):
self.assertEqual(requests.get(self.url_prompt_styles).status_code, 200)
-
- def test_artist_categories(self):
- self.assertEqual(requests.get(self.url_artist_categories).status_code, 200)
-
- def test_artists(self):
- self.assertEqual(requests.get(self.url_artists).status_code, 200)
+ def test_embeddings(self):
+ self.assertEqual(requests.get(self.url_embeddings).status_code, 200)
if __name__ == "__main__":
unittest.main()