def test_successful_ssl(self, conn_class_mock):
HttpConnection.set_connection_params('test', ssl=True)
conn_class_mock.return_value = conn_mock = MagicMock()
with HttpConnection('test') as (conn, headers):
self.assertEqual(conn_mock, conn)
conn_class_mock.assert_called_with('localhost')
conn_mock.close.assert_called_with()
def test_successful_ssl(self, conn_class_mock):
HttpConnection.set_connection_params('test', ssl=True)
conn_class_mock.return_value = conn_mock = MagicMock()
with HttpConnection('test') as (conn, headers):
self.assertEqual(conn_mock, conn)
conn_class_mock.assert_called_with('localhost')
conn_mock.close.assert_called_with()
def test_successful_auth(self, conn_class_mock):
HttpConnection.set_connection_params('test',
user='******',
password='******')
conn_class_mock.return_value = conn_mock = MagicMock()
with HttpConnection('test') as (conn, headers):
self.assertEqual(conn_mock, conn)
expected = {'Authorization': b'Basic dGVzdHVzZXI6dGVzdHBhc3M='}
self.assertEqual(expected, headers)
conn_class_mock.assert_called_with('localhost')
conn_mock.close.assert_called_with()
def test_successful_auth(self, conn_class_mock):
HttpConnection.set_connection_params('test',
user='******',
password='******')
conn_class_mock.return_value = conn_mock = MagicMock()
with HttpConnection('test') as (conn, headers):
self.assertEqual(conn_mock, conn)
expected = {'Authorization': b'Basic dGVzdHVzZXI6dGVzdHBhc3M='}
self.assertEqual(expected, headers)
conn_class_mock.assert_called_with('localhost')
conn_mock.close.assert_called_with()
def test_set_connection_params(self):
HttpConnection.set_connection_params('test',
host='testhost',
user='******',
password='******',
ssl=True)
self.assertEqual({'test': {'host': 'testhost',
'user': '******',
'password': '******',
'ssl': True}},
HttpConnection._connection_params)
def test_set_connection_params(self):
HttpConnection.set_connection_params('test',
host='testhost',
user='******',
password='******',
ssl=True)
self.assertEqual(
{
'test': {
'host': 'testhost',
'user': '******',
'password': '******',
'ssl': True
}
}, HttpConnection._connection_params)
def test_reset_connection_params(self):
HttpConnection.set_connection_params('test', host='testhost')
HttpConnection.reset_connection_params()
self.assertEqual({}, HttpConnection._connection_params)
def test_reset_connection_params(self):
HttpConnection.set_connection_params('test', host='testhost')
HttpConnection.reset_connection_params()
self.assertEqual({}, HttpConnection._connection_params)
