mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-28 11:28:38 +00:00
148 lines
4.5 KiB
Python
Executable File
148 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# Copyright (c) 2004 Trent Mick
|
|
|
|
"""Test adding licenses to addlicense_inputs/... with relic.py."""
|
|
|
|
import sys
|
|
import os
|
|
import unittest
|
|
import difflib
|
|
import pprint
|
|
import shutil
|
|
import StringIO
|
|
|
|
import testsupport
|
|
|
|
#---- globals
|
|
|
|
gInputsDir = "addlicense_inputs"
|
|
gOutputsDir = "addlicense_outputs"
|
|
gTmpDir = "addlicense_tmp"
|
|
|
|
|
|
#----- test cases
|
|
|
|
class RelicInputsTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
if not os.path.exists(gTmpDir):
|
|
os.mkdir(gTmpDir)
|
|
|
|
def tearDown(self):
|
|
testsupport.rmtree(gTmpDir)
|
|
|
|
|
|
def _testOneInputFile(self, fname):
|
|
import relic
|
|
_debug = 0 # Set to true to dump status info for each test run.
|
|
|
|
infile = os.path.join(gInputsDir, fname) # input
|
|
outfile = os.path.join(gOutputsDir, fname) # expected output
|
|
tmpfile = os.path.join(gTmpDir, fname) # actual output
|
|
errfile = os.path.join(gOutputsDir, fname+'.error') # expected error
|
|
# An options file is a set of kwargs for the relic.addlicense()
|
|
# method call. One key-value pair per-line like this:
|
|
# key=value
|
|
# Whitespace is stripped off the value.
|
|
optsfile = os.path.join(gInputsDir, fname+'.options') # input options
|
|
|
|
if _debug:
|
|
print
|
|
print "*"*50, "relic '%s'" % fname
|
|
|
|
# Determine input options to use, if any.
|
|
opts = {}
|
|
if os.path.exists(optsfile):
|
|
for line in open(optsfile, 'r').read().splitlines(0):
|
|
name, value = line.split('=', 1)
|
|
value = value.strip()
|
|
try: # allow value to be a type other than string
|
|
value = eval(value)
|
|
except Exception:
|
|
pass
|
|
opts[name] = value
|
|
if _debug:
|
|
print "*"*50, "options"
|
|
pprint.pprint(opts)
|
|
else:
|
|
# If no options were specified, we presume the equivalent of the
|
|
# command-line --defaults option.
|
|
opts = {
|
|
"original_code_is": "mozilla.org Code",
|
|
"initial_copyright_date": "2001",
|
|
"initial_developer": "Netscape Communications Corporation",
|
|
}
|
|
|
|
# Copy the input file to the tmp location where relicensing is done.
|
|
shutil.copy(infile, tmpfile)
|
|
|
|
# Relicense the file, capturing stdout and stderr and any possible
|
|
# error.
|
|
oldStdout = sys.stdout
|
|
oldStderr = sys.stderr
|
|
sys.stdout = StringIO.StringIO()
|
|
sys.stderr = StringIO.StringIO()
|
|
try:
|
|
try:
|
|
relic.addlicense([tmpfile], **opts)
|
|
except relic.RelicError, ex:
|
|
error = ex
|
|
else:
|
|
error = None
|
|
finally:
|
|
stdout = sys.stdout.getvalue()
|
|
stderr = sys.stderr.getvalue()
|
|
sys.stdout = oldStdout
|
|
sys.stderr = oldStderr
|
|
if _debug:
|
|
print "*"*50, "stdout"
|
|
print stdout
|
|
print "*"*50, "stderr"
|
|
print stderr
|
|
print "*"*50, "error"
|
|
print str(error)
|
|
print "*" * 50
|
|
|
|
# Verify that the results are as expected.
|
|
if os.path.exists(outfile) and error:
|
|
self.fail("adding license '%s' raised an error but success was "
|
|
"expected: error='%s'" % (fname, str(error)))
|
|
elif os.path.exists(outfile):
|
|
expected = open(outfile, 'r').readlines()
|
|
actual = open(tmpfile, 'r').readlines()
|
|
if expected != actual:
|
|
diff = list(difflib.ndiff(expected, actual))
|
|
self.fail("%r != %r:\n%s"\
|
|
% (outfile, tmpfile, pprint.pformat(diff)))
|
|
elif os.path.exists(errfile):
|
|
# There is no reference output file. This means that processing
|
|
# this file is expected to fail.
|
|
expectedError = open(errfile, 'r').read()
|
|
actualError = str(error)
|
|
self.failUnlessEqual(actualError.strip(), expectedError.strip())
|
|
else:
|
|
self.fail("No reference ouput file or error file for '%s'." % infile)
|
|
|
|
# Ensure next test file gets a clean relic.
|
|
del sys.modules['relic']
|
|
|
|
|
|
#for fname in ["separated_license_comment_blocks.pl"]:
|
|
for fname in os.listdir(gInputsDir):
|
|
if fname.endswith(".options"): continue # skip input option files
|
|
testFunction = lambda self, fname=fname: _testOneInputFile(self, fname)
|
|
name = 'test_addlicense_'+fname
|
|
setattr(RelicInputsTestCase, name, testFunction)
|
|
|
|
|
|
#---- mainline
|
|
|
|
def suite():
|
|
"""Return a unittest.TestSuite to be used by test.py."""
|
|
return unittest.makeSuite(RelicInputsTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
runner = unittest.TextTestRunner(sys.stdout, verbosity=2)
|
|
result = runner.run(suite())
|
|
|