def test_loading_book_mark_good(self):
model = Model()
self.assertEqual(model.getStatus(), State.NO_OPEN_FILE)
model.loadBookmark(os.getcwd() + "\\bookmarks.html")
self.assertEqual(model.getStatus(), State.FILE_OPENED)
self.assertEqual(model.numberOfVideos(), 3)
def test_benchmark_model_bookmark_regex(self):
#just curious how long this takes
model = Model()
start = time()
clock()
model.loadBookmark(os.getcwd() + "\\test-files\\bookmark_1k_bomb.html")
seconds = time() - start
print("Loading Bookmark Files Took:", clock(), "Cycles. Seconds:", seconds)
def test_loading_book_mark_bad(self):
model = Model()
#test to make sure that if the file does not exist, the state
#is still NO_OPEN_FILE. We don't check if its EMPTY_FILE since
#that is just a flag state for the UI to display a error dialog box
#and instantly changes back to NO_OPEN_FILE after the UI has been informed.
self.assertEqual(model.getStatus(), State.NO_OPEN_FILE)
model.loadBookmark("")
self.assertEqual(model.getStatus(), State.NO_OPEN_FILE)
self.assertEqual(model.numberOfVideos(), 0)
def test_video_skip(self):
model = Model()
model.loadBookmark(os.getcwd() + "\\bookmarks.html")
#test skipping
self.assertEqual(model.videoStatus(0), videoState.QUEUED)
model.removeItemFromList(0)
self.assertEqual(model.videoStatus(0), videoState.SKIPPED)
model.queueItemFromList(0)
self.assertEqual(model.videoStatus(0), videoState.QUEUED)
#test cancelling
model.changeVideoStatus(0, videoState.DOWNLOADING)
model.removeItemFromList(0)
self.assertEqual(model.videoStatus(0), videoState.CANCELLED)
#queueing this should not work, should stay canceled
model.queueItemFromList(0)
self.assertEqual(model.videoStatus(0), videoState.CANCELLED)
def test_using_youtube_title_if_non_bookmark(self):
model = Model()
model.loadBookmark(os.getcwd() + "\\youtube-videos.txt")
self.assertEqual(model.output_title_format, titleFormat.USE_YOUTUBE_TITLE)
