+ def test_clean_dir(self):
+ self._setup_files()
+ try:
+ self.assertEqual(len(os.listdir(self.basedir)), 3)
+ util.clean_dir(self.basedir)
+ self.assertTrue(os.path.exists(self.basedir))
+ self.assertEqual(len(os.listdir(self.basedir)), 0)
+ finally:
+ if os.path.exists(self.basedir):
+ shutil.rmtree(self.basedir)