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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# coding=utf-8
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
import uuid
import pytest
import functools
from devtools_testutils.aio import recorded_by_proxy_async
from devtools_testutils import set_bodiless_matcher
from azure.core.exceptions import ResourceNotFoundError
from azure.ai.documentintelligence.aio import DocumentIntelligenceAdministrationClient
from azure.ai.documentintelligence.models import (
BuildDocumentClassifierRequest,
AzureBlobContentSource,
AuthorizeClassifierCopyRequest,
ClassifierDocumentTypeDetails,
)
from asynctestcase import AsyncDocumentIntelligenceTest
from preparers import DocumentIntelligencePreparer, GlobalClientPreparerAsync as _GlobalClientPreparer
DocumentModelAdministrationClientPreparer = functools.partial(
_GlobalClientPreparer, DocumentIntelligenceAdministrationClient
)
class TestCopyClassifierAsync(AsyncDocumentIntelligenceTest):
@DocumentIntelligencePreparer()
@DocumentModelAdministrationClientPreparer()
@recorded_by_proxy_async
async def test_copy_classifier_none_classifier_id(self, **kwargs):
client = kwargs.pop("client")
with pytest.raises(ValueError) as e:
await client.begin_copy_classifier_to(classifier_id=None, body={})
assert "No value for given attribute" in str(e.value)
@DocumentIntelligencePreparer()
@DocumentModelAdministrationClientPreparer()
@recorded_by_proxy_async
async def test_copy_classifier_empty_classifier_id(self, **kwargs):
client = kwargs.pop("client")
with pytest.raises(ResourceNotFoundError):
await client.begin_copy_classifier_to(classifier_id="", body={})
@DocumentIntelligencePreparer()
@DocumentModelAdministrationClientPreparer()
@recorded_by_proxy_async
async def test_copy_classifier_successful(
self, client, documentintelligence_training_data_classifier_sas_url, **kwargs
):
set_bodiless_matcher()
recorded_variables = kwargs.pop("variables", {})
recorded_variables.setdefault("classifier_id", str(uuid.uuid4()))
recorded_variables.setdefault("classifier_id_copy", str(uuid.uuid4()))
request = BuildDocumentClassifierRequest(
classifier_id=recorded_variables.get("classifier_id"),
description="IRS document classifier",
doc_types={
"IRS-1040-A": ClassifierDocumentTypeDetails(
azure_blob_source=AzureBlobContentSource(
container_url=documentintelligence_training_data_classifier_sas_url, prefix="IRS-1040-A/train"
)
),
"IRS-1040-B": ClassifierDocumentTypeDetails(
azure_blob_source=AzureBlobContentSource(
container_url=documentintelligence_training_data_classifier_sas_url, prefix="IRS-1040-B/train"
)
),
"IRS-1040-C": ClassifierDocumentTypeDetails(
azure_blob_source=AzureBlobContentSource(
container_url=documentintelligence_training_data_classifier_sas_url, prefix="IRS-1040-C/train"
)
),
"IRS-1040-D": ClassifierDocumentTypeDetails(
azure_blob_source=AzureBlobContentSource(
container_url=documentintelligence_training_data_classifier_sas_url, prefix="IRS-1040-D/train"
)
),
"IRS-1040-E": ClassifierDocumentTypeDetails(
azure_blob_source=AzureBlobContentSource(
container_url=documentintelligence_training_data_classifier_sas_url, prefix="IRS-1040-E/train"
)
),
},
)
poller = await client.begin_build_classifier(request)
classifier = await poller.result()
copy_auth = await client.authorize_classifier_copy(
AuthorizeClassifierCopyRequest(
classifier_id=recorded_variables.get("classifier_id_copy"),
tags={"testkey": "testvalue"},
)
)
poller = await client.begin_copy_classifier_to(classifier.classifier_id, body=copy_auth)
copy = await poller.result()
assert copy.api_version == classifier.api_version
assert copy.classifier_id != classifier.classifier_id
assert copy.classifier_id == copy_auth["targetClassifierId"]
assert copy.base_classifier_id == classifier.base_classifier_id
assert copy.base_classifier_id is None
assert copy.description == classifier.description
return recorded_variables
|
