237899745
0f9948ffc3
Some checks are pending
Docker Image CI / build-and-push-image (push) Waiting to run
95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
import asyncio
|
|
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
|
|
from src.config.constants import EmailServiceType
|
|
from src.database.models import Base, EmailService
|
|
from src.database.session import DatabaseSessionManager
|
|
from src.services.base import EmailServiceFactory
|
|
from src.web.routes import email as email_routes
|
|
from src.web.routes import registration as registration_routes
|
|
|
|
|
|
class DummySettings:
|
|
custom_domain_base_url = ""
|
|
custom_domain_api_key = None
|
|
|
|
|
|
def test_duck_mail_service_registered():
|
|
service_type = EmailServiceType("duck_mail")
|
|
service_class = EmailServiceFactory.get_service_class(service_type)
|
|
assert service_class is not None
|
|
assert service_class.__name__ == "DuckMailService"
|
|
|
|
|
|
def test_email_service_types_include_duck_mail():
|
|
result = asyncio.run(email_routes.get_service_types())
|
|
duckmail_type = next(item for item in result["types"] if item["value"] == "duck_mail")
|
|
|
|
assert duckmail_type["label"] == "DuckMail"
|
|
field_names = [field["name"] for field in duckmail_type["config_fields"]]
|
|
assert "base_url" in field_names
|
|
assert "default_domain" in field_names
|
|
assert "api_key" in field_names
|
|
|
|
|
|
def test_filter_sensitive_config_marks_duckmail_api_key():
|
|
filtered = email_routes.filter_sensitive_config({
|
|
"base_url": "https://api.duckmail.test",
|
|
"api_key": "dk_test_key",
|
|
"default_domain": "duckmail.sbs",
|
|
})
|
|
|
|
assert filtered["base_url"] == "https://api.duckmail.test"
|
|
assert filtered["default_domain"] == "duckmail.sbs"
|
|
assert filtered["has_api_key"] is True
|
|
assert "api_key" not in filtered
|
|
|
|
|
|
def test_registration_available_services_include_duck_mail(monkeypatch):
|
|
runtime_dir = Path("tests_runtime")
|
|
runtime_dir.mkdir(exist_ok=True)
|
|
db_path = runtime_dir / "duckmail_routes.db"
|
|
if db_path.exists():
|
|
db_path.unlink()
|
|
|
|
manager = DatabaseSessionManager(f"sqlite:///{db_path}")
|
|
Base.metadata.create_all(bind=manager.engine)
|
|
|
|
with manager.session_scope() as session:
|
|
session.add(
|
|
EmailService(
|
|
service_type="duck_mail",
|
|
name="DuckMail 主服务",
|
|
config={
|
|
"base_url": "https://api.duckmail.test",
|
|
"default_domain": "duckmail.sbs",
|
|
"api_key": "dk_test_key",
|
|
},
|
|
enabled=True,
|
|
priority=0,
|
|
)
|
|
)
|
|
|
|
@contextmanager
|
|
def fake_get_db():
|
|
session = manager.SessionLocal()
|
|
try:
|
|
yield session
|
|
finally:
|
|
session.close()
|
|
|
|
monkeypatch.setattr(registration_routes, "get_db", fake_get_db)
|
|
|
|
import src.config.settings as settings_module
|
|
|
|
monkeypatch.setattr(settings_module, "get_settings", lambda: DummySettings())
|
|
|
|
result = asyncio.run(registration_routes.get_available_email_services())
|
|
|
|
assert result["duck_mail"]["available"] is True
|
|
assert result["duck_mail"]["count"] == 1
|
|
assert result["duck_mail"]["services"][0]["name"] == "DuckMail 主服务"
|
|
assert result["duck_mail"]["services"][0]["type"] == "duck_mail"
|
|
assert result["duck_mail"]["services"][0]["default_domain"] == "duckmail.sbs"
|