Harshil Jani
136 lines
5.5 KiB
Python
136 lines
5.5 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# The OpenSearch Contributors require contributions made to
|
|
# this file be licensed under the Apache-2.0 license or a
|
|
# compatible open source license.
|
|
#
|
|
# Modifications Copyright OpenSearch Contributors. See
|
|
# GitHub history for details.
|
|
#
|
|
# Licensed to Elasticsearch B.V. under one or more contributor
|
|
# license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright
|
|
# ownership. Elasticsearch B.V. licenses this file to you under
|
|
# the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
import pytest
|
|
|
|
from opensearchpy.exceptions import RequestError
|
|
|
|
from . import OpenSearchTestCase
|
|
|
|
|
|
class TestSpecialCharacters(OpenSearchTestCase):
|
|
def test_index_with_slash(self) -> None:
|
|
index_name = "movies/shmovies"
|
|
with pytest.raises(RequestError) as e:
|
|
self.client.indices.create(index=index_name)
|
|
self.assertEqual(
|
|
str(e.value),
|
|
"RequestError(400, 'invalid_index_name_exception', 'Invalid index name [movies/shmovies], must not contain the following characters [ , \", *, \\\\, <, |, ,, >, /, ?]')",
|
|
)
|
|
|
|
|
|
class TestUnicode(OpenSearchTestCase):
|
|
def test_indices_lifecycle_english(self) -> None:
|
|
index_name = "movies"
|
|
|
|
index_create_result = self.client.indices.create(index=index_name)
|
|
self.assertTrue(index_create_result["acknowledged"])
|
|
self.assertEqual(index_name, index_create_result["index"])
|
|
|
|
document = {"name": "Solaris", "director": "Andrei Tartakovsky", "year": "2011"}
|
|
id = "solaris@2011"
|
|
doc_insert_result = self.client.index(
|
|
index=index_name, body=document, id=id, refresh=True
|
|
)
|
|
self.assertEqual("created", doc_insert_result["result"])
|
|
self.assertEqual(index_name, doc_insert_result["_index"])
|
|
self.assertEqual(id, doc_insert_result["_id"])
|
|
|
|
doc_delete_result = self.client.delete(index=index_name, id=id)
|
|
self.assertEqual("deleted", doc_delete_result["result"])
|
|
self.assertEqual(index_name, doc_delete_result["_index"])
|
|
self.assertEqual(id, doc_delete_result["_id"])
|
|
|
|
index_delete_result = self.client.indices.delete(index=index_name)
|
|
self.assertTrue(index_delete_result["acknowledged"])
|
|
|
|
def test_indices_lifecycle_russian(self) -> None:
|
|
index_name = "кино"
|
|
index_create_result = self.client.indices.create(index=index_name)
|
|
self.assertTrue(index_create_result["acknowledged"])
|
|
self.assertEqual(index_name, index_create_result["index"])
|
|
|
|
document = {"название": "Солярис", "автор": "Андрей Тарковский", "год": "2011"}
|
|
id = "соларис@2011"
|
|
doc_insert_result = self.client.index(
|
|
index=index_name, body=document, id=id, refresh=True
|
|
)
|
|
self.assertEqual("created", doc_insert_result["result"])
|
|
self.assertEqual(index_name, doc_insert_result["_index"])
|
|
self.assertEqual(id, doc_insert_result["_id"])
|
|
|
|
doc_delete_result = self.client.delete(index=index_name, id=id)
|
|
self.assertEqual("deleted", doc_delete_result["result"])
|
|
self.assertEqual(index_name, doc_delete_result["_index"])
|
|
self.assertEqual(id, doc_delete_result["_id"])
|
|
|
|
index_delete_result = self.client.indices.delete(index=index_name)
|
|
self.assertTrue(index_delete_result["acknowledged"])
|
|
|
|
def test_indices_analyze(self) -> None:
|
|
self.client.indices.analyze(body='{"text": "привет"}')
|
|
|
|
|
|
class TestBulk(OpenSearchTestCase):
|
|
def test_bulk_works_with_string_body(self) -> None:
|
|
docs = '{ "index" : { "_index" : "bulk_test_index", "_id" : "1" } }\n{"answer": 42}'
|
|
response = self.client.bulk(body=docs)
|
|
|
|
self.assertFalse(response["errors"])
|
|
self.assertEqual(1, len(response["items"]))
|
|
|
|
def test_bulk_works_with_bytestring_body(self) -> None:
|
|
docs = b'{ "index" : { "_index" : "bulk_test_index", "_id" : "2" } }\n{"answer": 42}'
|
|
response = self.client.bulk(body=docs)
|
|
|
|
self.assertFalse(response["errors"])
|
|
self.assertEqual(1, len(response["items"]))
|
|
|
|
def test_bulk_works_with_delete(self) -> None:
|
|
docs = '{ "index" : { "_index" : "bulk_test_index", "_id" : "1" } }\n{"answer": 42}\n{ "delete" : { "_index" : "bulk_test_index", "_id": "1" } }'
|
|
response = self.client.bulk(body=docs)
|
|
|
|
self.assertFalse(response["errors"])
|
|
self.assertEqual(2, len(response["items"]))
|
|
|
|
# Check insertion status
|
|
self.assertEqual(201, response["items"][0]["index"]["status"])
|
|
# Check deletion status
|
|
self.assertEqual(200, response["items"][1]["delete"]["status"])
|
|
|
|
|
|
class TestClose(OpenSearchTestCase):
|
|
def test_close_doesnt_break_client(self) -> None:
|
|
self.client.cluster.health()
|
|
self.client.close()
|
|
self.client.cluster.health()
|
|
|
|
def test_with_doesnt_break_client(self) -> None:
|
|
for _ in range(2):
|
|
with self.client as client:
|
|
client.cluster.health()
|