def test_es_client_with_api_key(self) -> None:
"""Get an Elasticsearch client that can access a server on AWS."""
client = sync_user_elasticsearch.get_es_client_from_env()
http_auth = client.transport.kwargs['api_key']
self.assertEqual('my-api-key', http_auth[0])
self.assertEqual('this-api-key', http_auth[1])
def test_es_client_for_aws_access(self) -> None:
"""Get an Elasticsearch client that can access a server on AWS."""
client = sync_user_elasticsearch.get_es_client_from_env()
http_auth = client.transport.kwargs['http_auth']
self.assertEqual('my-access-key', http_auth.access_id)
self.assertEqual('mega-secret', http_auth.signing_key.secret_key)
self.assertEqual('es', http_auth.service)
def test_es_client_from_env_aws_in_docker(
self, mock_requests: '_RequestsMock') -> None:
"""Get an Elasticsearch client for a task running in AWS ECS Docker."""
mock_requests.get('http://169.254.170.2/get-my-credentials',
json={
'AccessKeyId': 'my-access-key',
'SecretAccessKey': 'super-secret'
})
client = sync_user_elasticsearch.get_es_client_from_env()
http_auth = client.transport.kwargs['http_auth']
self.assertEqual('my-access-key', http_auth.access_id)
self.assertEqual('super-secret', http_auth.signing_key.secret_key)
self.assertEqual('es', http_auth.service)
def test_es_client_for_aws_access_without_key(self) -> None:
"""Try getting an Elasticsearch client that can access a server on AWS with no key."""
client = sync_user_elasticsearch.get_es_client_from_env()
http_auth = client.transport.kwargs['http_auth']
self.assertFalse(http_auth)