def __init__(self) -> None:
self.client = get_proxy_client()
self.parser = reqparse.RequestParser()
self.parser.add_argument('tag_type', type=str, required=False, default='default')
super(TableTagAPI, self).__init__()
self._tag_common = TagCommon(client=self.client)
def test_badge_on_unreserved_badge_value(self) -> None:
self.app.config['WHITELIST_BADGES'] = [BADGE_NAME]
mock_proxy = MagicMock()
tag_common = TagCommon(client=mock_proxy)
response = tag_common.put(id='',
resource_type=ResourceType.Dashboard,
tag=TAG_NAME,
tag_type='badge')
self.assertEqual(response[1], HTTPStatus.NOT_FOUND)
def test_badge_on_reserved_badge_value(self) -> None:
self.app.config['WHITELIST_BADGES'] = [
Badge(badge_name=BADGE_NAME, category='table_status')
]
mock_proxy = MagicMock()
tag_common = TagCommon(client=mock_proxy)
response = tag_common.put(id='',
resource_type=ResourceType.Dashboard,
tag=BADGE_NAME,
tag_type='badge')
self.assertEqual(response[1], HTTPStatus.NOT_ACCEPTABLE)
