237899745
0f9948ffc3
Some checks are pending
Docker Image CI / build-and-push-image (push) Waiting to run
144 lines
4.4 KiB
Python
144 lines
4.4 KiB
Python
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"
|