def test_connection_make_query(self, MockUrllib): mock_instance1 = mock.Mock() mock_instance1.headers = {'X-HOLVI-RESULT': 'OK'} MockUrllib.return_value = mock_instance1 headers = {} headers['Test-header'] = 'value' connection = Connection(self.server) response = connection.make_query(headers) self.assertEquals(MockUrllib.call_args[0][0].headers, headers) self.assertEquals(response['X-HOLVI-RESULT'], 'OK') mock_instance1.headers = {'X-HOLVI-RESULT': 'ERROR: 404 Not found'} with self.assertRaises(HolviDataItemException): response = connection.make_query(headers) mock_instance1.headers = {'X-HOLVI-RESULT': 'ERROR: 404 Not found'}