python-telegram-bot/tests/test_botan.py

#!/usr/bin/env python
"""This module contains a object that represents Tests for Botan analytics integration"""

import sys
import unittest
import os

from flaky import flaky

sys.path.append('.')

from telegram.utils.botan import Botan
from tests.base import BaseTest


class MessageMock(object):
    chat_id = None

    def __init__(self, chat_id):
        self.chat_id = chat_id

    def to_json(self):
        return "{}"


@flaky(3, 1)
class BotanTest(BaseTest, unittest.TestCase):
    """This object represents Tests for Botan analytics integration."""

    token = os.environ.get('BOTAN_TOKEN')

    def test_track(self):
        botan = Botan(self.token)
        message = MessageMock(self._chat_id)
        result = botan.track(message, 'named event')
        self.assertTrue(result)

    def test_track_fail(self):
        botan = Botan(self.token)
        botan.url_template = 'https://api.botan.io/traccc?token={token}&uid={uid}&name={name}'
        message = MessageMock(self._chat_id)
        result = botan.track(message, 'named event')
        self.assertFalse(result)

    def test_wrong_message(self):
        botan = Botan(self.token)
        message = MessageMock(self._chat_id)
        message = delattr(message, 'chat_id')
        result = botan.track(message, 'named event')
        self.assertFalse(result)

    def test_wrong_endpoint(self):
        botan = Botan(self.token)
        botan.url_template = 'https://api.botaaaaan.io/traccc?token={token}&uid={uid}&name={name}'
        message = MessageMock(self._chat_id)
        result = botan.track(message, 'named event')
        self.assertFalse(result)


if __name__ == '__main__':
    unittest.main()