From 4037de58d0c8b615e54dddf5aa01aea8f17845a1 Mon Sep 17 00:00:00 2001 From: Njal Karevoll Date: Tue, 1 Oct 2013 23:40:34 +0200 Subject: [PATCH] requests: add tests to verify that providing http_auth as a list works and that the authorization headers are added to the request --- test_elasticsearch/test_connection.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/test_elasticsearch/test_connection.py b/test_elasticsearch/test_connection.py index d7009b9f..006c54c6 100644 --- a/test_elasticsearch/test_connection.py +++ b/test_elasticsearch/test_connection.py @@ -87,6 +87,10 @@ class TestRequestsConnection(TestCase): con = RequestsHttpConnection(http_auth=('username', 'secret')) self.assertEquals(('username', 'secret'), con.session.auth) + def test_http_auth_list(self): + con = RequestsHttpConnection(http_auth=['username', 'secret']) + self.assertEquals(('username', 'secret'), con.session.auth) + def test_repr(self): con = self._get_mock_connection({"host": "elasticsearch.com", "port": 443}) self.assertEquals('', repr(con)) @@ -169,6 +173,12 @@ class TestRequestsConnection(TestCase): self.assertEquals('GET', request.method) self.assertEquals('{"answer": 42}', request.body) + def test_http_auth_attached(self): + con = self._get_mock_connection(dict(http_auth='username:secret')) + request = self._get_request(con, 'GET', '/') + + self.assertEquals(request.headers['authorization'], 'Basic dXNlcm5hbWU6c2VjcmV0') + @patch('elasticsearch.connection.base.tracer') def test_url_prefix(self, tracer): con = self._get_mock_connection({"url_prefix": "/some-prefix/"})