class MacTestBuildTreeModel(unittest.TestCase):
"""
Tests how a tree model is built in different path/scan situations, and with different input data - esp. between
the mac and windows implementations.
"""
def setUp(self):
self.t = DirectoryScanner()
self.builder = DirectoryTreeBuilder()
# construct a known directory structure, suitable for testing - it includes files, directories, ACL's, etc
self.contents_path = self.builder.make_dir("BB/TextEdit.app/Contents", 0755)
self.builder.make_dir("BB/TextEdit.app/Contents/Resources", 0755)
self.builder.make_dir("BB/TextEdit.app/Contents/Frameworks", 0755)
self.builder.create_file("BB/TextEdit.app/Contents/Stupid.txt", 0755, 425)
# produce a single scan of the fake file-system entries
self.p = PersistentScanningState("tree-tests.sqlite")
self.t.addPathsForScanning([self.builder.rootDir])
self.initialScan()
self.mergeScan()
def initialScan(self):
for value in self.p.storeFilesystemSnapshot(self.t.performScan()):
pass
self.assertTrue(self.p.numberOfScannedFiles() > 0)
def mergeScan(self):
for value in self.p.storeSecondScan(self.t.performScan()):
pass
self.assertTrue(self.p.numberOfMergedFiles() > 0)
def test_simple_tree_model(self):
# grab a fake scan of something we know the content of, e.g. one directory containing one file
builder = FileSystemTreeModelBuilder(self.p)
builder.buildModel()
topLevel = builder.itemRootedAtPath(self.builder.rootDir)
itemBB = builder.childOfItem(topLevel, "BB")
self.assertTrue(itemBB is not None)
self.assertEqual(itemBB.data().toString(), "BB")
self.assertEqual(itemBB.rowCount(), 1)
# fetch something that doesn't exist - should bring me back None
self.assertEqual(None, builder.childOfItem(itemBB, "Not Here"))
self.assertEqual(None, builder.itemRootedAtPath(os.path.join(self.builder.rootDir, "WOOT")))
itemTextEdit = builder.childOfItem(itemBB, "TextEdit.app")
self.assertTrue(itemTextEdit is not None)
self.assertEqual(itemTextEdit.data().toString(), "TextEdit.app")
self.assertEqual(itemTextEdit.rowCount(), 1)
itemContents = builder.childOfItem(itemTextEdit, "Contents")
self.assertTrue(itemContents is not None)
self.assertEqual(itemContents.data().toString(), "Contents")
self.assertEqual(itemContents.rowCount(), 2)
# go grab both - they should be the frameworks and resources
items = builder.childrenOfItem(itemContents, ["Resources", "Frameworks"])
self.assertEqual(items["Resources"].data().toString(), "Resources")
self.assertEqual(items["Frameworks"].data().toString(), "Frameworks")
