def testCoreTree_BuildTree(dummyGUI, dummyItems): """Test building a project tree from a list of items. """ theProject = NWProject(dummyGUI) theTree = NWTree(theProject) theTree.setSeed(42) assert theTree._handleSeed == 42 # Check that tree is empty (calls NWTree.__bool__) assert not theTree # Check for archive and trash folders assert theTree.trashRoot() is None assert theTree.archiveRoot() is None assert not theTree.isTrashRoot("a000000000003") aHandles = [] for tHandle, pHande, nwItem in dummyItems: aHandles.append(tHandle) assert theTree.append(tHandle, pHande, nwItem) assert theTree._treeChanged # Check that tree is not empty (calls __bool__) assert theTree # Check the number of elements (calls __len__) assert len(theTree) == len(dummyItems) # Check that we have the correct handles assert theTree.handles() == aHandles # Check by iterator (calls __iter__, __next__ and __getitem__) for theItem, theHandle in zip(theTree, aHandles): assert theItem.itemHandle == theHandle # Check that we have the correct archive and trash folders assert theTree.trashRoot() == "a000000000003" assert theTree.archiveRoot() == "a000000000002" assert theTree.isTrashRoot("a000000000003") # Try to add another trash folder itemT = NWItem(theProject) itemT.itemName = "Trash" itemT.itemType = nwItemType.TRASH itemT.itemClass = nwItemClass.TRASH itemT.isExpanded = False assert not theTree.append("1234567890abc", None, itemT) assert len(theTree) == len(dummyItems) # Generate handle automatically itemT = NWItem(theProject) itemT.itemName = "New File" itemT.itemType = nwItemType.FILE itemT.itemClass = nwItemClass.NOVEL itemT.itemLayout = nwItemLayout.SCENE assert theTree.append(None, None, itemT) assert len(theTree) == len(dummyItems) + 1 theList = theTree.handles() assert theList[-1] == "73475cb40a568" # Try to add existing handle assert not theTree.append("73475cb40a568", None, itemT) assert len(theTree) == len(dummyItems) + 1 # Delete a non-existing item del theTree["dummy"] assert len(theTree) == len(dummyItems) + 1 # Delete the last item del theTree["73475cb40a568"] assert len(theTree) == len(dummyItems) assert "73475cb40a568" not in theTree # Delete the Novel, Archive and Trash folders del theTree["a000000000001"] assert len(theTree) == len(dummyItems) - 1 assert "a000000000001" not in theTree del theTree["a000000000002"] assert len(theTree) == len(dummyItems) - 2 assert "a000000000002" not in theTree assert theTree.archiveRoot() is None del theTree["a000000000003"] assert len(theTree) == len(dummyItems) - 3 assert "a000000000003" not in theTree assert theTree.trashRoot() is None
def dummyItems(dummyGUI): """Create a list of dummy items. """ theProject = NWProject(dummyGUI) itemA = NWItem(theProject) itemA.itemName = "Novel" itemA.itemType = nwItemType.ROOT itemA.itemClass = nwItemClass.NOVEL itemA.isExpanded = True itemB = NWItem(theProject) itemB.itemName = "Act One" itemB.itemType = nwItemType.FOLDER itemB.itemClass = nwItemClass.NOVEL itemB.isExpanded = True itemC = NWItem(theProject) itemC.itemName = "Chapter One" itemC.itemType = nwItemType.FILE itemC.itemClass = nwItemClass.NOVEL itemC.itemLayout = nwItemLayout.CHAPTER itemC.charCount = 300 itemC.wordCount = 50 itemC.paraCount = 2 itemD = NWItem(theProject) itemD.itemName = "Scene One" itemD.itemType = nwItemType.FILE itemD.itemClass = nwItemClass.NOVEL itemD.itemLayout = nwItemLayout.SCENE itemD.charCount = 3000 itemD.wordCount = 500 itemD.paraCount = 20 itemE = NWItem(theProject) itemE.itemName = "Outtakes" itemE.itemType = nwItemType.ROOT itemE.itemClass = nwItemClass.ARCHIVE itemE.isExpanded = False itemF = NWItem(theProject) itemF.itemName = "Trash" itemF.itemType = nwItemType.TRASH itemF.itemClass = nwItemClass.TRASH itemF.isExpanded = False itemG = NWItem(theProject) itemG.itemName = "Characters" itemG.itemType = nwItemType.ROOT itemG.itemClass = nwItemClass.CHARACTER itemG.isExpanded = True itemH = NWItem(theProject) itemH.itemName = "Jane Doe" itemH.itemType = nwItemType.FILE itemH.itemClass = nwItemClass.CHARACTER itemH.itemLayout = nwItemLayout.NOTE itemH.charCount = 2000 itemH.wordCount = 400 itemH.paraCount = 16 theItems = [ ("a000000000001", None, itemA), ("b000000000001", "a000000000001", itemB), ("c000000000001", "b000000000001", itemC), ("c000000000002", "b000000000001", itemD), ("a000000000002", None, itemE), ("a000000000003", None, itemF), ("a000000000004", None, itemG), ("b000000000002", "a000000000002", itemH), ] return theItems