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
|
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------
from unittest import mock
from azure.core.credentials import AzureKeyCredential
from azure.search.documents._generated.models import SearchDocumentsResult, SearchResult
from azure.search.documents.aio import SearchClient
from azure.search.documents.aio._search_client_async import AsyncSearchPageIterator
from test_search_index_client_async import await_prepared_test
CREDENTIAL = AzureKeyCredential(key="test_api_key")
class TestSearchClientAsync:
@await_prepared_test
@mock.patch(
"azure.search.documents._generated.aio.operations._documents_operations.DocumentsOperations.search_post"
)
async def test_get_count_reset_continuation_token(self, mock_search_post):
client = SearchClient("endpoint", "index name", CREDENTIAL)
result = await client.search(search_text="search text")
assert result._page_iterator_class is AsyncSearchPageIterator
search_result = SearchDocumentsResult()
search_result.results = [SearchResult(additional_properties={"key": "val"})]
mock_search_post.return_value = search_result
await result.__anext__()
result._first_page_iterator_instance.continuation_token = "fake token"
await result.get_count()
assert not result._first_page_iterator_instance.continuation_token
|
