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