Return slack api call response in slack_hook (apache#21107)

Author: pingzh

airflow/providers/slack/hooks/slack.py

@@ -19,6 +19,7 @@
 from typing import Any, Optional
 
 from slack_sdk import WebClient
+from slack_sdk.web.slack_response import SlackResponse
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base import BaseHook
@@ -78,7 +79,7 @@ def __get_token(self, token: Any, slack_conn_id: Any) -> str:
 
         raise AirflowException('Cannot get token: No valid Slack token nor slack_conn_id supplied.')
 
-    def call(self, api_method: str, **kwargs) -> None:
+    def call(self, api_method: str, **kwargs) -> SlackResponse:
         """
         Calls Slack WebClient `WebClient.api_call` with given arguments.
 
@@ -89,5 +90,8 @@ def call(self, api_method: str, **kwargs) -> None:
             form-encoding will take place. Optional.
         :param params: The URL parameters to append to the URL. Optional.
         :param json: JSON for the body to attach to the request. Optional.
+        :return: The server's response to an HTTP request. Data from the response can be
+            accessed like a dict.  If the response included 'next_cursor' it can be
+            iterated on to execute subsequent requests.
         """
-        self.client.api_call(api_method, **kwargs)
+        return self.client.api_call(api_method, **kwargs)

docs/spelling_wordlist.txt

@@ -333,6 +333,7 @@ Seedlist
 Sendgrid
 SerializedDAG
 SlackHook
+SlackResponse
 SnowflakeHook
 Spark
 SparkPi

tests/providers/slack/hooks/test_slack.py

@@ -122,11 +122,14 @@ def test_call_with_failure(self, slack_client_class_mock):
         with pytest.raises(SlackApiError):
             slack_hook.call(test_method, data=test_api_params)
 
-    @mock.patch('airflow.providers.slack.hooks.slack.WebClient.api_call', autospec=True)
     @mock.patch('airflow.providers.slack.hooks.slack.WebClient')
-    def test_api_call(self, mock_slack_client, mock_slack_api_call):
+    def test_api_call(self, slack_client_class_mock):
+        slack_client_mock = mock.Mock()
+        slack_client_class_mock.return_value = slack_client_mock
+        slack_client_mock.api_call.return_value = {'ok': True}
+
         slack_hook = SlackHook(token='test_token')
         test_api_json = {'channel': 'test_channel'}
 
         slack_hook.call("chat.postMessage", json=test_api_json)
-        mock_slack_api_call.assert_called_once_with(mock_slack_client, "chat.postMessage", json=test_api_json)
+        slack_client_mock.api_call.assert_called_with("chat.postMessage", json=test_api_json)