def test_add_task(self):
root = TaskTreeItem()
root.addSubTask(name = "Folder", size = 1035, index = 555, selected = False)
self.assertListEqual(list(root.children.keys()), ["Folder"])
folder = root / "Folder"
self.assertEqual(folder.name, "Folder")
self.assertEqual(folder.size, 1035)
self.assertEqual(folder.index, 555)
self.assertEqual(folder.selected, Qt.Unchecked)
def test_add_task_task(self):
root = TaskTreeItem()
root.addSubTask(name = "Folder", size = 1035, index = 555, selected = False)
root.addSubTask(name = "Folder/document", size = 1354, index = 334, selected = False)
self.assertListEqual(list(root.children.keys()), ["Folder"])
# folder size is the total size of children
folder = root / "Folder"
self.assertEqual(folder.size, 1354)
doc = root / "Folder" / "document"
self.assertListEqual(list(doc.children.keys()), [])
self.assertEqual(doc.name, "document")
self.assertEqual(doc.size, 1354)
self.assertEqual(doc.index, 334)
self.assertEqual(doc.selected, Qt.Unchecked)
def test_create_root(self):
tti = TaskTreeItem()
self.assertTrue(tti.isRoot())
self.assertEqual(tti.ancestryTree, "ROOT")
self.assertEqual(tti.name, "UNKNOWN")
self.assertEqual(tti.size, 0)
self.assertEqual(tti.index, -1)
self.assertIsNone(tti.parent)
self.assertEqual(tti.selected, Qt.Unchecked)
self.assertFalse(tti.children)
self.assertEqual(tti.childrenCount(), 0)
self.assertFalse(tti.siblings)
with self.assertRaises(ValueError):
tti.siblingNumber()
def test_selection_size(self):
root = TaskTreeItem()
root.addSubTask(name = "Folder", size = 0, index = 1, selected = False)
root.addSubTask(name = "Dir/video.mp4", size = 5000, index = 2, selected = False)
root.addSubTask(name = "Folder/document", size = 3000, index = 3, selected = True)
self.assertEqual(root.selected, Qt.PartiallyChecked)
self.assertEqual(root.size, 8000)
dir_ = root / "Dir"
self.assertEqual(dir_.selected, Qt.Unchecked)
self.assertEqual(dir_.size, 5000)
video = root / "Dir" / "video.mp4"
self.assertEqual(video.selected, Qt.Unchecked)
self.assertEqual(video.size, 5000)
folder = root / "Folder"
self.assertEqual(folder.selected, Qt.Checked)
self.assertEqual(folder.size, 3000)
doc = root / "Folder" / "document"
self.assertEqual(doc.selected, Qt.Checked)
self.assertEqual(doc.size, 3000)
def test_set_name(self):
tti = TaskTreeItem()
tti.name = "Root"
self.assertEqual(tti.name, "Root")
