QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-17 23:25:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(reports): add task for slack channels warm-up (#32585)

Browse Source
This commit is contained in:
Usiel Riedl
2025-04-01 01:30:21 +08:00
committed by GitHub
parent 56bf17f879
commit 662f0fa8f4
4 changed files with 84 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tests/unit_tests/utils/slack_test.py
View File

@@ -17,7 +17,7 @@
import pytest
from superset.utils.slack import get_channels_with_search
from superset.utils.slack import get_channels_with_search, SlackChannelTypes
class MockResponse:
@@ -150,15 +150,35 @@ class TestGetChannelsWithSearch:
The server responded with: missing scope: channels:read"""
)
def test_filter_channels_by_specified_types(self, mocker):
@pytest.mark.parametrize(
"types, expected_channel_ids",
[
([SlackChannelTypes.PUBLIC], {"public_channel_id"}),
([SlackChannelTypes.PRIVATE], {"private_channel_id"}),
(
[SlackChannelTypes.PUBLIC, SlackChannelTypes.PRIVATE],
{"public_channel_id", "private_channel_id"},
),
([], {"public_channel_id", "private_channel_id"}),
],
)
def test_filter_channels_by_specified_types(
self, types: list[SlackChannelTypes], expected_channel_ids: set[str], mocker
):
mock_data = {
"channels": [
{
"id": "C12345",
"name": "general",
"id": "public_channel_id",
"name": "open",
"is_member": False,
"is_private": False,
},
{
"id": "private_channel_id",
"name": "secret",
"is_member": False,
"is_private": True,
},
],
"response_metadata": {"next_cursor": None},
}
@@ -168,15 +188,8 @@ The server responded with: missing scope: channels:read"""
mock_client.conversations_list.return_value = mock_response_instance
mocker.patch("superset.utils.slack.get_slack_client", return_value=mock_client)
result = get_channels_with_search(types=["public"])
assert result == [
{
"id": "C12345",
"name": "general",
"is_member": False,
"is_private": False,
}
]
result = get_channels_with_search(types=types)
assert {channel["id"] for channel in result} == expected_channel_ids
def test_handle_pagination_multiple_pages(self, mocker):
mock_data_page1 = {