def create_test_files(): """Create the directories and files used for testing. These are the same directories and files created in AStyleTestI18n_I18n. """ prtdir = testdir prtdir = prtdir.replace('\\', '/') prtdir = libextract.strip_directory_prefix(prtdir) # create the test directory imax = 2 # max number of retries for i in range(0, imax): try: os.mkdir(testdir) except WindowsError: # print a message and retry print "mkdir WindowsError" if os.path.isdir(testdir): break if i == imax - 1: libastyle.system_exit("mkdir WindowsError " + testdir) time.sleep(1) # create Russian sub directories os.mkdir(testdir + subdir1) os.mkdir(testdir + subdir1a) os.mkdir(testdir + subdir1b) os.mkdir(testdir + subdir2) # create the test files text = ( "\nvoid foo()\n" "{\n" "bar();\n" "}\n" ) fileNames = [] fileNames.append(testdir + "/recursive1.cpp") fileNames.append(testdir + subdir1 + "/recursive2.cpp") fileNames.append(testdir + subdir1 + "/recursive3.cpp") fileNames.append(testdir + subdir1a + "/recursive4.cpp") fileNames.append(testdir + subdir1a + "/recursive5.cpp") fileNames.append(testdir + subdir1b + "/recursive6.cpp") fileNames.append(testdir + subdir1b + "/recursive7.cpp") fileNames.append(testdir + subdir2 + "/recursive8.cpp") fileNames.append(testdir + subdir2 + "/recursive9.cpp") for file in fileNames: outfile = open(file, 'w') outfile.write(text) outfile.close() return fileNames
def clean_test_directory(testdir): """Find and remove ALL pre-existing files and directories. """ imax = 2 # max number of retries files = glob.glob(testdir) for file in files: file = file.replace('\\', '/') prtfile = libextract.strip_directory_prefix(file) # removed the directory - this is a problem with Windows only for i in range(0, imax): shutil.rmtree(file, True) if not os.path.isdir(file): break if i == imax - 1: print "Directory not removed: " + file libastyle.system_exit("Remove the directory manually") time.sleep(1) # Windows has to wait for the all directories to be removed for i in range(0, imax): if not os.path.isdir(testdir): break time.sleep(1)