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)