42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import requests
|
|
import unittest
|
|
import importlib
|
|
utils = importlib.import_module(
|
|
'extensions.sd-webui-controlnet.tests.utils', 'utils')
|
|
utils.setup_test_env()
|
|
|
|
|
|
class TestDetectEndpointWorking(unittest.TestCase):
|
|
def setUp(self):
|
|
self.base_detect_args = {
|
|
"controlnet_module": "canny",
|
|
"controlnet_input_images": [utils.readImage("test/test_files/img2img_basic.png")],
|
|
"controlnet_processor_res": 512,
|
|
"controlnet_threshold_a": 0,
|
|
"controlnet_threshold_b": 0,
|
|
}
|
|
|
|
def test_detect_with_invalid_module_performed(self):
|
|
detect_args = self.base_detect_args.copy()
|
|
detect_args.update({
|
|
"controlnet_module": "INVALID",
|
|
})
|
|
self.assertEqual(utils.detect(detect_args).status_code, 422)
|
|
|
|
def test_detect_with_no_input_images_performed(self):
|
|
detect_args = self.base_detect_args.copy()
|
|
detect_args.update({
|
|
"controlnet_input_images": [],
|
|
})
|
|
self.assertEqual(utils.detect(detect_args).status_code, 422)
|
|
|
|
def test_detect_with_valid_args_performed(self):
|
|
detect_args = self.base_detect_args
|
|
response = utils.detect(detect_args)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|