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_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)