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