improved tests

This commit is contained in:
aceisace 2022-09-08 23:23:05 +02:00
parent a431ffb683
commit 1c68895d94

View File

@ -1,14 +1,15 @@
#!python3 #!python3
import os
import unittest import unittest
from inkycal.modules import Todoist as Module from inkycal.modules import Todoist as Module
from helper_functions import * from helper_functions import *
environment = get_environment() environment = get_environment()
# Set to True to preview images. Only works on Raspberry Pi OS with Desktop # Set to True to preview images. Only works on Raspberry Pi OS with Desktop
use_preview = False use_preview = False
api_key = "" api_key = os.environ["TODOIST_API_KEY"] or ""
tests = [ tests = [
{ {
@ -22,6 +23,7 @@ tests = [
}, },
] ]
class module_test(unittest.TestCase): class module_test(unittest.TestCase):
def test_get_config(self): def test_get_config(self):
@ -36,13 +38,13 @@ class module_test(unittest.TestCase):
module = Module(test) module = Module(test)
im_black, im_colour = module.generate_image() im_black, im_colour = module.generate_image()
print('OK') print('OK')
if use_preview == True and environment == 'Raspberry': if use_preview and environment == 'Raspberry':
preview(merge(im_black, im_colour)) preview(merge(im_black, im_colour))
else: else:
print('No api key given, omitting test') print('No api key given, omitting test')
if __name__ == '__main__':
if __name__ == '__main__':
logger = logging.getLogger() logger = logging.getLogger()
logger.level = logging.DEBUG logger.level = logging.DEBUG
logger.addHandler(logging.StreamHandler(sys.stdout)) logger.addHandler(logging.StreamHandler(sys.stdout))