class UserFollowAPITest(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 = UserFollowAPI()

    def test_table_put(self) -> None:
        response = self.api.put(user_id='username',
                                resource_type='table',
                                resource_id='3')
        self.assertEqual(list(response)[1], HTTPStatus.OK)
        self.mock_client.add_resource_relation_by_user.assert_called_with(
            id='3',
            user_id='username',
            relation_type=UserResourceRel.follow,
            resource_type=ResourceType.Table)

    def test_dashboard_put(self) -> None:
        response = self.api.put(user_id='username',
                                resource_type='dashboard',
                                resource_id='3')
        self.assertEqual(list(response)[1], HTTPStatus.OK)
        self.mock_client.add_resource_relation_by_user.assert_called_with(
            id='3',
            user_id='username',
            relation_type=UserResourceRel.follow,
            resource_type=ResourceType.Dashboard)

    def test_table_delete(self) -> None:
        response = self.api.delete(user_id='username',
                                   resource_type='table',
                                   resource_id='3')
        self.assertEqual(list(response)[1], HTTPStatus.OK)
        self.mock_client.delete_resource_relation_by_user.assert_called_with(
            id='3',
            user_id='username',
            relation_type=UserResourceRel.follow,
            resource_type=ResourceType.Table)

    def test_dashboard_delete(self) -> None:
        response = self.api.delete(user_id='username',
                                   resource_type='dashboard',
                                   resource_id='3')
        self.assertEqual(list(response)[1], HTTPStatus.OK)
        self.mock_client.delete_resource_relation_by_user.assert_called_with(
            id='3',
            user_id='username',
            relation_type=UserResourceRel.follow,
            resource_type=ResourceType.Dashboard)
Exemple #2
0
class UserFollowAPITest(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 = UserFollowAPI()

    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_table_relation_by_user.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_table_relation_by_user.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 = UserFollowAPI()