class Message(Base): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.slack = SlackClient() def run(self): if self.options.get('-') is True: msg = sys.stdin.read().strip() to = self.options.get('TO') sys.stdout.write('{}\n'.format(dumps({'msg': msg, 'to': to}))) self.slack.send_message(to, msg)
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.slack = SlackClient()
def test_get_channel_by_name_with_pound_prefix(slack_client: api.SlackClient): channel = slack_client.get_channel_by_name('#testchannel') slack_client.client.channels.list.assert_called_once_with() assert channel['id'] == 'testchannelid' assert channel['name'] == 'testchannel'
def test_get_member_by_name_without_at_prefix(slack_client: api.SlackClient): member = slack_client.get_member_by_name('testmember') slack_client.client.users.list.assert_called_once_with() assert member['id'] == 'testmemberid' assert member['name'] == 'testmember'
def test_list_channels(slack_client: api.SlackClient): channels = slack_client.list_channels() slack_client.client.channels.list.assert_called_once_with() assert len(channels) == 1 assert channels[0]['id'] == 'testchannelid' assert channels[0]['name'] == 'testchannel'
def test_send_message_to_user_with_at_prefix(slack_client: api.SlackClient): resp = slack_client.send_message('@testmember', 'sam test') slack_client.client.users.list.assert_called_once_with() slack_client.client.im.open.assert_called_once_with('testmemberid') slack_client.client.chat.post_message.assert_called_once_with( channel='testimchannelid', text='sam test', as_user=True)
def test_send_message_to_channel_without_prefix(slack_client: api.SlackClient): with pytest.raises(api.InvalidMessageToException): slack_client.send_message('testchannel', 'sam test')
def test_send_message_to_channel_with_pound_prefix( slack_client: api.SlackClient): resp = slack_client.send_message('#testchannel', 'sam test') slack_client.client.chat.post_message.assert_called_once_with( channel='#testchannel', text='sam test')