| 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() |
|
|
