test_translation_deepl_backend.py
2.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from translation.backends.deepl import DeepLTranslationBackend
class _FakeResponse:
def __init__(self, status_code, payload=None, text=""):
self.status_code = status_code
self._payload = payload or {}
self.text = text
def json(self):
return self._payload
class _FakeSession:
def __init__(self, responses):
self.responses = list(responses)
self.calls = []
def post(self, url, *, headers, json, timeout):
self.calls.append(
{
"url": url,
"headers": headers,
"json": json,
"timeout": timeout,
}
)
response = self.responses.pop(0)
if isinstance(response, Exception):
raise response
return response
def test_deepl_translate_uses_persistent_session_and_connect_timeout():
backend = DeepLTranslationBackend(
api_key="test-key",
api_url="https://api.deepl.com/v2/translate",
timeout=5.0,
glossary_id="glossary-1",
)
fake_session = _FakeSession(
[
_FakeResponse(
200,
payload={"translations": [{"text": "Product title"}]},
)
]
)
backend._session = fake_session
result = backend.translate(
"商品标题",
target_lang="en",
source_lang="zh",
scene="ecommerce_search_query",
)
assert result == "Product title"
assert len(fake_session.calls) == 1
call = fake_session.calls[0]
assert call["timeout"] == (2.0, 5.0)
assert call["json"]["target_lang"] == "EN"
assert call["json"]["source_lang"] == "ZH"
assert call["json"]["glossary_id"] == "glossary-1"
assert call["json"]["context"]
def test_deepl_translate_retries_retryable_failures(monkeypatch):
monkeypatch.setattr("translation.backends.deepl.time.sleep", lambda *_args, **_kwargs: None)
backend = DeepLTranslationBackend(
api_key="test-key",
api_url="https://api.deepl.com/v2/translate",
timeout=5.0,
)
fake_session = _FakeSession(
[
_FakeResponse(429, text="too many requests"),
_FakeResponse(200, payload={"translations": [{"text": "dress"}]}),
]
)
backend._session = fake_session
result = backend.translate(
"连衣裙",
target_lang="en",
source_lang="zh",
scene="general",
)
assert result == "dress"
assert len(fake_session.calls) == 2