from test.helpers.syscmd import SysCommand
import unittest
import time
from test.helpers.cv_display_test import pattern_detect

class Qscreen(unittest.TestCase):

    def __init__(self, testname, testfunc, display):
        super(Qscreen, self).__init__(testfunc)
        self.__display = display
        self._testMethodDoc = testname

    def execute(self):
        str_cmd = "fbi -T 1 --noverbose -d /dev/{} test/files/test_pattern.png".format(self.__display)
        display_image = SysCommand("display_image", str_cmd)
        if display_image.execute() == -1:
            test_screen = pattern_detect(1)
            if not test_screen=="0":
                self.fail("failed: {}".format(test_screen))
            str_cmd= "fbi -T 1 /home/root/result_hdmi_img.jpg -d /dev/fb0 --noverbose -a"
            show_img = SysCommand("show-image", str_cmd)
            show_img.execute()
        else:
            self.fail("failed: could not display the image")
        try:
            str_cmd= "fbi -T 1 /home/root/result_hdmi_img.jpg -d /dev/fb0 --noverbose -a"
            show_img = SysCommand("show-image", str_cmd)
            show_img.execute()
        except ValueError:
            print("COULD NOT DISPLAY IMAGE")