def test_build_staging(self): if os.path.isdir(self.stage_dir): shutil.rmtree(self.stage_dir) b = DNSBuilder(STAGE_DIR=self.stage_dir, PROD_DIR=self.prod_dir, LOCK_FILE=self.lock_file) b.build_staging() # Make sure it made the staging dir self.assertTrue(os.path.isdir(self.stage_dir)) # Ensure if fails if the directory exists self.assertRaises(BuildError, b.build_staging) # There shouldn't be errors because force=True b.build_staging(force=True) self.assertTrue(os.path.isdir(self.stage_dir)) b.clear_staging() self.assertFalse(os.path.isdir(self.stage_dir)) self.assertRaises(BuildError, b.clear_staging) b.clear_staging(force=True) self.assertFalse(os.path.isdir(self.stage_dir))