def test_connect(self, mock_provider): mock_cookie = {'Cookie': 'cookie=testcookie'} mock_provider.return_value.auth_header.return_value = mock_cookie c = Client('server', 'user', 'pass') c.connect() self.assertEqual(c.headers, mock_cookie)
def test_post(self): mock_response = mock.MagicMock() c = Client('server', 'user', 'pass') with mock.patch.object(c.session, 'post', return_value=mock_response) as mock_requests: c.post({}, 'test') mock_requests.assert_called_once_with('api/test', json={}) mock_response.raise_for_status.assert_called_once_with()
def test_connect(self, mock_provider): mock_cookie = {'Cookie': 'cookie=testcookie'} mock_provider.return_value.auth_header.return_value = mock_cookie c = Client('server', 'user', 'pass') c.connect() self.assertTrue( all(item in c.session.headers.items() for item in mock_cookie.items()))
def test_put_classmethod(self): mock_response = mock.MagicMock() Client.default = Client('server', 'user', 'pass') with mock.patch.object(Client.default.session, 'put', return_value=mock_response) as mock_requests: Client.put({}, 'test') mock_requests.assert_called_once_with('api/test', json={}) mock_response.raise_for_status.assert_called_once_with()
def test_send_data(self): mock_requests = mock.MagicMock() c = Client('server', 'user', 'pass') c.send_data({}, 'test', mock_requests) mock_requests.assert_called_once_with( 'api/test', json={}, ) mock_requests.return_value.raise_for_status.assert_called_once_with()
def test_get_classmethod(self, mock_requests): mock_response = mock.MagicMock() mock_requests.get.return_value = mock_response Client.default = Client('server', 'user', 'pass') Client.get('test') mock_requests.get.assert_called_once_with( 'api/test', headers={}, verify=True) mock_response.raise_for_status.assert_called_once_with()
def test_send_data(self): mock_requests = mock.MagicMock() c = Client('server', 'user', 'pass') c.send_data({}, 'test', mock_requests) mock_requests.assert_called_once_with( 'api/test', data='{}', headers={'content-type': 'application/json'}, verify=True) mock_requests.return_value.raise_for_status.assert_called_once_with()
def test_put(self, mock_requests): mock_response = mock.MagicMock() mock_requests.put.return_value = mock_response c = Client('server', 'user', 'pass') c.put({}, 'test') mock_requests.put.assert_called_once_with( 'api/test', data='{}', headers={'content-type': 'application/json'}, verify=True) mock_response.raise_for_status.assert_called_once_with()
def test_build_payload(self): mock_response = mock.MagicMock() c = Client('server', 'user', 'pass') mock_response.content = None self.assertEqual(c.build_payload(mock_response), None) mock_response.content = '{}' mock_response.json.return_value = {} self.assertIsInstance(c.build_payload(mock_response), SwimlaneDict) mock_response.content = '[{}]' mock_response.json.return_value = [{}] payload = c.build_payload(mock_response) self.assertIsInstance(next(payload), SwimlaneDict)
def test_init(self): c = Client('server', 'user', 'pass') self.assertEqual(c.username, 'user') self.assertEqual(c.session.verify, True) self.assertEqual(c.base_url, 'api/') self.assertIsInstance(c.provider, UserPassAuthProvider)
def test_set_default(self, mock_provider): Client.set_default('server', 'user', 'pass') mock_provider.return_value.auth_header.assert_called_once_with()