"""Unit testcases for slack client.""" from unittest.mock import MagicMock from unittest.mock import patch from royalty_collection_territory_backfill.connectors.slack_client import SlackClient @patch('royalty_collection_territory_backfill.connectors.slack_client.WebClient') def test_get_client(mock_web_client): """Test get_client method.""" with patch.multiple( 'royalty_collection_territory_backfill.connectors.slack_client', SLACK_BOT_TOKEN='1234' ): SlackClient().get_client() mock_web_client.assert_called_once_with(token='1234') @patch('royalty_collection_territory_backfill.connectors.slack_client.WebClient') def test_post_text_message(mock_web_client): """Test post_text_message method.""" SlackClient.get_client = MagicMock() SlackClient.get_client.return_value.chat_postMessage.return_value = True with patch.multiple( 'royalty_collection_territory_backfill.connectors.slack_client', SLACK_BOT_TOKEN='1234' ): SlackClient().post_text_message('1234', 'test') SlackClient.get_client.return_value.chat_postMessage.assert_called_once_with( channel='1234', text='test' ) @patch('royalty_collection_territory_backfill.connectors.slack_client.WebClient') def test_post_block_message(mock_web_client): """Test post_block_message method.""" SlackClient.get_client = MagicMock() SlackClient.get_client.return_value.chat_postMessage.return_value = True with patch.multiple( 'royalty_collection_territory_backfill.connectors.slack_client', SLACK_BOT_TOKEN='1234' ): SlackClient().post_block_message('1234', ['test'], 'test') SlackClient.get_client.return_value.chat_postMessage.assert_called_once_with( channel='1234', blocks=['test'], text='test' )