class UserOwnAPITest(unittest.TestCase): @mock.patch('metadata_service.api.user.get_proxy_client') def setUp(self, mock_get_proxy_client: MagicMock) -> None: self.mock_client = mock.Mock() mock_get_proxy_client.return_value = self.mock_client self.api = UserOwnAPI() def test_put(self) -> None: response = self.api.put(user_id='username', resource_type='2', table_uri='3') self.assertEqual(list(response)[1], HTTPStatus.OK) self.mock_client.add_owner.assert_called_once() def test_delete(self) -> None: response = self.api.delete(user_id='username', resource_type='2', table_uri='3') self.assertEqual(list(response)[1], HTTPStatus.OK) self.mock_client.delete_owner.assert_called_once()
def setUp(self, mock_get_proxy_client: MagicMock) -> None: self.mock_client = mock.Mock() mock_get_proxy_client.return_value = self.mock_client self.api = UserOwnAPI()