51 lines
1.9 KiB
Python
51 lines
1.9 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
import datetime
|
|
import db
|
|
import pathlib
|
|
import scanner
|
|
import unittest
|
|
|
|
test_album = pathlib.Path("./data/Test Album")
|
|
test_track = test_album / "01 - Test Track.ogg"
|
|
|
|
class TestScanner(unittest.TestCase):
|
|
def tearDown(self):
|
|
scanner.Queue.tasks_finished()
|
|
|
|
def test_init(self):
|
|
self.assertIsInstance(scanner.Queue, scanner.queue.TaskQueue)
|
|
|
|
def test_update_library(self):
|
|
lib = db.library.Table.find(test_album)
|
|
scanner.update_library(lib)
|
|
self.assertIsInstance(scanner.Queue.tasks[0],
|
|
scanner.task.CheckSchedulerTask)
|
|
self.assertIsInstance(scanner.Queue.tasks[1],
|
|
scanner.task.DirectoryTask)
|
|
|
|
def test_remove_library(self):
|
|
lib = db.library.Table.find(test_album)
|
|
scanner.remove_library(lib)
|
|
self.assertIsInstance(scanner.Queue.tasks[0],
|
|
scanner.task.RemoveLibrarySchedulerTask)
|
|
|
|
def test_commit(self):
|
|
scanner.commit()
|
|
self.assertIsInstance(scanner.Queue.tasks[0], scanner.task.CommitTask)
|
|
|
|
def test_widgets(self):
|
|
lib = db.library.Table.find(test_album)
|
|
|
|
self.assertIsInstance(scanner.ProgressBar(),
|
|
scanner.widgets.ProgressBar)
|
|
self.assertIsInstance(scanner.AddFolderButton(),
|
|
scanner.widgets.AddFolderButton)
|
|
self.assertIsInstance(scanner.UpdateButton(lib),
|
|
scanner.widgets.UpdateButton)
|
|
self.assertIsInstance(scanner.UpdateAllButton(),
|
|
scanner.widgets.UpdateAllButton)
|
|
self.assertIsInstance(scanner.RemoveButton(lib),
|
|
scanner.widgets.RemoveButton)
|
|
self.assertIsInstance(scanner.EnableSwitch(lib),
|
|
scanner.widgets.EnableSwitch)
|