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)
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()