fluxdev/stable-diffusion-webui-forge

1import pytest2import requests3 4 5def test_options_write(base_url):6    url_options = f"{base_url}/sdapi/v1/options"7    response = requests.get(url_options)8    assert response.status_code == 2009 10    pre_value = response.json()["send_seed"]11 12    assert requests.post(url_options, json={'send_seed': (not pre_value)}).status_code == 20013 14    response = requests.get(url_options)15    assert response.status_code == 20016    assert response.json()['send_seed'] == (not pre_value)17 18    requests.post(url_options, json={"send_seed": pre_value})19 20 21@pytest.mark.parametrize("url", [22    "sdapi/v1/cmd-flags",23    "sdapi/v1/samplers",24    "sdapi/v1/upscalers",25    "sdapi/v1/sd-models",26    "sdapi/v1/hypernetworks",27    "sdapi/v1/face-restorers",28    "sdapi/v1/realesrgan-models",29    "sdapi/v1/prompt-styles",30    "sdapi/v1/embeddings",31])32def test_get_api_url(base_url, url):33    assert requests.get(f"{base_url}/{url}").status_code == 20034