tests/test_external_sources.py

from pathlib import Path
import unittest


ROOT = Path(__file__).resolve().parents[1]


def read_text(relative_path: str) -> str:
    return (ROOT / relative_path).read_text(encoding="utf-8")


class ExternalSourcePolicyTest(unittest.TestCase):
    def test_app_uses_phr00t_v23_single_file_source(self):
        app = read_text("app.py")

        self.assertNotIn("Sneak-Moose", app)
        self.assertIn("Phr00t/Qwen-Image-Edit-Rapid-AIO", app)
        self.assertIn("v23/Qwen-Rapid-AIO-NSFW-v23.safetensors", app)
        self.assertIn("load_phr00t_rapid_transformer", app)
        self.assertIn('PHR00T_TRANSFORMER_PREFIX = "model.diffusion_model."', app)
        self.assertIn("safe_open", app)
        self.assertNotIn("from_single_file", app)

    def test_private_diagnostics_callback_is_removed(self):
        app = read_text("app.py")

        self.assertNotIn("QUALITY_ENHANCEMENT_URL", app)
        self.assertNotIn("QUALITY_ENHANCEMENT_TOKEN", app)
        self.assertNotIn("requests.post", app)
        self.assertNotIn("_emit_diagnostics", app)

    def test_third_party_video_space_is_not_hardcoded(self):
        app = read_text("app.py")

        self.assertNotIn("multimodalart/wan-2-2-first-last-frame", app)
        self.assertIn('VIDEO_SPACE_ID = os.environ.get("VIDEO_SPACE_ID", "").strip()', app)
        self.assertIn("Client(VIDEO_SPACE_ID)", app)

    def test_no_external_logo_image_or_sneak_moose_docs(self):
        relevant_text = "\n".join(
            read_text(path)
            for path in ("app.py", "README.md", "AGENT.md", "CLAUDE.md")
        )

        self.assertNotIn("Sneak-Moose", relevant_text)
        self.assertNotIn("qianwen-res.oss-cn-beijing.aliyuncs.com", relevant_text)
        self.assertNotIn("Rapid-AIO v18", relevant_text)


if __name__ == "__main__":
    unittest.main()