def health_check(validate=True, scope=""): stbt.press("KEY_TV") dismissPopupIfPresent() stbt.press("KEY_INFO") if stbt.wait_until(lambda: infobanner.Infobanner()): print "Box is Up and responds to key presses" try: stbt.wait_for_motion() print "Video playback is present in Live TV" return {"status": True} except stbt.MotionTimeout: print "up key pressed" stbt.press("KEY_CHANNELUP") try: stbt.wait_for_motion() print "Video playback is present in Live TV" return {"status": True} except stbt.MotionTimeout: assert False, "Could not detect Video playback in Live TV" else: # TO DO else: <do one more IR event check> # if stbt.is_screen_black(): stbt.press("KEY_MENU") if stbt.wait_until(lambda: not stbt.is_screen_black()): if stbt.wait_until(lambda: mainhub_page.MainHubPage()): sleep(1) print "Box is Up and responds to key presses" stbt.press("KEY_OK") if stbt.wait_until(lambda: not mainhub_page.MainHubPage()): print "Dismissed Main Hub and tuned to live TV" else: assert False, "Failed to tune to Live TV by OK press on Televisie from Main Hub" try: stbt.wait_for_motion() print "Video playback is present in Live TV" return {"status": True} except stbt.MotionTimeout: assert False, "Could not detect Video playback in Live TV" else: assert False, "Main_hub did not load. Health check failed" else: stbt.press("KEY_POWER") if stbt.wait_until(lambda: not stbt.is_screen_black(), timeout_secs=30): if stbt.wait_until(lambda: mainhub_page.MainHubPage()): sleep(1) print "Box is Up and responds to key presses" stbt.press("KEY_OK") if stbt.wait_until(lambda: not mainhub_page.MainHubPage()): print "Dismissed Main Hub and tuned to live TV" else: assert False, "Failed to tune to Live TV by OK press on Televisie from Main Hub" try: stbt.wait_for_motion() print "Video playback is present in Live TV" return {"status": True} except stbt.MotionTimeout: assert False, "Could not detect Video playback in Live TV" else: assert False, "TV frozen with black screen"
def test_is_screen_black_with_numpy_mask_and_region(): frame = stbt.load_image("videotestsrc-full-frame.png") region = stbt.Region(x=160, y=180, right=320, bottom=240) mask = numpy.zeros((60, 160), dtype=numpy.uint8) mask[:, :80] = 255 assert stbt.is_screen_black(frame, mask, 20, region) mask[:, :] = 255 assert not stbt.is_screen_black(frame, mask, 20, region)
def test_is_screen_black_with_numpy_mask_and_region(): frame = stbt.load_image("videotestsrc-full-frame.png") region = stbt.Region(x=160, y=180, right=320, bottom=240) mask = numpy.zeros((60, 160), dtype=numpy.uint8) mask[:, :80] = 255 assert stbt.is_screen_black(frame, mask, 20, region) mask[:, :] = 255 assert not stbt.is_screen_black(frame, mask, 20, region)
def stbySwitchDetection(maxAttempts=10, sleepTime=1): counter = 0 result = stbt.is_screen_black() while result.black == False and counter <= maxAttempts: time.sleep(sleepTime) result = stbt.is_screen_black() counter += 1 if counter > maxAttempts: return False return True
def test_DVR_on_reboot(): count = 0 while True: if stbt.is_screen_black(): break count += 1 sleep(3) assert count < 10, \ "STB did not reboot within 30 seconds with osdiag RebootNow" count = 0 while True: stbt.press('KEY_POWER') sleep(5) if stbt.match('images/menu/stick_around.png') or stbt.match( 'images/menu/reboot_logo.png'): break sleep(5) count += 1 assert count < 18, \ "Stick Around screen is not shown after reboot within 3 minutes" stbt.wait_for_motion(timeout_secs=300) stbt.press('KEY_RECORD') assert stbt.wait_until(lambda: stbt.match("images/dvr/edit_ep_rec.png") or stbt.match("images/dvr/ch_bar_rec.png")), \ "RECORD press on live TV did not set recording, or prompt for edit recording" mydvr_launch() count = 0 while True: stbt.press('KEY_ENTER') if stbt.match('images/cta/watch.png'): break count += 1 assert count < 16, \ "Could not find recording to play in DVR page" stbt.press('KEY_ENTER') assert stbt.wait_for_motion(timeout_secs=20)
def measure_channel_change_time(key, mask): import google # start timer after key press stbt.press(key) start_time = time.time() assert stbt.wait_until(lambda: stbt.is_screen_black(mask=mask)), \ "Screen never went black" for frame, _ in stbt.frames(timeout_secs=30): if not stbt.is_screen_black(mask=mask, frame=frame): end_time = frame.time break else: assert False, "Channel change didn't complete after 30s" google.GoogleSheet().record_measurement( start_time, "channel_change", {"duration": end_time - start_time})
def check_black_screen(self): """ Checks for a black screen. Asserts if blackscreen is still seen after <timeout> """ stbt.draw_text( 'Action : Wait until black screen is dismissed' ) if not stbt.wait_until( lambda: stbt.is_screen_black(mask=tests.core.bitmap.get_bitmap( self, "BlackScreenMaskCropped.png")), timeout_secs=float(profile["MOTION_IMAGE_TIMEOUT"])): assert True, stbt.draw_text( 'Black Screen : Visible after timeout %s seconds', timeout=float(profile["MOTION_IMAGE_TIMEOUT"]))
def test_is_screen_black_debug(): # So that the output directory name doesn't depend on how many tests # were run before this one. ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access f = stbt.load_image("videotestsrc-full-frame.png") with scoped_curdir(), scoped_debug_level(2): stbt.is_screen_black(f) stbt.is_screen_black(f, mask="videotestsrc-mask-non-black.png") stbt.is_screen_black(f, mask="videotestsrc-mask-no-video.png") stbt.is_screen_black(f, region=stbt.Region(0, 0, 160, 120)) files = subprocess.check_output("find stbt-debug | sort", shell=True) \ .decode("utf-8") assert files == dedent("""\ stbt-debug stbt-debug/00001 stbt-debug/00001/grey.png stbt-debug/00001/index.html stbt-debug/00001/non_black.png stbt-debug/00001/source.png stbt-debug/00002 stbt-debug/00002/grey.png stbt-debug/00002/index.html stbt-debug/00002/mask.png stbt-debug/00002/non_black.png stbt-debug/00002/source.png stbt-debug/00003 stbt-debug/00003/grey.png stbt-debug/00003/index.html stbt-debug/00003/mask.png stbt-debug/00003/non_black.png stbt-debug/00003/source.png stbt-debug/00004 stbt-debug/00004/grey.png stbt-debug/00004/index.html stbt-debug/00004/non_black.png stbt-debug/00004/source.png """) assert_expected("stbt-debug-expected-output/is_screen_black")
def test_is_screen_black_debug(): # So that the output directory name doesn't depend on how many tests # were run before this one. ImageLogger._frame_number = itertools.count(1) # pylint:disable=protected-access f = stbt.load_image("videotestsrc-full-frame.png") with scoped_curdir(), scoped_debug_level(2): stbt.is_screen_black(f) stbt.is_screen_black(f, mask="videotestsrc-mask-non-black.png") stbt.is_screen_black(f, mask="videotestsrc-mask-no-video.png") stbt.is_screen_black(f, region=stbt.Region(0, 0, 160, 120)) files = subprocess.check_output("find stbt-debug | sort", shell=True) assert files == dedent("""\ stbt-debug stbt-debug/00001 stbt-debug/00001/grey.png stbt-debug/00001/index.html stbt-debug/00001/non_black.png stbt-debug/00001/source.png stbt-debug/00002 stbt-debug/00002/grey.png stbt-debug/00002/index.html stbt-debug/00002/mask.png stbt-debug/00002/non_black.png stbt-debug/00002/source.png stbt-debug/00003 stbt-debug/00003/grey.png stbt-debug/00003/index.html stbt-debug/00003/mask.png stbt-debug/00003/non_black.png stbt-debug/00003/source.png stbt-debug/00004 stbt-debug/00004/grey.png stbt-debug/00004/index.html stbt-debug/00004/non_black.png stbt-debug/00004/source.png """) assert_expected("stbt-debug-expected-output/is_screen_black")
def test_is_screen_black_with_numpy_mask(): frame = stbt.load_image("videotestsrc-full-frame.png") mask = numpy.zeros((240, 320), dtype=numpy.uint8) mask[180:240, 160:213] = 255 assert stbt.is_screen_black(frame, mask)
def test_is_screen_black_result(): frame = stbt.load_image("almost-black.png") result = stbt.is_screen_black(frame) assert result assert numpy.all(result.frame == frame) assert result.black is True
def test_is_screen_black(frame, mask, threshold, region, expected): frame = stbt.load_image(frame) assert expected == bool( stbt.is_screen_black(frame, mask, threshold, region))
def testAutodiag(self): scanedSN #zeskanowany SN dekodera self.sprawdzSN(scannedSN) stbt.press("KEY_OK", Test.timeBeforePress) for i in range(10): result = stbt.match_text("OK", None, stbt.Region(x=665, y=293 + 16 * i, width=30, height=16)) if result == False: self.testsResults["hds_autodiag_hardware_test"] = False return stbt.prsss("KEY_DOWN", Test.timeBeforePress) for i in range(3): result = stbt.match_text("OK", None, stbt.Region(x=665, y=293 + 16 * i, width=30, height=16)) if result == False: self.testsResults["hds_autodiag_hardware_test"] = False return stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press_until_match("KEY_DOWN", "./images/t1/hds_hard_disk.png", self.timeBeforePress, 5, region = stbt.Region(x=464, y=357, width=117, height=13)) stbt.press("KEY_OK", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) result = stbt.match_text("PASS") if result == False: self.testsResults["hdd"] = False return stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press("KEY_DOWN", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) while True: time.sleep(2) if stbt.match_text("PASS", region = stbt.Region(x=663, y=290, width=70, height=20)): self.testsResults["hdd"] = True break elif stbt.match_text("FAIL", region = stbt.Region(x=663, y=290, width=70, height=20)): self.testsResults["hdd"] = False return stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press("KEY_DOWN", Test.timeBeforePress) stbt.press("KEY_DOWN", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) self.typeAutodiagPassword() while True: time.sleep(2) if stbt.match_text("PASS", region = stbt.Region(x=663, y=290, width=70, height=20)): self.testsResults["hdd"] = True break elif stbt.match_text("FAIL", region = stbt.Region(x=663, y=290, width=70, height=20)): self.testsResults["hdd"] = False return stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press_until_match("KEY_DOWN", "./images/t1/hds_autodiag_factory_settings.png", Test.timeBeforePress, 5, region = stbt.Region(x=464, y=371, width=212, height=21)) stbt.press("KEY_OK", Test.timeBeforePress) self.typeAutodiagPassword() stbt.press("KEY_OK", Test.timeBeforePress) stbt.press("KEY_DOWN", Test.timeBeforePress) stbt.press("KEY_DOWN", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) while True: time.sleep(2) if stbt.match_text("DONE", region = stbt.Region(x=637, y=292, width=78, height=14))): while True: time.sleep(1) if stbt.match_text("DONE", region = stbt.Region(x=636, y=309, width=70, height=14): self.testsResults["hds_factory_settings"] = True break elif stbt.match_text("FAIL", region = stbt.Region(x=636, y=309, width=70, height=14): self.testsResults["hds_factory_settings"] = False return break elif stbt.match_text("FAIL", region = stbt.Region(x=637, y=292, width=78, height=14)): self.testsResults["hds_factory_settings"] = False return stbt.press("KEY_MENU", Test.timeBeforePress) stbt.press("KEY_MENU", Test.timeBeforePress) def wejdzWUstawienia(self): stbt.press("KEY_MENU", self.timeBeforePress) self.myPressUntilMatchText("USTAWIENIA", "KEY_RIGHT", stbt.Region(x=550, y=605, width=180, height=41), 10) stbt.press("KEY_OK", self.timeBeforePress) def typeAutodiagPassword(self): stbt.press("KEY_5", Test.timeBeforePress) stbt.press("KEY_5", Test.timeBeforePress) stbt.press("KEY_5", Test.timeBeforePress) def sprawdzRachunki(self): result = stbt.match_text("BRAK", None, stbt.Region(x=845, y=235, width=72, height=26)) if result: self.testsResults["bills"] = True else: self.testsResults["bills"] = False def sprawdzWiadomosci(self): result = stbt.ocr(region = stbt.Region(x=101, y=135, width=1075, height=32)) if result == "": self.testsResults["messages"] = True else: self.testsResults["messages"] = False def sprawdzHDD(self): today = stbt.ocr(region = stbt.Region(x=1026, y=35, width=86, height=33)) recordingDate = ocr(region = stbt.Region(x=673, y=543, width=62, height=22)) if recordingDate == today: stbt.press("KEY_PLAY", Test.timeBeforePress) time.sleep(10) stbt.press("KEY_PAUSE", Test.timeBeforePress) try: stbt.wait_for_motion(timeout_sec = 3, region = stbt.Region(x=10, y=11, width=1260, height=540)) except MotionTimeout: stbt.press("KEY_REV", Test.timeBeforePress) if stbt.match_text("x4", region = stbt.Region(x=186, y=588, width=30, height=21)): stbt.press("KEY_FWD", Test.timeBeforePress) stbt.press("KEY_FWD", Test.timeBeforePress) if stbt.match_text("x8", region = stbt.Region(x=186, y=588, width=30, height=21)): selft.testsResults["hdd"] = True else: self.testsResults["hdd"] = False else: self.testsResults["hdd"] = False else: self.testsResults["hdd"] = False stbt.press("KEY_STOP", Test.timeBeforePress) stbt.press("KEY_3", Test.timeBeforePress) stbt.press("KEY_OK", Test.timeBeforePress) else: self.testsResults["hdd"] = False def sprawdzSN(self, scannedSN): SN = stbt.ocr(region = stbt.Region(x=600, y=251, width=164, height=19)) print("Odczytany SN: ", SN) if SN != scannedSN: self.testsResults["sn"] = False else: self.testsResults["sn"] = True def sprawdzCzytnikKart(self): print("Wloz karte do czytnika.") result = myWaitWhileMatchText("BRAK KARTY", stbt.Region(x=555, y=127, width=165, height=33)) if result: result = stbt.match("./images/t1/hds_smart_card_error.png", region = stbt.Region(x=555, y=127, width=165, height=33)) if result: self.testsResults["card_reader"] = False else: self.testsResults["card_reader"] = True else: self.testsResults["card_reader"] = False def sprawdzMaxIndex(self): maxIndex = stbt.ocr(region = stbt.Region(x=669, y=418, width=56, height=35)) #stbt.ocr(None, stbt.Region(x=669, y=418, width=56, height=35), stbt.OcrMode.PAGE_SEGMENTATION_WITHOUT_OSD) if maxIndex == "0": self.testsResults["max_index"] = True else: self.testsResults["max_index"] = False def sprawdzSiec(self): ipTestResult = stbt.match_text("172.16", None, stbt.Region(x=655, y=233, width=56, height=24)) if ipTestResult == False: self.testsResults["ethernet"] = False else: self.testsResults["ethernet"] = True def sprawdzWersjeOprogramowania(self): softVersion = stbt.ocr(region = stbt.Region(x=811, y=323, width=47, height=21)) #stbt.ocr(None, stbt.Region(x=811, y=323, width=47, height=21), stbt.OcrMode.PAGE_SEGMENTATION_WITHOUT_OSD) if softVersion == currentUpdateVersions[0]uv or softVersion == currentUpdateVersions[1]: self.testsResults["soft_version"] = True else: self.testsResults["soft_version"] = False def sprawdzSileIJakosSygnalu(self): self.testsResults["signal_level"] = stbt.match("./images/t1/hds_signal_level.png", region = stbt.Region(x=743, y=324, width=111, height=19)) def sprawdzUSB(self, port): print("Wloz Pendrive do gniazda USB umieszczonego z boku.") input = input("Czy dioda Pendrive swieci?\n1) Tak\n2) Nie") while input != "1" and input != "2": input = input("Czy dioda Pendrive swieci?\n1) Tak\n2) Nie") if input == "1": self.testsResults[port] = True else: self.testsResults[port] = False def sprawdzTimeshifting(self): stbt.press("KEY_PAUSE", Test.timeBeforePress) try: stbt.wait_for_motion(timeout_sec = 3, region = stbt.Region(x=10, y=70, width=1260, height=480)) except MotionTimeout: stbt.press("KEY_STOP", Test.timeBeforePress) try: stbt.wait_for_motion() except MotionTimeout: self.testsResults["timeshifting"] = False else: self.testsResults["timeshifting"] = True else: self.testsResults["timeshifting"] = False def sprawdzWyswietlacz(self): self.runChannel("888") input = input("Czy wyswietlacz swieci prawidlowo?\n1) Tak\n2) Nie") while input != "1" and input != "2": input = input("Czy wyswietlacz swieci prawidlowo?\n1) Tak\n2) Nie") if input == "1": self.testsResults["display"] = True else self.testsResults["dispaly"] = False def sprawdzTVNaziemna(self): self.runChannel("998") input = input("Czy obraz na ekranie wyswietlany jest prawidlowo?\n1) Tak\n2) Nie") while input != "1" and input != "2": input = input("Czy obraz na ekranie wyswietlany jest prawidlowo?\n1) Tak\n2) Nie") if input == "1": self.testsResults["terrestrial_siganl_quality"] = True else: self.runChannel("994") input = input("Czy obraz na ekranie wyswietlany jest prawidlowo?\n1) Tak\n2) Nie") while input != "1" and input != "2": input = input("Czy obraz na ekranie wyswietlany jest prawidlowo?\n1) Tak\n2) Nie") if input == "1": self.testsResults["terrestrial_siganl_quality"] = True else: self.testsResults["terrestrial_siganl_quality"] = False def sprawdzPrzyciski(self, maxAttempts = 5): result = {} counter = 0 print("Wcisnij kalwisz P-") detection = channelSwitchDetection(self.mainTestChannels[0]) while detection[0] != 1 and counter < maxAttempts: detection = channelSwitchDetection(self.mainTestChannels[0]) counter += 1 if counter == maxAttempts: result["P-"] = False else: result["P-"] = True time.sleep(5) print("WciĆnij klawisz 'P+'") counter = 0 detection = channelSwitchDetection(detection[1]) while detection[0] != 2 and counter < maxAttempts: detection = channelSwitchDetection(detection[1]) counter += 1 if counter == maxAttempts: result["P+"] = False else: result["P+"] = True print("Wcisnij klawisz 'V+'") result["V+"] = self.volumeUpSwitchDetection() print("Wcisnij klawisz 'V-'") result["V-"] = self.volumeDownSwitchDetection() time.sleep(5) print("Wcisnij klawisz 'St-By'") counter = 0 detection = stbySwitchDetection() while detection == False and counter < maxAttempts: detection = stbySwitchDetection() counter += 1 if counter == maxAttempts: result["St-By"] = False else: result["St-By"] = True self.testsResults["switches"] = result def channelSwitchDetection(channelToCompare): result = [0, 0] try: stbt.wait_for_match("./images/t1/hds_up_arrow.png", region = stbt.Region(x=255, y=539, width=39, height=31)) # oczekiwanie na wcisnienie przycisku except stbt.MatchTimeout: result[0] = -1 result[1] = channelToCompare return result channelNumber = stbt.ocr(None, stbt.Region(90, 592, width=45, height=41), stbt.OcrMode.PAGE_SEGMENTATION_WITHOUT_OSD) cn = int(channelNumber) result[1] = cn if cn < channelToCompare: result[0] = 1 elif cn > channelToCompare: result[0] = 2 return result def stbySwitchDetection(maxAttempts = 5, sleepTime = 1): counter = 0 result = stbt.is_screen_black() while result.black == False and counter <= maxAttempts: time.sleep(sleepTime) result = stbt.is_screen_black() counter += 1 if counter > maxAttempts: return False return True def volumeUpSwitchDetection(self): try: stbt.wait_for_match("./images/t1/hds_max_volume_belt.png", region = stbt.Region(x=178, y=94, width=118, height=16)) except stbt.MatchTimeout: return False return True def volumeDownSwitchDetection(self): try: stbt.wait_for_match("./hds_part_volume_belt.png", region = stbt.Region(x=178, y=94, width=64, height=16)) except stbt.MatchTimeout: return False else if stbt.match("./images/t1/hds_max_volume_belt.png", region = stbt.Region(x=178, y=94, width=118, height=16)): return False return True def lodeConfiguration(self, path): domeTree = minidom.parse(path) parameters = domeTree.firstChild # <parameters> paramChilds = parameters.childNodes soft = paramChilds[1] # <soft> softChilds = soft.getElementsByTagName("version") self.currentUpdateVersions.append(int(soft.childNodes[1].firstChild.data)) # <version> self.currentUpdateVersions.append(int(soft.childNodes[3].firstChild.data)) # <version> testChannels = paramChilds[7] # <test_channels> channels = testChannels.childNodes self.mainTestChannels.append(int(testChannels.childNodes[1].firstChild.data)) # <channel> self.mainTestChannels.append(int(testChannels.childNodes[3].firstChild.data)) # <channel> self.mainTestChannels.append(int(testChannels.childNodes[5].firstChild.data)) # <channel> self.mainTestChannels.append(int(testChannels.childNodes[7].firstChild.data)) # <channel> self.alternativeTestChannels.append(int(testChannels.childNodes[9].firstChild.data)) # <channel> self.alternativeTestChannels.append(int(testChannels.childNodes[11].firstChild.data)) # <channel> self.alternativeTestChannels.append(int(testChannels.childNodes[13].firstChild.data)) # <channel> self.alternativeTestChannels.append(int(testChannels.childNodes[15].firstChild.data)) # <channel> self.mainTerrestrialTestChannel = testChannels.childNodes[17].firstChild.data # <channel> self.alternativeTerrestrialTestChannel = testChannels.childNodes[19].firstChild.data # <channel>
def test_is_screen_black(frame, mask, threshold, region, expected): frame = stbt.load_image(frame) assert expected == bool( stbt.is_screen_black(frame, mask, threshold, region))
def test_is_screen_black_with_numpy_mask(): frame = stbt.load_image("videotestsrc-full-frame.png") mask = numpy.zeros((240, 320), dtype=numpy.uint8) mask[180:240, 160:213] = 255 assert stbt.is_screen_black(frame, mask)
def test_is_screen_black_result(): frame = stbt.load_image("almost-black.png") result = stbt.is_screen_black(frame) assert result assert numpy.all(result.frame == frame) assert result.black is True