ubot2/tests/test_rtmbot_core.py

# -*- coding: utf-8 -*-
try:
    from unittest.mock import Mock
except ImportError:
    from mock import Mock

from testfixtures import LogCapture
from rtmbot.core import RtmBot

def init_rtmbot():
    ''' Initializes an instance of RTMBot with some default values '''
    rtmbot = RtmBot({
        'SLACK_TOKEN': 'test-12345',
        'BASE_PATH': '/tmp/',
        'LOGFILE': '/tmp/rtmbot.log',
        'DEBUG': True
    })
    return rtmbot

def test_init():
    with LogCapture() as l:
        rtmbot = init_rtmbot()

    assert rtmbot.token == 'test-12345'
    assert rtmbot.directory == '/tmp/'
    assert rtmbot.debug == True

    l.check(
        ('root', 'INFO', 'Initialized in: /tmp/')
    )

def test_output():
    ''' Test that sending a message behaves as expected '''
    rtmbot = init_rtmbot()

    # Mock the slack_client object
    slackclient_mock = Mock()
    channel_mock = Mock()
    slackclient_mock.server.channels.find.return_value = channel_mock
    rtmbot.slack_client = slackclient_mock

    # mock the plugin object to return a sample response
    plugin_mock = Mock()
    plugin_mock.do_output.return_value = [['C12345678', 'test message']]
    rtmbot.bot_plugins.append(plugin_mock)

    rtmbot.output()


    # test that the output matches the expected value
    channel_mock.send_message.assert_called_with('test message')

    # test that unicode messages work as expected
    channel_mock.reset_mock()
    plugin_mock.reset_mock()
    plugin_mock.do_output.return_value = [['C12345678', '🚀 testing']]
    rtmbot.output()

    channel_mock.send_message.assert_called_with('🚀 testing')