Change all instances of es->opensearch and ES->OPENSEARCH

Signed-off-by: Rushi Agrawal <rushi.agr@gmail.com>
This commit is contained in:
Rushi Agrawal
2021-08-18 00:41:42 +05:30
parent c6cae11f97
commit 004e346bb0
10 changed files with 40 additions and 40 deletions
+2 -2
View File
@@ -5,7 +5,7 @@
# Version 1.0.1
# - Initial version after refactor
if [[ -z $es_node_name ]]; then
if [[ -z $opensearch_node_name ]]; then
# only set these once
set -euo pipefail
export TEST_SUITE=${TEST_SUITE-oss}
@@ -14,7 +14,7 @@ if [[ -z $es_node_name ]]; then
export CLEANUP=${CLEANUP-false}
export OPENSEARCH_URL_EXTENSION=${OPENSEARCH_URL_EXTENSION-http}
export es_node_name=instance
export opensearch_node_name=instance
export opensearch_image=opensearchproject/opensearch
if [[ "$CLUSTER" == "opendistro" ]]; then
export opensearch_image=amazon/opendistro-for-elasticsearch
+6 -6
View File
@@ -23,14 +23,14 @@ source $script_path/functions/imports.sh
set -euo pipefail
echo -e "\033[34;1mINFO:\033[0m Take down node if called twice with the same arguments (DETACH=true) or on seperate terminals \033[0m"
cleanup_node $es_node_name
cleanup_node $opensearch_node_name
master_node_name=${es_node_name}
master_node_name=${opensearch_node_name}
cluster_name=search-rest-test
declare -a volumes
environment=($(cat <<-END
--env node.name=$es_node_name
--env node.name=$opensearch_node_name
--env cluster.name=$cluster_name
--env cluster.initial_master_nodes=$master_node_name
--env discovery.seed_hosts=$master_node_name
@@ -46,9 +46,9 @@ END
NUMBER_OF_NODES=${NUMBER_OF_NODES-1}
http_port=9200
for (( i=0; i<$NUMBER_OF_NODES; i++, http_port++ )); do
node_name=${es_node_name}$i
node_name=${opensearch_node_name}$i
node_url=${external_opensearch_url/9200/${http_port}}$i
if [[ "$i" == "0" ]]; then node_name=$es_node_name; fi
if [[ "$i" == "0" ]]; then node_name=$opensearch_node_name; fi
environment+=($(cat <<-END
--env node.name=$node_name
END
@@ -93,7 +93,7 @@ END
$CLUSTER;
set +x
if wait_for_container "$es_node_name" "$network_name"; then
if wait_for_container "$opensearch_node_name" "$network_name"; then
echo -e "\033[32;1mSUCCESS:\033[0m Running on: $node_url\033[0m"
fi
+2 -2
View File
@@ -132,7 +132,7 @@ def _make_path(*parts):
GLOBAL_PARAMS = ("pretty", "human", "error_trace", "format", "filter_path")
def query_params(*es_query_params):
def query_params(*opensearch_query_params):
"""
Decorator that pops all accepted parameters from method's kwargs and puts
them in the params argument.
@@ -164,7 +164,7 @@ def query_params(*es_query_params):
elif api_key is not None:
headers["authorization"] = "ApiKey %s" % (_base64_auth_header(api_key),)
for p in es_query_params + GLOBAL_PARAMS:
for p in opensearch_query_params + GLOBAL_PARAMS:
if p in kwargs:
v = kwargs.pop(p)
if v is not None:
+2 -2
View File
@@ -98,7 +98,7 @@ class Connection(object):
if cloud_id:
try:
_, cloud_id = cloud_id.split(":")
parent_dn, es_uuid = (
parent_dn, opensearch_uuid = (
binascii.a2b_base64(cloud_id.encode("utf-8"))
.decode("utf-8")
.split("$")[:2]
@@ -110,7 +110,7 @@ class Connection(object):
except (ValueError, IndexError):
raise ImproperlyConfigured("'cloud_id' is not properly formatted")
host = "%s.%s" % (es_uuid, parent_dn)
host = "%s.%s" % (opensearch_uuid, parent_dn)
use_ssl = True
if http_compress is None:
http_compress = True
+6 -6
View File
@@ -80,7 +80,7 @@ class OpenSearchTestCase(TestCase):
def teardown_method(self, _):
# Hidden indices expanded in wildcards in ES 7.7
expand_wildcards = ["open", "closed"]
if self.es_version() >= (7, 7):
if self.opensearch_version() >= (7, 7):
expand_wildcards.append("hidden")
self.client.indices.delete(
@@ -88,10 +88,10 @@ class OpenSearchTestCase(TestCase):
)
self.client.indices.delete_template(name="*", ignore=404)
def es_version(self):
if not hasattr(self, "_es_version"):
self._es_version = es_version(self.client)
return self._es_version
def opensearch_version(self):
if not hasattr(self, "_opensearch_version"):
self._opensearch_version = opensearch_version(self.client)
return self._opensearch_version
def _get_version(version_string):
@@ -101,5 +101,5 @@ def _get_version(version_string):
return tuple(int(v) if v.isdigit() else 999 for v in version)
def es_version(client):
def opensearch_version(client):
return _get_version(client.info()["version"]["number"])
+1 -1
View File
@@ -41,4 +41,4 @@ class OpenSearchTestCase(TestCase):
@classmethod
def setup_class(cls) -> None: ...
def teardown_method(self, _: Any) -> None: ...
def es_version(self) -> Tuple[int, ...]: ...
def opensearch_version(self) -> Tuple[int, ...]: ...
+6 -6
View File
@@ -33,13 +33,13 @@ from os import environ
from os.path import abspath, dirname, exists, join, pardir
def fetch_es_repo():
def fetch_opensearch_repo():
# user is manually setting YAML dir, don't tamper with it
if "TEST_ES_YAML_DIR" in environ:
if "TEST_OPENSEARCH_YAML_DIR" in environ:
return
repo_path = environ.get(
"TEST_ES_REPO",
"TEST_OPENSEARCH_REPO",
abspath(join(dirname(__file__), pardir, pardir, "opensearch")),
)
@@ -51,12 +51,12 @@ def fetch_es_repo():
)
# set YAML test dir
environ["TEST_ES_YAML_DIR"] = join(
environ["TEST_OPENSEARCH_YAML_DIR"] = join(
repo_path, "rest-api-spec", "src", "main", "resources", "rest-api-spec", "test"
)
# fetching of yaml tests disabled, we'll run with what's there
if environ.get("TEST_ES_NOFETCH", False):
if environ.get("TEST_OPENSEARCH_NOFETCH", False):
return
from test_opensearch.test_cases import SkipTest
@@ -87,7 +87,7 @@ def run_all(argv=None):
# fetch yaml tests anywhere that's not GitHub Actions
if "GITHUB_ACTION" not in environ:
fetch_es_repo()
fetch_opensearch_repo()
# always insert coverage when running tests
if argv is None:
@@ -47,7 +47,7 @@ from ...test_server.test_rest_api_spec import (
pytestmark = pytest.mark.asyncio
ES_VERSION = None
OPENSEARCH_VERSION = None
async def await_if_coro(x):
@@ -80,15 +80,15 @@ class AsyncYamlRunner(YamlRunner):
self.section("teardown")
await self.run_code(self._teardown_code)
async def es_version(self):
global ES_VERSION
if ES_VERSION is None:
async def opensearch_version(self):
global OPENSEARCH_VERSION
if OPENSEARCH_VERSION is None:
version_string = (await self.client.info())["version"]["number"]
if "." not in version_string:
return ()
version = version_string.strip().split(".")
ES_VERSION = tuple(int(v) if v.isdigit() else 999 for v in version)
return ES_VERSION
OPENSEARCH_VERSION = tuple(int(v) if v.isdigit() else 999 for v in version)
return OPENSEARCH_VERSION
def section(self, name):
print(("=" * 10) + " " + name + " " + ("=" * 10))
@@ -211,7 +211,7 @@ class AsyncYamlRunner(YamlRunner):
min_version, max_version = version.split("-")
min_version = _get_version(min_version) or (0,)
max_version = _get_version(max_version) or (999,)
if min_version <= (await self.es_version()) <= max_version:
if min_version <= (await self.opensearch_version()) <= max_version:
pytest.skip(reason)
async def _feature_enabled(self, name):
+1 -1
View File
@@ -92,7 +92,7 @@ class TestStreamingBulk(OpenSearchTestCase):
assert False, "exception should have been raised"
def test_different_op_types(self):
if self.es_version() < (0, 90, 1):
if self.opensearch_version() < (0, 90, 1):
raise SkipTest("update supported since 0.90.1")
self.client.index(index="i", id=45, body={})
self.client.index(index="i", id=42, body={})
@@ -116,7 +116,7 @@ SKIP_TESTS = {
}
ES_VERSION = None
OPENSEARCH_VERSION = None
RUN_ASYNC_REST_API_TESTS = (
sys.version_info >= (3, 6)
and os.environ.get("PYTHON_CONNECTION_CLASS") == "RequestsHttpConnection"
@@ -163,15 +163,15 @@ class YamlRunner:
self.section("teardown")
self.run_code(self._teardown_code)
def es_version(self):
global ES_VERSION
if ES_VERSION is None:
def opensearch_version(self):
global OPENSEARCH_VERSION
if OPENSEARCH_VERSION is None:
version_string = (self.client.info())["version"]["number"]
if "." not in version_string:
return ()
version = version_string.strip().split(".")
ES_VERSION = tuple(int(v) if v.isdigit() else 999 for v in version)
return ES_VERSION
OPENSEARCH_VERSION = tuple(int(v) if v.isdigit() else 999 for v in version)
return OPENSEARCH_VERSION
def section(self, name):
print(("=" * 10) + " " + name + " " + ("=" * 10))
@@ -311,7 +311,7 @@ class YamlRunner:
min_version, max_version = version.split("-")
min_version = _get_version(min_version) or (0,)
max_version = _get_version(max_version) or (999,)
if min_version <= (self.es_version()) <= max_version:
if min_version <= (self.opensearch_version()) <= max_version:
pytest.skip(reason)
def run_gt(self, action):