class DiskDirectoryTest(DirectoryTest, unittest.TestCase): def setUp(self): self.tempdir = tempfile.mkdtemp() self.dir = DiskDirectory(self.tempdir) super(DiskDirectoryTest, self).setUp() def tearDown(self): super(DiskDirectoryTest, self).tearDown() shutil.rmtree(self.tempdir) def addFile(self, name, mtime, content): path = os.path.join(self.tempdir, name) f = open(path, "wb") f.write(content) f.close() os.utime(path, (mtime, mtime)) def addDirectory(self, name): os.makedirs(os.path.join(self.tempdir, name)) def testGetDiskPath(self): self.assertEquals(os.path.join(self.tempdir, "foo/bar"), self.dir.get_disk_path("foo/bar")) def testGlob(self): self.dir.write("foo.qux", "") self.dir.write("bar.qux", "") self.dir.write("baz.qux", "") self.dir.write("foo.corge", "") self.dir.write("bar.corge", "") self.dir.write("baz.corge", "") self.assertEquals(set(["foo.qux"]), set(self.dir.expand_glob("foo.qux"))) self.assertEquals(set(["foo.qux", "bar.qux", "baz.qux"]), set(self.dir.expand_glob("*.qux"))) self.assertEquals(set(["foo.corge", "bar.corge", "baz.corge"]), set(self.dir.expand_glob("*.corge"))) self.assertEquals(set(["foo.qux", "foo.corge"]), set(self.dir.expand_glob("foo.*"))) self.assertEquals( set([ "foo.qux", "bar.qux", "baz.qux", "foo.corge", "bar.corge", "baz.corge" ]), set(self.dir.expand_glob("*"))) self.assertEquals(set([]), set(self.dir.expand_glob("grault")))
class DiskDirectoryTest(DirectoryTest, unittest.TestCase): def setUp(self): self.tempdir = tempfile.mkdtemp() self.dir = DiskDirectory(self.tempdir) super(DiskDirectoryTest, self).setUp() def tearDown(self): super(DiskDirectoryTest, self).tearDown() shutil.rmtree(self.tempdir) def addFile(self, name, mtime, content): path = os.path.join(self.tempdir, name) f = open(path, "wb") f.write(content) f.close() os.utime(path, (mtime, mtime)) def addDirectory(self, name): os.makedirs(os.path.join(self.tempdir, name)) def testGetDiskPath(self): self.assertEquals(os.path.join(self.tempdir, "foo/bar"), self.dir.get_disk_path("foo/bar")) def testGlob(self): self.dir.write("foo.qux", "") self.dir.write("bar.qux", "") self.dir.write("baz.qux", "") self.dir.write("foo.corge", "") self.dir.write("bar.corge", "") self.dir.write("baz.corge", "") self.assertEquals(set(["foo.qux"]), set(self.dir.expand_glob("foo.qux"))) self.assertEquals(set(["foo.qux", "bar.qux", "baz.qux"]), set(self.dir.expand_glob("*.qux"))) self.assertEquals(set(["foo.corge", "bar.corge", "baz.corge"]), set(self.dir.expand_glob("*.corge"))) self.assertEquals(set(["foo.qux", "foo.corge"]), set(self.dir.expand_glob("foo.*"))) self.assertEquals(set(["foo.qux", "bar.qux", "baz.qux", "foo.corge", "bar.corge", "baz.corge"]), set(self.dir.expand_glob("*"))) self.assertEquals(set([]), set(self.dir.expand_glob("grault")))
def testGlob(self): # Must test with disk directory... tempdir = tempfile.mkdtemp() try: disk_dir = DiskDirectory(tempdir) disk_dir.write("foo.qux", "") disk_dir.write("bar.qux", "") disk_dir.write("baz.qux", "") disk_dir.write("foo.corge", "") disk_dir.write("bar.corge", "") disk_dir.write("baz.corge", "") class PrefixRemovingMapping(MappedDirectory.Mapping): def __init__(self, prefix, real_dir): super(PrefixRemovingMapping, self).__init__() self.__prefix = prefix self.__real_dir = real_dir def map(self, filename): assert filename.startswith(self.__prefix) return (self.__real_dir, filename[len(self.__prefix):]) dir = MappedDirectory(PrefixRemovingMapping("a/", disk_dir)) self.assertEquals(set(["a/foo.qux"]), set(dir.expand_glob("a/foo.qux"))) self.assertEquals(set(["a/foo.qux", "a/bar.qux", "a/baz.qux"]), set(dir.expand_glob("a/*.qux"))) self.assertEquals( set(["a/foo.corge", "a/bar.corge", "a/baz.corge"]), set(dir.expand_glob("a/*.corge"))) self.assertEquals(set(["a/foo.qux", "a/foo.corge"]), set(dir.expand_glob("a/foo.*"))) self.assertEquals( set([ "a/foo.qux", "a/bar.qux", "a/baz.qux", "a/foo.corge", "a/bar.corge", "a/baz.corge" ]), set(dir.expand_glob("a/*"))) self.assertEquals(set([]), set(dir.expand_glob("a/grault"))) finally: shutil.rmtree(tempdir)
def testGlob(self): # Must test with disk directory... tempdir = tempfile.mkdtemp() try: disk_dir = DiskDirectory(tempdir) disk_dir.write("foo.qux", "") disk_dir.write("bar.qux", "") disk_dir.write("baz.qux", "") disk_dir.write("foo.corge", "") disk_dir.write("bar.corge", "") disk_dir.write("baz.corge", "") class PrefixRemovingMapping(MappedDirectory.Mapping): def __init__(self, prefix, real_dir): super(PrefixRemovingMapping, self).__init__() self.__prefix = prefix self.__real_dir = real_dir def map(self, filename): assert filename.startswith(self.__prefix) return (self.__real_dir, filename[len(self.__prefix):]) dir = MappedDirectory(PrefixRemovingMapping("a/", disk_dir)) self.assertEquals(set(["a/foo.qux"]), set(dir.expand_glob("a/foo.qux"))) self.assertEquals(set(["a/foo.qux", "a/bar.qux", "a/baz.qux"]), set(dir.expand_glob("a/*.qux"))) self.assertEquals(set(["a/foo.corge", "a/bar.corge", "a/baz.corge"]), set(dir.expand_glob("a/*.corge"))) self.assertEquals(set(["a/foo.qux", "a/foo.corge"]), set(dir.expand_glob("a/foo.*"))) self.assertEquals(set(["a/foo.qux", "a/bar.qux", "a/baz.qux", "a/foo.corge", "a/bar.corge", "a/baz.corge"]), set(dir.expand_glob("a/*"))) self.assertEquals(set([]), set(dir.expand_glob("a/grault"))) finally: shutil.rmtree(tempdir)