test: tweaks for consistency

Haidra-Org · Jun 3, 2024 · ef5334b · ef5334b
1 parent 4debaf5
commit ef5334b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -23,7 +23,7 @@ def api_key() -> str:
     raise ValueError("No api key file found")
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture(autouse=True, scope="session")
 def increase_kudos(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     headers = {"apikey": api_key, "Client-Agent": f"aihorde_ci_client:{CIVERSION}:(discord)db0#1625", "user_id": "1"}
 

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -38,6 +38,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
         "name": "CICD Fake Dreamer",
         "models": TEST_MODELS,
         "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "nsfw": True,
         "amount": 10,
         "max_pixels": 4194304,
         "allow_img2img": True,

diff --git a/tests/test_image_extra_sources.py b/tests/test_image_extra_sources.py
@@ -56,6 +56,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
         "name": "CICD Fake Dreamer",
         "models": TEST_MODELS,
         "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "nsfw": True, 
         "amount": 10,
         "max_pixels": 4194304,
         "allow_img2img": True,