| from src.services.duck_mail import DuckMailService | |
| class FakeResponse: | |
| def __init__(self, status_code=200, payload=None, text=""): | |
| self.status_code = status_code | |
| self._payload = payload | |
| self.text = text | |
| self.headers = {} | |
| def json(self): | |
| if self._payload is None: | |
| raise ValueError("no json payload") | |
| return self._payload | |
| class FakeHTTPClient: | |
| def __init__(self, responses): | |
| self.responses = list(responses) | |
| self.calls = [] | |
| def request(self, method, url, **kwargs): | |
| self.calls.append({ | |
| "method": method, | |
| "url": url, | |
| "kwargs": kwargs, | |
| }) | |
| if not self.responses: | |
| raise AssertionError(f"未准备响应: {method} {url}") | |
| return self.responses.pop(0) | |
| def test_create_email_creates_account_and_fetches_token(): | |
| service = DuckMailService({ | |
| "base_url": "https://api.duckmail.test", | |
| "default_domain": "duckmail.sbs", | |
| "api_key": "dk_test_key", | |
| "password_length": 10, | |
| }) | |
| fake_client = FakeHTTPClient([ | |
| FakeResponse( | |
| status_code=201, | |
| payload={ | |
| "id": "account-1", | |
| "address": "tester@duckmail.sbs", | |
| "authType": "email", | |
| }, | |
| ), | |
| FakeResponse( | |
| payload={ | |
| "id": "account-1", | |
| "token": "token-123", | |
| } | |
| ), | |
| ]) | |
| service.http_client = fake_client | |
| email_info = service.create_email() | |
| assert email_info["email"] == "tester@duckmail.sbs" | |
| assert email_info["service_id"] == "account-1" | |
| assert email_info["account_id"] == "account-1" | |
| assert email_info["token"] == "token-123" | |
| create_call = fake_client.calls[0] | |
| assert create_call["method"] == "POST" | |
| assert create_call["url"] == "https://api.duckmail.test/accounts" | |
| assert create_call["kwargs"]["json"]["address"].endswith("@duckmail.sbs") | |
| assert len(create_call["kwargs"]["json"]["password"]) == 10 | |
| assert create_call["kwargs"]["headers"]["Authorization"] == "Bearer dk_test_key" | |
| token_call = fake_client.calls[1] | |
| assert token_call["method"] == "POST" | |
| assert token_call["url"] == "https://api.duckmail.test/token" | |
| assert token_call["kwargs"]["json"] == { | |
| "address": "tester@duckmail.sbs", | |
| "password": email_info["password"], | |
| } | |
| def test_get_verification_code_reads_message_detail_and_extracts_code(): | |
| service = DuckMailService({ | |
| "base_url": "https://api.duckmail.test", | |
| "default_domain": "duckmail.sbs", | |
| }) | |
| fake_client = FakeHTTPClient([ | |
| FakeResponse( | |
| status_code=201, | |
| payload={ | |
| "id": "account-1", | |
| "address": "tester@duckmail.sbs", | |
| "authType": "email", | |
| }, | |
| ), | |
| FakeResponse( | |
| payload={ | |
| "id": "account-1", | |
| "token": "token-123", | |
| } | |
| ), | |
| FakeResponse( | |
| payload={ | |
| "hydra:member": [ | |
| { | |
| "id": "msg-1", | |
| "from": { | |
| "name": "OpenAI", | |
| "address": "noreply@openai.com", | |
| }, | |
| "subject": "Your verification code", | |
| "createdAt": "2026-03-19T10:00:00Z", | |
| } | |
| ] | |
| } | |
| ), | |
| FakeResponse( | |
| payload={ | |
| "id": "msg-1", | |
| "text": "Your OpenAI verification code is 654321", | |
| "html": [], | |
| } | |
| ), | |
| ]) | |
| service.http_client = fake_client | |
| email_info = service.create_email() | |
| code = service.get_verification_code( | |
| email=email_info["email"], | |
| email_id=email_info["service_id"], | |
| timeout=1, | |
| ) | |
| assert code == "654321" | |
| messages_call = fake_client.calls[2] | |
| assert messages_call["method"] == "GET" | |
| assert messages_call["url"] == "https://api.duckmail.test/messages" | |
| assert messages_call["kwargs"]["headers"]["Authorization"] == "Bearer token-123" | |
| detail_call = fake_client.calls[3] | |
| assert detail_call["method"] == "GET" | |
| assert detail_call["url"] == "https://api.duckmail.test/messages/msg-1" | |
| assert detail_call["kwargs"]["headers"]["Authorization"] == "Bearer token-123" | |