58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import shutil
|
|
from colorama import Fore
|
|
import argparse
|
|
import zipTest
|
|
import unzipTest
|
|
|
|
parser = argparse.ArgumentParser(description='Process command-line arguments')
|
|
parser.add_argument('--file', '-f', metavar='path/to/file', type=str, nargs='?', help='Path to file to use for test')
|
|
parser.add_argument('--level', '-l', metavar='#', type=int, nargs='?', help='Compression level')
|
|
parser.add_argument('--no-delete', const=True, default=False, nargs='?', help='Don\'t delete files produced for test')
|
|
parser.add_argument('--test', default='both', nargs='?', help='Which test to run (zip, unzip, both)')
|
|
|
|
args = parser.parse_args()
|
|
|
|
allPassed = True
|
|
|
|
outDir = 'test-outs'
|
|
|
|
# make the test-outs directory
|
|
try:
|
|
os.mkdir(outDir)
|
|
except:
|
|
pass
|
|
|
|
delete = not getattr(args, 'no_delete')
|
|
level = getattr(args, 'level')
|
|
inFile = getattr(args, 'file')
|
|
test = getattr(args, 'test')
|
|
|
|
if test == 'zip' or test == 'both':
|
|
print Fore.CYAN + 'Running zip tests' + Fore.RESET
|
|
# if the user specifies a file, only run that test
|
|
if inFile != None:
|
|
allPassed = zipTest.runTest(inFile, level)
|
|
else:
|
|
allPassed = zipTest.runAll(level)
|
|
|
|
if test == 'unzip' or test == 'both':
|
|
print Fore.CYAN + 'Running unzip tests' + Fore.RESET
|
|
# if the user specifies a file, only run that test
|
|
if inFile != None:
|
|
allPassed = unzipTest.runTest(inFile, level)
|
|
else:
|
|
allPassed = unzipTest.runAll(level)
|
|
|
|
if delete:
|
|
shutil.rmtree(outDir)
|
|
|
|
if allPassed:
|
|
print Fore.GREEN + 'All tests passed!' + Fore.RESET
|
|
else:
|
|
print Fore.RED + 'Automated test failed' + Fore.RESET
|
|
sys.exit(1)
|