63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
|
import unittest
|
||
|
|
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"
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
def test_samplers(self):
|
||
|
|
self.assertEqual(requests.get(self.url_samplers).status_code, 200)
|
||
|
|
|
||
|
|
def test_upscalers(self):
|
||
|
|
self.assertEqual(requests.get(self.url_upscalers).status_code, 200)
|
||
|
|
|
||
|
|
def test_sd_models(self):
|
||
|
|
self.assertEqual(requests.get(self.url_sd_models).status_code, 200)
|
||
|
|
|
||
|
|
def test_hypernetworks(self):
|
||
|
|
self.assertEqual(requests.get(self.url_hypernetworks).status_code, 200)
|
||
|
|
|
||
|
|
def test_face_restorers(self):
|
||
|
|
self.assertEqual(requests.get(self.url_face_restorers).status_code, 200)
|
||
|
|
|
||
|
|
def test_realesrgan_models(self):
|
||
|
|
self.assertEqual(requests.get(self.url_realesrgan_models).status_code, 200)
|
||
|
|
|
||
|
|
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)
|