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
|
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
import functools
from testcase import DocumentTranslationTest, Document
from preparer import (
DocumentTranslationPreparer,
DocumentTranslationClientPreparer as _DocumentTranslationClientPreparer,
)
from devtools_testutils import recorded_by_proxy
from azure.ai.translation.document import DocumentTranslationInput, TranslationTarget
from azure.ai.translation.document import DocumentTranslationClient
DocumentTranslationClientPreparer = functools.partial(_DocumentTranslationClientPreparer, DocumentTranslationClient)
class TestDocumentStatus(DocumentTranslationTest):
@DocumentTranslationPreparer()
@DocumentTranslationClientPreparer()
@recorded_by_proxy
def test_list_statuses(self, **kwargs):
client = kwargs.pop("client")
variables = kwargs.pop("variables", {})
# prepare containers and test data
blob_data = [Document(data=b"This is some text")]
source_container_sas_url = self.create_source_container(data=blob_data, variables=variables)
target_container_sas_url = self.create_target_container(variables=variables)
target_language = "es"
# prepare translation inputs
translation_inputs = [
DocumentTranslationInput(
source_url=source_container_sas_url,
targets=[TranslationTarget(target_url=target_container_sas_url, language=target_language)],
)
]
# submit and validate translation operation
translation_id = self._begin_and_validate_translation(
client, translation_inputs, len(blob_data), target_language
)
# get doc statuses
doc_statuses = client.list_document_statuses(translation_id)
assert doc_statuses is not None
# get first doc
first_doc = next(doc_statuses)
assert first_doc.id is not None
# get doc details
doc_status = client.get_document_status(translation_id=translation_id, document_id=first_doc.id)
self._validate_doc_status(doc_status, target_language)
return variables
|
