def test_disconnect_exception(self, conn_class_mock):
conn_class_mock.return_value = conn_mock = MagicMock()
conn_mock.disconnect.side_effect = ValueError('disconnect failure')
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_close_exception(self, conn_class_mock):
conn_class_mock.return_value = conn_mock = MagicMock()
conn_mock.close.side_effect = ValueError('close failure')
try:
with HttpConnection('test') as (conn, headers):
self.assertEqual(conn_mock, conn)
except ValueError as exc:
self.assertEqual('close failure', str(exc))
else:
self.fail('ValueError was suppressed or not raised')
conn_class_mock.assert_called_with('localhost')
conn_mock.close.assert_called_with()
