|  | 
|  | 1 | +import io | 
|  | 2 | +import re | 
|  | 3 | +from functools import wraps | 
|  | 4 | +from typing import Any, Callable, Dict, List, Optional, Union | 
|  | 5 | +from unittest import TestCase | 
|  | 6 | +from unittest.mock import DEFAULT, patch | 
|  | 7 | + | 
|  | 8 | +from integrations.clickup import zulip_clickup | 
|  | 9 | +from integrations.clickup.zulip_clickup import ClickUpAPIHandler | 
|  | 10 | + | 
|  | 11 | +MOCK_WEBHOOK_URL = ( | 
|  | 12 | +    "https://YourZulipApp.com/api/v1/external/clickup?api_key=TJ9DnIiNqt51bpfyPll5n2uT4iYxMBW9" | 
|  | 13 | +) | 
|  | 14 | + | 
|  | 15 | +MOCK_AUTH_CODE = "332KKA3321NNAK3MADS" | 
|  | 16 | +MOCK_AUTH_CODE_URL = f"https://YourZulipApp.com/?code={MOCK_AUTH_CODE}" | 
|  | 17 | +MOCK_API_KEY = "X" * 32 | 
|  | 18 | + | 
|  | 19 | +SCRIPT_PATH = "integrations.clickup.zulip_clickup" | 
|  | 20 | + | 
|  | 21 | +MOCK_CREATED_WEBHOOK_ID = "13-13-13-13-1313-13" | 
|  | 22 | +MOCK_DELETE_WEBHOOK_ID = "12-12-12-12-12" | 
|  | 23 | +MOCK_GET_WEBHOOK_IDS = {"endpoint": MOCK_WEBHOOK_URL, "id": MOCK_DELETE_WEBHOOK_ID} | 
|  | 24 | + | 
|  | 25 | +CLICKUP_TEAM_ID = "teamid123" | 
|  | 26 | +CLICKUP_CLIENT_ID = "clientid321" | 
|  | 27 | +CLICKUP_CLIENT_SECRET = "clientsecret322"  # noqa: S105 | 
|  | 28 | + | 
|  | 29 | + | 
|  | 30 | +def make_clickup_request_side_effect( | 
|  | 31 | +    path: str, query: Dict[str, Union[str, List[str]]], method: str | 
|  | 32 | +) -> Optional[Dict[str, Any]]: | 
|  | 33 | +    clickup_api = ClickUpAPIHandler(CLICKUP_CLIENT_ID, CLICKUP_CLIENT_SECRET, CLICKUP_TEAM_ID) | 
|  | 34 | +    api_data_mapper: Dict[str, Dict[str, Dict[str, Any]]] = {  # path -> method -> response | 
|  | 35 | +        clickup_api.ENDPOINTS["oauth"]: { | 
|  | 36 | +            "POST": {"access_token": MOCK_API_KEY}, | 
|  | 37 | +        }, | 
|  | 38 | +        clickup_api.ENDPOINTS["team"]: { | 
|  | 39 | +            "POST": {"id": MOCK_CREATED_WEBHOOK_ID}, | 
|  | 40 | +            "GET": {"webhooks": [MOCK_GET_WEBHOOK_IDS]}, | 
|  | 41 | +        }, | 
|  | 42 | +        clickup_api.ENDPOINTS["webhook"].format(webhook_id=MOCK_DELETE_WEBHOOK_ID): {"DELETE": {}}, | 
|  | 43 | +    } | 
|  | 44 | +    return api_data_mapper.get(path, {}).get(method, DEFAULT) | 
|  | 45 | + | 
|  | 46 | + | 
|  | 47 | +def mock_script_args(selected_events: str = "1,2,3,4,5") -> Callable[[Any], Callable[..., Any]]: | 
|  | 48 | +    def decorator(func: Callable[..., Any]) -> Callable[..., Any]: | 
|  | 49 | +        @wraps(func) | 
|  | 50 | +        def wrapper(*args: Any, **kwargs: Any) -> Any: | 
|  | 51 | +            mock_user_inputs = [MOCK_AUTH_CODE_URL, selected_events] | 
|  | 52 | +            with patch( | 
|  | 53 | +                "sys.argv", | 
|  | 54 | +                [ | 
|  | 55 | +                    "zulip_clickup.py", | 
|  | 56 | +                    "--clickup-team-id", | 
|  | 57 | +                    CLICKUP_TEAM_ID, | 
|  | 58 | +                    "--clickup-client-id", | 
|  | 59 | +                    CLICKUP_CLIENT_ID, | 
|  | 60 | +                    "--clickup-client-secret", | 
|  | 61 | +                    CLICKUP_CLIENT_SECRET, | 
|  | 62 | +                    "--zulip-webhook-url", | 
|  | 63 | +                    MOCK_WEBHOOK_URL, | 
|  | 64 | +                ], | 
|  | 65 | +            ), patch("sys.exit"), patch("builtins.input", side_effect=mock_user_inputs), patch( | 
|  | 66 | +                SCRIPT_PATH + ".ClickUpAPIHandler.make_clickup_request", | 
|  | 67 | +                side_effect=make_clickup_request_side_effect, | 
|  | 68 | +            ): | 
|  | 69 | +                result = func(*args, **kwargs) | 
|  | 70 | + | 
|  | 71 | +            return result | 
|  | 72 | + | 
|  | 73 | +        return wrapper | 
|  | 74 | + | 
|  | 75 | +    return decorator | 
|  | 76 | + | 
|  | 77 | + | 
|  | 78 | +class ZulipClickUpScriptTest(TestCase): | 
|  | 79 | +    @mock_script_args() | 
|  | 80 | +    def test_valid_arguments(self) -> None: | 
|  | 81 | +        with patch(SCRIPT_PATH + ".run") as mock_run, patch( | 
|  | 82 | +            "sys.stdout", new=io.StringIO() | 
|  | 83 | +        ) as mock_stdout: | 
|  | 84 | +            zulip_clickup.main() | 
|  | 85 | +            self.assertRegex(mock_stdout.getvalue(), r"Running Zulip Clickup Integration...") | 
|  | 86 | +            mock_run.assert_called_once_with( | 
|  | 87 | +                CLICKUP_CLIENT_ID, CLICKUP_CLIENT_SECRET, CLICKUP_TEAM_ID, MOCK_WEBHOOK_URL | 
|  | 88 | +            ) | 
|  | 89 | + | 
|  | 90 | +    def test_missing_arguments(self) -> None: | 
|  | 91 | +        with self.assertRaises(SystemExit) as cm: | 
|  | 92 | +            with patch("sys.stderr", new=io.StringIO()) as mock_stderr: | 
|  | 93 | +                zulip_clickup.main() | 
|  | 94 | +        self.assertEqual(cm.exception.code, 2) | 
|  | 95 | +        self.assertRegex( | 
|  | 96 | +            mock_stderr.getvalue(), | 
|  | 97 | +            r"""the following arguments are required: --clickup-team-id, --clickup-client-id, --clickup-client-secret, --zulip-webhook-url\n""", | 
|  | 98 | +        ) | 
|  | 99 | + | 
|  | 100 | +    @mock_script_args() | 
|  | 101 | +    def test_redirect_to_auth_page(self) -> None: | 
|  | 102 | +        with patch("webbrowser.open") as mock_open, patch( | 
|  | 103 | +            "sys.stdout", new=io.StringIO() | 
|  | 104 | +        ) as mock_stdout: | 
|  | 105 | +            zulip_clickup.main() | 
|  | 106 | +            redirect_uri = "https://YourZulipApp.com" | 
|  | 107 | +            mock_open.assert_called_once_with( | 
|  | 108 | +                f"https://app.clickup.com/api?client_id={CLICKUP_CLIENT_ID}&redirect_uri={redirect_uri}" | 
|  | 109 | +            ) | 
|  | 110 | +            expected_output = r""" | 
|  | 111 | +STEP 1 | 
|  | 112 | +---- | 
|  | 113 | +ClickUp authorization page will open in your browser\. | 
|  | 114 | +Please authorize your workspace\(s\)\. | 
|  | 115 | +
 | 
|  | 116 | +Click 'Connect Workspace' on the page to proceed...""" | 
|  | 117 | + | 
|  | 118 | +            self.assertRegex( | 
|  | 119 | +                mock_stdout.getvalue(), | 
|  | 120 | +                expected_output, | 
|  | 121 | +            ) | 
|  | 122 | + | 
|  | 123 | +    @mock_script_args() | 
|  | 124 | +    def test_query_for_auth_code(self) -> None: | 
|  | 125 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 126 | +            zulip_clickup.main() | 
|  | 127 | +            expected_output = r""" | 
|  | 128 | +STEP 2 | 
|  | 129 | +---- | 
|  | 130 | +After you've authorized your workspace, | 
|  | 131 | +you should be redirected to your home URL. | 
|  | 132 | +Please copy your home URL and paste it below. | 
|  | 133 | +It should contain a code, and look similar to this: | 
|  | 134 | +
 | 
|  | 135 | +e.g. """ + re.escape(MOCK_AUTH_CODE_URL) | 
|  | 136 | +            self.assertRegex( | 
|  | 137 | +                mock_stdout.getvalue(), | 
|  | 138 | +                expected_output, | 
|  | 139 | +            ) | 
|  | 140 | + | 
|  | 141 | +    @mock_script_args() | 
|  | 142 | +    def test_select_clickup_events(self) -> None: | 
|  | 143 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 144 | +            zulip_clickup.main() | 
|  | 145 | +            expected_output = r""" | 
|  | 146 | +STEP 3 | 
|  | 147 | +---- | 
|  | 148 | +Please select which ClickUp event notification\(s\) you'd | 
|  | 149 | +like to receive in your Zulip app\. | 
|  | 150 | +EVENT CODES: | 
|  | 151 | +    1 = task | 
|  | 152 | +    2 = list | 
|  | 153 | +    3 = folder | 
|  | 154 | +    4 = space | 
|  | 155 | +    5 = goal | 
|  | 156 | +
 | 
|  | 157 | +    Or, enter \* to subscribe to all events\. | 
|  | 158 | +
 | 
|  | 159 | +Here's an example input if you intend to only receive notifications | 
|  | 160 | +related to task, list and folder: 1,2,3 | 
|  | 161 | +""" | 
|  | 162 | +            self.assertRegex( | 
|  | 163 | +                mock_stdout.getvalue(), | 
|  | 164 | +                expected_output, | 
|  | 165 | +            ) | 
|  | 166 | + | 
|  | 167 | +    @mock_script_args() | 
|  | 168 | +    def test_success_message(self) -> None: | 
|  | 169 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 170 | +            zulip_clickup.main() | 
|  | 171 | +            expected_output = r"SUCCESS: Completed integrating your Zulip app with ClickUp!\s*webhook_id: \d+-\d+-\d+-\d+-\d+-\d+\s*You may delete this script or run it again to reconfigure\s*your integration\." | 
|  | 172 | +            self.assertRegex(mock_stdout.getvalue(), expected_output) | 
|  | 173 | + | 
|  | 174 | +    @mock_script_args(selected_events="*") | 
|  | 175 | +    def test_select_all_events(self) -> None: | 
|  | 176 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 177 | +            zulip_clickup.main() | 
|  | 178 | +            expected_output = ( | 
|  | 179 | +                r"Please enter a valid set of options and only select each option once" | 
|  | 180 | +            ) | 
|  | 181 | +            self.assertNotRegex( | 
|  | 182 | +                mock_stdout.getvalue(), | 
|  | 183 | +                expected_output, | 
|  | 184 | +            ) | 
|  | 185 | + | 
|  | 186 | +    @mock_script_args(selected_events="123123") | 
|  | 187 | +    def test_select_invalid_events(self) -> None: | 
|  | 188 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 189 | +            with self.assertRaises(StopIteration): | 
|  | 190 | +                zulip_clickup.main() | 
|  | 191 | + | 
|  | 192 | +            expected_output = ( | 
|  | 193 | +                r"Please enter a valid set of options and only select each option once" | 
|  | 194 | +            ) | 
|  | 195 | +            self.assertRegex( | 
|  | 196 | +                mock_stdout.getvalue(), | 
|  | 197 | +                expected_output, | 
|  | 198 | +            ) | 
|  | 199 | + | 
|  | 200 | +    @mock_script_args(selected_events="1,1,1,1") | 
|  | 201 | +    def test_invalid_input_multiple_events(self) -> None: | 
|  | 202 | +        with patch("webbrowser.open"), patch("sys.stdout", new=io.StringIO()) as mock_stdout: | 
|  | 203 | +            with self.assertRaises(StopIteration): | 
|  | 204 | +                zulip_clickup.main() | 
|  | 205 | + | 
|  | 206 | +            expected_output = ( | 
|  | 207 | +                r"Please enter a valid set of options and only select each option once" | 
|  | 208 | +            ) | 
|  | 209 | +            self.assertRegex( | 
|  | 210 | +                mock_stdout.getvalue(), | 
|  | 211 | +                expected_output, | 
|  | 212 | +            ) | 
0 commit comments