anti
741e6bb0d3
- Rename project to stealergram throughout - Add pyproject.toml (replaces requirements.txt split, folds pytest.ini) - Replace all em-dashes with hyphens across all source files Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import os
|
|
|
|
# Must be set before config.py is imported by any module.
|
|
# load_dotenv() runs at import time; these setdefaults fill the gap when .env is absent.
|
|
os.environ.setdefault("API_ID", "12345")
|
|
os.environ.setdefault("API_HASH", "dummy_hash_for_tests")
|
|
os.environ.setdefault("BOT_TOKEN", "0:dummy_bot_token")
|
|
os.environ.setdefault("NOTIFY_CHAT_ID", "99999")
|
|
|
|
# Web frontend test defaults - set once here so all web test files see the same values.
|
|
os.environ.setdefault("WEB_SECRET_KEY", "test-secret-key-for-pytest")
|
|
os.environ.setdefault("WEB_ADMIN_USER", "superadmin")
|
|
os.environ.setdefault("WEB_ADMIN_PASS", "superpass")
|
|
|
|
import pytest
|
|
import config
|
|
import utils.scorer as scorer
|
|
|
|
# Two test keywords:
|
|
# @testcorp\.com - employee email domain (triggers CRITICAL)
|
|
# testcorp\.com - plain domain match (triggers LOW baseline)
|
|
TEST_KEYWORDS = [r"@testcorp\.com", r"testcorp\.com"]
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_keywords(monkeypatch):
|
|
"""
|
|
Override TARGET_KEYWORDS for the duration of a test and rebuild the
|
|
scorer's module-level globals so scoring logic uses known test patterns.
|
|
|
|
scorer.py now reads _config.TARGET_KEYWORDS at call time via `import config as _config`,
|
|
so patching config.TARGET_KEYWORDS is sufficient - no direct scorer patch needed.
|
|
"""
|
|
monkeypatch.setattr(config, "TARGET_KEYWORDS", TEST_KEYWORDS)
|
|
monkeypatch.setattr(scorer, "EMPLOYEE_DOMAINS", scorer._build_employee_domains())
|
|
monkeypatch.setattr(scorer, "ORG_DOMAINS", scorer._build_org_domains())
|