Merge pull request #308 from MITLibraries/expand-local-hosts

Support 'opensearch' as local host variation

Author: ghukill

tests/test_opensearch.py

@@ -25,6 +25,11 @@ def test_configure_opensearch_client_for_localhost():
     assert str(result) == "<OpenSearch([{'host': 'localhost', 'port': '9200'}])>"
 
 
+def test_configure_opensearch_client_for_local_opensearch_host():
+    result = tim_os.configure_opensearch_client("opensearch")
+    assert str(result) == "<OpenSearch([{'host': 'opensearch', 'port': '9200'}])>"
+
+
 @mock.patch("boto3.session.Session")
 def test_configure_opensearch_client_for_aws(mocked_boto3_session):  # noqa
     result = tim_os.configure_opensearch_client("fake-dev.us-east-1.es.amazonaws.com")

tim/opensearch.py

@@ -34,7 +34,7 @@ def configure_opensearch_client(url: str) -> OpenSearch:
     Includes the appropriate AWS credentials configuration if the URL is not localhost.
     """
     logger.info("OpenSearch request configurations: %s", REQUEST_CONFIG)
-    if url == "localhost":
+    if url in ["localhost", "opensearch"]:
         return OpenSearch(
             hosts=[{"host": url, "port": "9200"}],
             http_auth=("admin", "admin"),