def organize(self):
fields = DocumentFields()
fields.setField("title", "document about internet")
fields.setField("year", "2014")
fields.setField("authors", "John Bruf, Tommy Bamo")
fields.setField("keywords", "Computer science, Photography")
fields.setField("journal", "Journal of Computer Science")
dir = Directory("document_about_internet")
dir.removeDir()
d = DocumentFolder()
d.setDocFields(fields)
d.create()
assert(not d.hasVideoFolder())
assert(not d.hasImageFolder())
assert(not d.hasNotesFolder())
assert(not d.hasDescFolder())
d.organize()
assert(d.hasVideoFolder())
assert(d.hasImageFolder())
assert(d.hasNotesFolder())
assert(d.hasDescFolder())
notesPath = d.getNotesFilePath()
print notesPath
assert(File(notesPath).exists())
d.delete()
def test_create(self):
a = DocumentFields()
a.setField("title", "document")
d = DocumentFolder()
d.setDocFields(a)
# Creation in local folder
for title in self.titles:
a.setField("title", title)
d.create()
self.assertTrue(d.getDir().exists())
d.delete()
self.assertFalse(d.getDir().exists())
dir = Directory("documents")
dir.createDir()
# Creaton in another folder
for title in self.titles:
a.setField("title", title)
d.create(dir.getPath().getAbsolutePath())
self.assertTrue(d.getDir().exists())
d.delete()
self.assertFalse(d.getDir().exists())
dir.removeDir()
def create(self):
a = DocumentFields()
a.setField("title", "document")
d = DocumentFolder()
d.setDocFields(a)
for title in self.titles:
a.setField("title", title)
d.create()
assert(d.getDir().exists())
d.delete()
assert(not d.getDir().exists())
dir = Directory("documents")
dir.createDir()
for title in self.titles:
a.setField("title", title)
d.create(dir.getPath().getAbsolutePath())
assert(d.getDir().exists())
d.delete()
assert(not d.getDir().exists())
dir.removeDir()
def remove(self):
d = Directory("dir")
d.removeFile()
assert(d.exists())
d.removeDir()
assert(not d.exists())
def __init__(self):
Test.__init__(self, "Directory")
d = Directory("dir")
d.removeDir()
self.addTestFunction(self.create)
self.addTestFunction(self.remove)
self.addTestFunction(self.createTree)
self.addTestFunction(self.content)
self.addTestFunction(self.containsFolder)
def test_remove(self):
d = Directory("dir")
d.createDir()
d.removeFile()
self.assertTrue(d.exists())
d.removeDir()
self.assertFalse(d.exists())
def test_containsExtension(self):
d = Directory("dir")
f1 = File("dir/text.txt")
f2 = File("dir/image.png")
d.createDir()
f1.createFile()
f2.createFile()
self.assertTrue(d.containsFileExtension(".txt"), "Directory contains .txt")
self.assertTrue(d.containsFileExtension(".png"), "Directory contains .png")
self.assertFalse(d.containsFileExtension(".pdf"), "Directory does not contain .pdf")
d.removeDir()
def containsFolder(self):
d = Directory("dir")
d1 = Directory("dir/a/b")
d2 = Directory("dir/c")
d3 = Directory("dir/a")
d1.createTree()
d2.createTree()
assert(d.containsFolder("a"))
assert(d.containsFolder("c"))
assert(not d.containsFolder("b"))
assert(d3.containsFolder("b"))
assert(not d3.containsFolder("a"))
d.removeDir()
def test_containsFolder(self):
d = Directory("dir")
d1 = Directory("dir/a/b")
d2 = Directory("dir/c")
d3 = Directory("dir/a")
d1.createTree()
d2.createTree()
self.assertTrue(d.containsFolder("a"))
self.assertTrue(d.containsFolder("c"))
self.assertFalse(d.containsFolder("b"))
self.assertTrue(d3.containsFolder("b"))
self.assertFalse(d3.containsFolder("a"))
d.removeDir()
def createTree(self):
d = Directory("dir/fg/a")
assert(not d.exists())
d.createDir()
assert(not d.exists())
d.createTree()
assert(d.exists())
d.removeFile()
assert(d.exists())
d.removeDir()
assert(not d.exists())
d = Directory("dir")
assert(d.exists())
d.removeDir()
assert(not d.exists())
def test_createTree(self):
d = Directory("dir/fg/a")
self.assertFalse(d.exists())
with self.assertRaises(FileNotFoundError):
d.createDir()
self.assertFalse(d.exists())
d.createTree()
self.assertTrue(d.exists())
d.removeFile()
self.assertTrue(d.exists())
d.removeDir()
self.assertFalse(d.exists())
d = Directory("dir")
self.assertTrue(d.exists())
d.removeDir()
self.assertFalse(d.exists())
def content(self):
d = Directory("dir")
d1 = Directory("dir/a/b")
d2 = Directory("dir/c")
d3 = Directory("dir/d/e")
f1 = File("dir/g.txt")
f2 = File("dir/e.txt")
d1.createTree()
d2.createTree()
d3.createTree()
f1.createFile()
f2.createFile()
l = d.getFilesList()
assert (l == {"dir/e.txt", "dir/g.txt"})
l = d.getDirsList()
assert (l == {"dir/a", "dir/c", "dir/d"})
l = d.getContentList()
assert (l == {"dir/e.txt", "dir/g.txt", "dir/a", "dir/c", "dir/d"})
d.removeDir()
