salt/tests/unit/beacons/test_telegram_bot_msg_beacon.py

# coding: utf-8

# Python libs
from __future__ import absolute_import
import datetime

# Salt testing libs
from salttesting import TestCase, skipIf
from salttesting.helpers import ensure_in_syspath
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch

# Salt libs
from salt.beacons import telegram_bot_msg

# Third-party libs
try:
    import telegram
    HAS_TELEGRAM = True
except ImportError:
    HAS_TELEGRAM = False


ensure_in_syspath('../../')


@skipIf(not HAS_TELEGRAM, 'telegram is not available')
@skipIf(NO_MOCK, NO_MOCK_REASON)
class TelegramBotMsgBeaconTestCase(TestCase):
    '''
    Test case for salt.beacons.telegram_bot
    '''
    def setUp(self):
        telegram_bot_msg.__context__ = {}

    def test_validate_empty_config(self, *args, **kwargs):
        ret = telegram_bot_msg.__validate__(None)
        self.assertEqual(ret, (False, ('Configuration for telegram_bot_msg '
                                       'beacon must be a dictionary.')))

    def test_validate_missing_accept_from_config(self, *args, **kwargs):
        ret = telegram_bot_msg.__validate__({
            'token': 'bcd'
        })
        self.assertEqual(ret, (False, ('Not all required configuration for '
                                       'telegram_bot_msg are set.')))

    def test_validate_missing_token_config(self, *args, **kwargs):
        ret = telegram_bot_msg.__validate__({
            'accept_from': []
        })
        self.assertEqual(ret, (False, ('Not all required configuration for '
                                       'telegram_bot_msg are set.')))

    def test_validate_config_not_list_in_accept_from(self, *args, **kwargs):
        ret = telegram_bot_msg.__validate__({
            'token': 'bcd',
            'accept_from': {'nodict': "1"}
        })
        self.assertEqual(ret, (False, ('Configuration for telegram_bot_msg, '
                                       'accept_from must be a list of '
                                       'usernames.')))

    def test_validate_valid_config(self, *args, **kwargs):
        ret = telegram_bot_msg.__validate__({
            'token': 'bcd',
            'accept_from': [
                'username'
            ]
        })
        self.assertEqual(ret, (True, 'Valid beacon configuration.'))

    @patch("salt.beacons.telegram_bot_msg.telegram")
    def test_call_no_updates(self, telegram_api, *args, **kwargs):
        token = 'abc'
        config = {
            'token': token,
            'accept_from': ['tester']
        }
        inst = MagicMock(name='telegram.Bot()')
        telegram_api.Bot = MagicMock(name='telegram', return_value=inst)
        inst.get_updates.return_value = []

        ret = telegram_bot_msg.beacon(config)

        telegram_api.Bot.assert_called_once_with(token)
        self.assertEqual(ret, [])

    @patch("salt.beacons.telegram_bot_msg.telegram")
    def test_call_telegram_return_no_updates_for_user(
            self, telegram_api, *args, **kwargs):
        token = 'abc'
        username = 'tester'
        config = {
            'token': token,
            'accept_from': [username]
        }
        inst = MagicMock(name='telegram.Bot()')
        telegram_api.Bot = MagicMock(name='telegram', return_value=inst)

        username = 'different_user'
        user = telegram.User(id=1, first_name='', username=username)
        chat = telegram.Chat(1, 'private', username=username)
        date = datetime.datetime(2016, 12, 18, 0, 0)
        message = telegram.Message(1, user, date=date, chat=chat)
        update = telegram.update.Update(update_id=1, message=message)

        inst.get_updates.return_value = [update]

        ret = telegram_bot_msg.beacon(config)

        telegram_api.Bot.assert_called_once_with(token)
        self.assertEqual(ret, [])

    @patch("salt.beacons.telegram_bot_msg.telegram")
    def test_call_telegram_returning_updates(self, telegram_api,
                                             *args, **kwargs):
        token = 'abc'
        username = 'tester'
        config = {
            'token': token,
            'accept_from': [username]
        }
        inst = MagicMock(name='telegram.Bot()')
        telegram_api.Bot = MagicMock(name='telegram', return_value=inst)

        user = telegram.User(id=1, first_name='', username=username)
        chat = telegram.Chat(1, 'private', username=username)
        date = datetime.datetime(2016, 12, 18, 0, 0)
        message = telegram.Message(1, user, date=date, chat=chat)
        update = telegram.update.Update(update_id=1, message=message)

        inst.get_updates.return_value = [update]

        ret = telegram_bot_msg.beacon(config)

        telegram_api.Bot.assert_called_once_with(token)
        self.assertTrue(ret)
        self.assertEqual(ret[0]['msgs'][0], message.to_dict())