def test_connection_make_transaction(self, MockUrllib):
mock_instance1 = mock.Mock()
mock_instance1.headers = {'X-ELFCLOUD-RESULT': 'OK'}
MockUrllib.return_value = mock_instance1
headers = {}
headers['Test-header'] = 'value'
connection = Connection(self.server)
response = connection.make_transaction(headers, '/fetch')
self.assertEquals(MockUrllib.call_args[0][0].headers, headers)
self.assertEquals(response.headers['X-ELFCLOUD-RESULT'], 'OK')
mock_instance1.headers = {'X-ELFCLOUD-RESULT': 'ERROR: 404 Not found'}
self.assertRaises(ECDataItemException, connection.make_transaction,
headers, '/fetch')
def test_connection_auth(self, mock_method):
mock_method.return_value = {'result': 'success'}
params = {
'username': self.username,
'auth_data': self.password,
'auth_method': self.auth_method,
'apikey': self.apikey
}
connection = Connection(self.server)
self.assertFalse(connection._is_authed)
connection.auth(self.username, self.password, self.auth_method,
self.apikey)
connection.make_request.assert_called_once_with("auth", params)
self.assertTrue(connection._is_authed)
def test_connection_make_request(self, MockJSONDecoder, MockUrllib,
MockRequest):
mock_instance1 = mock.Mock()
mock_instance1.open.return_value = io.StringIO("Test response")
MockUrllib.return_value = mock_instance1
mock_instance2 = mock.Mock()
mock_instance2.decode.return_value = {'result': 'success'}
MockJSONDecoder.return_value = mock_instance2
MockRequest.return_value = mock.Mock()
connection = Connection(self.server)
response = connection.make_request('method', 'params')
self.assertEquals('success', response)
mock_instance2.decode.assert_called_once_with("Test response")
headers = {'Content-Type': 'application/json; charset=utf-8'}
MockRequest.assert_called_once_with(
self.server + '1.1/json',
'{"params": "params", "method": "method"}', headers)
def test_connection_init(self):
connection = Connection(self.server)
self.assertEquals(connection._server_url, self.server)