зеркало из https://github.com/mozilla/pjs.git
295 строки
10 KiB
Python
295 строки
10 KiB
Python
|
import pymake.data, pymake.parser, pymake.parserdata, pymake.functions
|
||
|
|
import unittest
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from cStringIO import StringIO
|
||
|
|
|
||
|
|
def multitest(cls):
|
||
|
|
for name in cls.testdata.iterkeys():
|
||
|
|
def m(self, name=name):
|
||
|
|
return self.runSingle(*self.testdata[name])
|
||
|
|
|
||
|
|
setattr(cls, 'test_%s' % name, m)
|
||
|
|
return cls
|
||
|
|
|
||
|
|
class TestBase(unittest.TestCase):
|
||
|
|
def assertEqual(self, a, b, msg=""):
|
||
|
|
"""Actually print the values which weren't equal, if things don't work out!"""
|
||
|
|
unittest.TestCase.assertEqual(self, a, b, "%s got %r expected %r" % (msg, a, b))
|
||
|
|
|
||
|
|
class DataTest(TestBase):
|
||
|
|
testdata = (
|
||
|
|
((("He\tllo", "f", 1, 0),),
|
||
|
|
((0, "f", 1, 0), (2, "f", 1, 2), (3, "f", 1, 4))),
|
||
|
|
((("line1 ", "f", 1, 4), ("l\tine2", "f", 2, 11)),
|
||
|
|
((0, "f", 1, 4), (5, "f", 1, 9), (6, "f", 2, 11), (7, "f", 2, 12), (8, "f", 2, 16))),
|
||
|
|
)
|
||
|
|
|
||
|
|
def runTest(self):
|
||
|
|
for datas, results in self.testdata:
|
||
|
|
d = pymake.parser.Data()
|
||
|
|
for line, file, lineno, col in datas:
|
||
|
|
d.append(line, pymake.parserdata.Location(file, lineno, col))
|
||
|
|
for pos, file, lineno, col in results:
|
||
|
|
loc = d.getloc(pos)
|
||
|
|
self.assertEqual(loc.path, file, "data file")
|
||
|
|
self.assertEqual(loc.line, lineno, "data line")
|
||
|
|
self.assertEqual(loc.column, col, "data %r col, got %i expected %i" % (d.data, loc.column, col))
|
||
|
|
|
||
|
|
class TokenTest(TestBase):
|
||
|
|
testdata = {
|
||
|
|
'wsmatch': (' ifdef FOO', 2, ('ifdef', 'else'), True, 'ifdef', 8),
|
||
|
|
'wsnomatch': (' unexpected FOO', 2, ('ifdef', 'else'), True, None, 2),
|
||
|
|
'wsnows': (' ifdefFOO', 2, ('ifdef', 'else'), True, None, 2),
|
||
|
|
'paren': (' "hello"', 1, ('(', "'", '"'), False, '"', 2),
|
||
|
|
}
|
||
|
|
|
||
|
|
def runSingle(self, s, start, tlist, needws, etoken, eoffset):
|
||
|
|
d = pymake.parser.Data.fromstring(s, None)
|
||
|
|
tl = pymake.parser.TokenList.get(tlist)
|
||
|
|
atoken, aoffset = d.findtoken(start, tl, needws)
|
||
|
|
self.assertEqual(atoken, etoken)
|
||
|
|
self.assertEqual(aoffset, eoffset)
|
||
|
|
multitest(TokenTest)
|
||
|
|
|
||
|
|
class IterTest(TestBase):
|
||
|
|
testdata = {
|
||
|
|
'plaindata': (
|
||
|
|
pymake.parser.iterdata,
|
||
|
|
"plaindata # test\n",
|
||
|
|
"plaindata # test\n"
|
||
|
|
),
|
||
|
|
'makecomment': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = val # comment",
|
||
|
|
"VAR = val "
|
||
|
|
),
|
||
|
|
'makeescapedcomment': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = val \# escaped hash\n",
|
||
|
|
"VAR = val # escaped hash"
|
||
|
|
),
|
||
|
|
'makeescapedslash': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = val\\\\\n",
|
||
|
|
"VAR = val\\\\",
|
||
|
|
),
|
||
|
|
'makecontinuation': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = VAL \\\n continuation # comment \\\n continuation",
|
||
|
|
"VAR = VAL continuation "
|
||
|
|
),
|
||
|
|
'makecontinuation2': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = VAL \\ \\\n continuation",
|
||
|
|
"VAR = VAL \\ continuation"
|
||
|
|
),
|
||
|
|
'makeawful': (
|
||
|
|
pymake.parser.itermakefilechars,
|
||
|
|
"VAR = VAL \\\\# comment\n",
|
||
|
|
"VAR = VAL \\"
|
||
|
|
),
|
||
|
|
'command': (
|
||
|
|
pymake.parser.itercommandchars,
|
||
|
|
"echo boo # comment\n",
|
||
|
|
"echo boo # comment",
|
||
|
|
),
|
||
|
|
'commandcomment': (
|
||
|
|
pymake.parser.itercommandchars,
|
||
|
|
"echo boo \# comment\n",
|
||
|
|
"echo boo \# comment",
|
||
|
|
),
|
||
|
|
'commandcontinue': (
|
||
|
|
pymake.parser.itercommandchars,
|
||
|
|
"echo boo # \\\n\t command 2\n",
|
||
|
|
"echo boo # \\\n command 2"
|
||
|
|
),
|
||
|
|
'define': (
|
||
|
|
pymake.parser.iterdefinechars,
|
||
|
|
"endef",
|
||
|
|
""
|
||
|
|
),
|
||
|
|
'definenesting': (
|
||
|
|
pymake.parser.iterdefinechars,
|
||
|
|
"""define BAR # comment
|
||
|
|
random text
|
||
|
|
endef not what you think!
|
||
|
|
endef # comment is ok\n""",
|
||
|
|
"""define BAR # comment
|
||
|
|
random text
|
||
|
|
endef not what you think!"""
|
||
|
|
),
|
||
|
|
'defineescaped': (
|
||
|
|
pymake.parser.iterdefinechars,
|
||
|
|
"""value \\
|
||
|
|
endef
|
||
|
|
endef\n""",
|
||
|
|
"value endef"
|
||
|
|
),
|
||
|
|
}
|
||
|
|
|
||
|
|
def runSingle(self, ifunc, idata, expected):
|
||
|
|
fd = StringIO(idata)
|
||
|
|
lineiter = enumerate(fd)
|
||
|
|
|
||
|
|
d = pymake.parser.DynamicData(lineiter, 'PlainIterTest-data')
|
||
|
|
d.readline()
|
||
|
|
|
||
|
|
actual = ''.join( (c for c, t, o, oo in ifunc(d, 0, pymake.parser._emptytokenlist)) )
|
||
|
|
self.assertEqual(actual, expected)
|
||
|
|
|
||
|
|
self.assertRaises(StopIteration, lambda: fd.next())
|
||
|
|
multitest(IterTest)
|
||
|
|
|
||
|
|
class MakeSyntaxTest(TestBase):
|
||
|
|
# (string, startat, stopat, stopoffset, expansion
|
||
|
|
testdata = {
|
||
|
|
'text': ('hello world', 0, (), None, ['hello world']),
|
||
|
|
'singlechar': ('hello $W', 0, (), None,
|
||
|
|
['hello ',
|
||
|
|
{'type': 'VariableRef',
|
||
|
|
'.vname': ['W']}
|
||
|
|
]),
|
||
|
|
'stopat': ('hello: world', 0, (':', '='), 6, ['hello']),
|
||
|
|
'funccall': ('h $(flavor FOO)', 0, (), None,
|
||
|
|
['h ',
|
||
|
|
{'type': 'FlavorFunction',
|
||
|
|
'[0]': ['FOO']}
|
||
|
|
]),
|
||
|
|
'escapedollar': ('hello$$world', 0, (), None, ['hello$world']),
|
||
|
|
'varref': ('echo $(VAR)', 0, (), None,
|
||
|
|
['echo ',
|
||
|
|
{'type': 'VariableRef',
|
||
|
|
'.vname': ['VAR']}
|
||
|
|
]),
|
||
|
|
'dynamicvarname': ('echo $($(VARNAME):.c=.o)', 0, (':',), None,
|
||
|
|
['echo ',
|
||
|
|
{'type': 'SubstitutionRef',
|
||
|
|
'.vname': [{'type': 'VariableRef',
|
||
|
|
'.vname': ['VARNAME']}
|
||
|
|
],
|
||
|
|
'.substfrom': ['.c'],
|
||
|
|
'.substto': ['.o']}
|
||
|
|
]),
|
||
|
|
'substref': (' $(VAR:VAL) := $(VAL)', 0, (':=', '+=', '=', ':'), 15,
|
||
|
|
[' ',
|
||
|
|
{'type': 'VariableRef',
|
||
|
|
'.vname': ['VAR:VAL']},
|
||
|
|
' ']),
|
||
|
|
'vadsubstref': (' $(VAR:VAL) = $(VAL)', 15, (), None,
|
||
|
|
[{'type': 'VariableRef',
|
||
|
|
'.vname': ['VAL']},
|
||
|
|
]),
|
||
|
|
}
|
||
|
|
|
||
|
|
def compareRecursive(self, actual, expected, path):
|
||
|
|
self.assertEqual(len(actual), len(expected),
|
||
|
|
"compareRecursive: %s" % (path,))
|
||
|
|
for i in xrange(0, len(actual)):
|
||
|
|
ipath = path + [i]
|
||
|
|
|
||
|
|
a, isfunc = actual[i]
|
||
|
|
e = expected[i]
|
||
|
|
if isinstance(e, str):
|
||
|
|
self.assertEqual(a, e, "compareRecursive: %s" % (ipath,))
|
||
|
|
else:
|
||
|
|
self.assertEqual(type(a), getattr(pymake.functions, e['type']),
|
||
|
|
"compareRecursive: %s" % (ipath,))
|
||
|
|
for k, v in e.iteritems():
|
||
|
|
if k == 'type':
|
||
|
|
pass
|
||
|
|
elif k[0] == '[':
|
||
|
|
item = int(k[1:-1])
|
||
|
|
proppath = ipath + [item]
|
||
|
|
self.compareRecursive(a[item], v, proppath)
|
||
|
|
elif k[0] == '.':
|
||
|
|
item = k[1:]
|
||
|
|
proppath = ipath + [item]
|
||
|
|
self.compareRecursive(getattr(a, item), v, proppath)
|
||
|
|
else:
|
||
|
|
raise Exception("Unexpected property at %s: %s" % (ipath, k))
|
||
|
|
|
||
|
|
def runSingle(self, s, startat, stopat, stopoffset, expansion):
|
||
|
|
d = pymake.parser.Data.fromstring(s, pymake.parserdata.Location('testdata', 1, 0))
|
||
|
|
|
||
|
|
a, t, offset = pymake.parser.parsemakesyntax(d, startat, stopat, pymake.parser.itermakefilechars)
|
||
|
|
self.compareRecursive(a, expansion, [])
|
||
|
|
self.assertEqual(offset, stopoffset)
|
||
|
|
|
||
|
|
multitest(MakeSyntaxTest)
|
||
|
|
|
||
|
|
class VariableTest(TestBase):
|
||
|
|
testdata = """
|
||
|
|
VAR = value
|
||
|
|
VARNAME = TESTVAR
|
||
|
|
$(VARNAME) = testvalue
|
||
|
|
$(VARNAME:VAR=VAL) = moretesting
|
||
|
|
IMM := $(VARNAME) # this is a comment
|
||
|
|
MULTIVAR = val1 \\
|
||
|
|
val2
|
||
|
|
VARNAME = newname
|
||
|
|
"""
|
||
|
|
expected = {'VAR': 'value',
|
||
|
|
'VARNAME': 'newname',
|
||
|
|
'TESTVAR': 'testvalue',
|
||
|
|
'TESTVAL': 'moretesting',
|
||
|
|
'IMM': 'TESTVAR ',
|
||
|
|
'MULTIVAR': 'val1 val2',
|
||
|
|
'UNDEF': None}
|
||
|
|
|
||
|
|
def runTest(self):
|
||
|
|
stream = StringIO(self.testdata)
|
||
|
|
stmts = pymake.parser.parsestream(stream, 'testdata')
|
||
|
|
|
||
|
|
m = pymake.data.Makefile()
|
||
|
|
stmts.execute(m)
|
||
|
|
for k, v in self.expected.iteritems():
|
||
|
|
flavor, source, val = m.variables.get(k)
|
||
|
|
if val is None:
|
||
|
|
self.assertEqual(val, v, 'variable named %s' % k)
|
||
|
|
else:
|
||
|
|
self.assertEqual(val.resolvestr(m, m.variables), v, 'variable named %s' % k)
|
||
|
|
|
||
|
|
class SimpleRuleTest(TestBase):
|
||
|
|
testdata = """
|
||
|
|
VAR = value
|
||
|
|
TSPEC = dummy
|
||
|
|
all: TSPEC = myrule
|
||
|
|
all:: test test2 $(VAR)
|
||
|
|
echo "Hello, $(TSPEC)"
|
||
|
|
|
||
|
|
%.o: %.c
|
||
|
|
$(CC) -o $@ $<
|
||
|
|
"""
|
||
|
|
|
||
|
|
def runTest(self):
|
||
|
|
stream = StringIO(self.testdata)
|
||
|
|
stmts = pymake.parser.parsestream(stream, 'testdata')
|
||
|
|
|
||
|
|
m = pymake.data.Makefile()
|
||
|
|
stmts.execute(m)
|
||
|
|
self.assertEqual(m.defaulttarget, 'all', "Default target")
|
||
|
|
|
||
|
|
self.assertTrue(m.hastarget('all'), "Has 'all' target")
|
||
|
|
target = m.gettarget('all')
|
||
|
|
rules = target.rules
|
||
|
|
self.assertEqual(len(rules), 1, "Number of rules")
|
||
|
|
prereqs = rules[0].prerequisites
|
||
|
|
self.assertEqual(prereqs, ['test', 'test2', 'value'], "Prerequisites")
|
||
|
|
commands = rules[0].commands
|
||
|
|
self.assertEqual(len(commands), 1, "Number of commands")
|
||
|
|
expanded = commands[0].resolvestr(m, target.variables)
|
||
|
|
self.assertEqual(expanded, 'echo "Hello, myrule"')
|
||
|
|
|
||
|
|
irules = m.implicitrules
|
||
|
|
self.assertEqual(len(irules), 1, "Number of implicit rules")
|
||
|
|
|
||
|
|
irule = irules[0]
|
||
|
|
self.assertEqual(len(irule.targetpatterns), 1, "%.o target pattern count")
|
||
|
|
self.assertEqual(len(irule.prerequisites), 1, "%.o prerequisite count")
|
||
|
|
self.assertEqual(irule.targetpatterns[0].match('foo.o'), 'foo', "%.o stem")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
unittest.main()
|