def test_merge_documents_missing(self, api_key, endpoint, index_name, **kwargs): client = SearchClient(endpoint, index_name, AzureKeyCredential(api_key)) batch_client = SearchIndexingBufferedSender( endpoint, index_name, AzureKeyCredential(api_key)) batch_client._batch_size = 2 batch_client.merge_documents([{ "hotelId": "1000", "rating": 1 }, { "hotelId": "4", "rating": 2 }]) batch_client.close() # There can be some lag before a document is searchable if self.is_live: time.sleep(TIME_TO_SLEEP) assert client.get_document_count() == 10 with pytest.raises(HttpResponseError): client.get_document(key="1000") result = client.get_document(key="4") assert result["rating"] == 2
def test_merge_documents_existing(self, api_key, endpoint, index_name, **kwargs): client = SearchClient( endpoint, index_name, AzureKeyCredential(api_key) ) batch_client = SearchIndexingBufferedSender( endpoint, index_name, AzureKeyCredential(api_key) ) batch_client._batch_action_count = 2 batch_client.merge_documents( [{"hotelId": "3", "rating": 1}, {"hotelId": "4", "rating": 2}] ) batch_client.close() # There can be some lag before a document is searchable if self.is_live: time.sleep(TIME_TO_SLEEP) assert client.get_document_count() == 10 result = client.get_document(key="3") assert result["rating"] == 1 result = client.get_document(key="4") assert result["rating"] == 2