Inkycal/inkycal/tests/inkycal_todoist_test.py

#!python3
"""
inkycal_todoist unittest
"""
import logging
import sys
import unittest
from inkycal.modules import Todoist as Module

from inkycal.modules.inky_image import Inkyimage
from inkycal.tests import Config
preview = Inkyimage.preview
merge = Inkyimage.merge

api_key = Config.TODOIST_API_KEY

tests = [
    {
        "name": "Todoist",
        "config": {
            "size": [400, 1000],
            "api_key": api_key,
            "project_filter": None,
            "padding_x": 10,
            "padding_y": 10,
            "fontsize": 12,
            "language": "en"
        }
    },
]


class module_test(unittest.TestCase):

    def test_get_config(self):
        print('getting data for web-ui...', end="")
        Module.get_config()
        print('OK')

    def test_generate_image(self):
        if api_key:
            for test in tests:
                print(f'test {tests.index(test) + 1} generating image..')
                module = Module(test)
                im_black, im_colour = module.generate_image()
                print('OK')
                if Config.USE_PREVIEW:
                    preview(merge(im_black, im_colour))
                merge(im_black, im_colour).show()
        else:
            print('No api key given, omitting test')


if __name__ == '__main__':
    logger = logging.getLogger()
    logger.level = logging.DEBUG
    logger.addHandler(logging.StreamHandler(sys.stdout))

    unittest.main()