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)