def testRmDirsWithSimplePath(self): self.mox.StubOutWithMock(shutil, 'rmtree') shutil.rmtree('directory') self.mox.ReplayAll() file_util.RmDirs('directory') self.mox.VerifyAll()
def testRmDirsForPermissionDeniedOnParentDirectory(self): self.mox.StubOutWithMock(os, 'rmdir') os.rmdir('path/to').AndRaise( OSError(errno.EACCES, 'Permission denied', 'path/to')) self.mox.StubOutWithMock(shutil, 'rmtree') shutil.rmtree('path/to/directory') self.mox.ReplayAll() file_util.RmDirs('path/to/directory') self.mox.VerifyAll()
def testRmDirsForNotEmptyDirectory(self): self.mox.StubOutWithMock(os, 'rmdir') os.rmdir('path/to').AndRaise( OSError(errno.ENOTEMPTY, 'Directory not empty', 'path/to')) self.mox.StubOutWithMock(shutil, 'rmtree') shutil.rmtree('path/to/directory') self.mox.ReplayAll() file_util.RmDirs('path/to/directory') self.mox.VerifyAll()
def testRmDirsForNonExistingDirectory(self): self.mox.StubOutWithMock(os, 'rmdir') os.rmdir('path/to') os.rmdir('path') self.mox.StubOutWithMock(shutil, 'rmtree') shutil.rmtree('path/to/directory').AndRaise( OSError(errno.ENOENT, "No such file or directory 'path/to/directory'")) self.mox.ReplayAll() file_util.RmDirs('path/to/directory') self.mox.VerifyAll()
def testRmDirs(self): test_sandbox = os.path.join(FLAGS.test_tmpdir, 'test-rm-dirs') test_dir = os.path.join(test_sandbox, 'test', 'dir') os.makedirs(test_sandbox) with open(os.path.join(test_sandbox, 'file'), 'w'): pass os.makedirs(test_dir) with open(os.path.join(test_dir, 'file'), 'w'): pass file_util.RmDirs(test_dir) self.assertFalse(os.path.exists(os.path.join(test_sandbox, 'test'))) self.assertTrue(os.path.exists(os.path.join(test_sandbox, 'file'))) shutil.rmtree(test_sandbox)