def test_characters_escaped_in_path(self):
"""
Assert we generate the 'correct' cannonical query string
and path a request with characters that need to be escaped
"""
url = 'http://search-foo.us-east-1.es.amazonaws.com:80/+foo.*/_stats'
mock_request = mock.Mock()
mock_request.url = url
self.assertEqual('/%2Bfoo.%2A/_stats',
AWSRequestsAuth.get_canonical_path(mock_request))
self.assertEqual(
'', AWSRequestsAuth.get_canonical_querystring(mock_request))
def test_path_with_querystring(self):
"""
Assert we generate the 'correct' cannonical query string
and path for request that includes a query stirng
"""
url = 'http://search-foo.us-east-1.es.amazonaws.com:80/my_index/?pretty=True'
mock_request = mock.Mock()
mock_request.url = url
self.assertEqual('/my_index/',
AWSRequestsAuth.get_canonical_path(mock_request))
self.assertEqual(
'pretty=True',
AWSRequestsAuth.get_canonical_querystring(mock_request))
def test_no_query_params(self):
"""
Assert we generate the 'correct' cannonical query string
and canonical path for a request with no query params
Correct is relative here b/c 'correct' simply means what
the AWS Elasticsearch service expects
"""
url = 'http://search-foo.us-east-1.es.amazonaws.com:80/'
mock_request = mock.Mock()
mock_request.url = url
self.assertEqual('/', AWSRequestsAuth.get_canonical_path(mock_request))
self.assertEqual(
'', AWSRequestsAuth.get_canonical_querystring(mock_request))
