def setUp(self):
self.token = 'xoxb-00000000000-ddddddddddddddddddddddddd'
self.user_id = 'U0MMM0MM'
self.client_mock = Mock()
self.subject = SlackBot(self.token, slack_client=self.client_mock)
self.subject.slack.server.login_data = {'self': {'id': self.user_id}}
self.subject.noop = Mock()
class SlackBotTestCase(unittest.TestCase):
def setUp(self):
self.token = 'xoxb-00000000000-ddddddddddddddddddddddddd'
self.user_id = 'U0MMM0MM'
self.client_mock = Mock()
self.subject = SlackBot(self.token, slack_client=self.client_mock)
self.subject.slack.server.login_data = {'self': {'id': self.user_id}}
self.subject.noop = Mock()
def test_start(self):
self.assertFalse(self.subject.slack.rtm_connect.called)
self.assertIsNone(self.subject.user_id)
self.subject.start()
self.assertTrue(self.subject.slack.rtm_connect.called)
self.assertEqual(self.subject.user_id, self.user_id)
def test_user_name(self):
self.subject.slack.server.username = "******"
name = self.subject.user_name
self.assertEqual(name, "TEST")
def test_is_mention(self):
self.subject.user_id = self.user_id
instr_expected = [
("U0MMM0MM", True),
("U0MMM0MM Speak", True),
("Speak U0MMM0MM ", True),
("<@U0MMM0MM>", True),
("<@U0MMM0MM> Speak", True),
("", False),
("not here", False),
]
for instr, expected in instr_expected:
self.assertEqual(self.subject.is_mention(instr), expected, instr)
def test_listen(self):
self.subject._event_generator = Mock(return_value=[1, 2, 3])
self.subject.process_event = Mock()
self.subject.listen()
self.assertEqual(self.subject.process_event.call_count, 3)
def test_process_event_message(self):
event = dict(type='message', channel='chan', text='text')
self.subject.process_message = Mock()
self.subject.user_id = "TEST"
self.subject.process_event(event)
self.subject.process_message.assert_called_with('chan', 'text')
def test_process_event_mention(self):
event = dict(type='message', channel='chan', text='Hello @TEST')
self.subject.process_mention = Mock()
self.subject.user_id = "TEST"
self.subject.process_event(event)
self.subject.process_mention.assert_called_with('chan', 'Hello @TEST')
def test_process_event_bad_event(self):
event = dict(type='weird type', channel='chan')
with self.assertRaises(NotImplementedError):
self.subject.process_event(event)
def test_process_message(self):
'''
This defined in the subclasses
'''
self.subject.process_message('chan', 'text')
self.assertTrue(self.subject.noop.called)
def test_process_mention(self):
'''
This defined in the subclasses
'''
self.subject.process_mention('chan', 'text')
self.assertTrue(self.subject.noop.called)
def test_process_user_typing(self):
self.subject.process_user_typing('chan', 'text')
self.assertTrue(self.subject.noop.called)
def test_process_hello(self):
self.subject.process_hello()
self.assertTrue(self.subject.noop.called)
def test_process_reconnect_url(self):
self.subject.process_reconnect_url()
self.assertTrue(self.subject.noop.called)
def test_process_presence_change(self):
self.subject.process_presence_change()
self.assertTrue(self.subject.noop.called)
def test_process_file_public(self):
self.subject.process_file_public()
self.assertTrue(self.subject.noop.called)
def test_process_file_shared(self):
self.subject.process_file_shared()
self.assertTrue(self.subject.noop.called)
def test_process_file_change(self):
self.subject.process_file_change()
self.assertTrue(self.subject.noop.called)
def test_process_channel_created(self):
self.subject.process_channel_created()
self.assertTrue(self.subject.noop.called)
def test_process_channel_joined(self):
self.subject.process_channel_joined()
self.assertTrue(self.subject.noop.called)
def test_process_emoji_changed(self):
self.subject.process_emoji_changed()
self.assertTrue(self.subject.noop.called)
def test_process_pin_removed(self):
self.subject.process_emoji_changed()
self.assertTrue(self.subject.noop.called)
def test_send_to_channel(self):
self.assertFalse(self.subject.slack.rtm_connect.called)
self.subject.send_to_channel('channel', 'message')
self.subject.slack.rtm_send_message.assert_called_with('channel', 'message')