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 = { 'oneline': ("He\tllo", "f", 1, 0, ((0, "f", 1, 0), (2, "f", 1, 2), (3, "f", 1, 4))), 'twoline': ("line1 \n\tl\tine2", "f", 1, 4, ((0, "f", 1, 4), (5, "f", 1, 9), (6, "f", 1, 10), (7, "f", 2, 0), (8, "f", 2, 4), (10, "f", 2, 8), (13, "f", 2, 11))), } def runSingle(self, data, filename, line, col, results): d = pymake.parser.Data(data, 0, len(data), pymake.parserdata.Location(filename, line, col)) for pos, file, lineno, col in results: loc = d.getloc(pos) self.assertEqual(loc.path, file, "data file offset %i" % pos) self.assertEqual(loc.line, lineno, "data line offset %i" % pos) self.assertEqual(loc.column, col, "data col offset %i" % pos) multitest(DataTest) class LineEnumeratorTest(TestBase): testdata = { 'simple': ( 'Hello, world', [ ('Hello, world', 1), ] ), 'multi': ( 'Hello\nhappy \n\nworld\n', [ ('Hello', 1), ('happy ', 2), ('', 3), ('world', 4), ('', 5), ] ), 'continuation': ( 'Hello, \\\n world\nJellybeans!', [ ('Hello, \\\n world', 1), ('Jellybeans!', 3), ] ), 'multislash': ( 'Hello, \\\\\n world', [ ('Hello, \\\\', 1), (' world', 2), ] ) } def runSingle(self, s, lines): gotlines = [(d.s[d.lstart:d.lend], d.loc.line) for d in pymake.parser.enumeratelines(s, 'path')] self.assertEqual(gotlines, lines) multitest(LineEnumeratorTest) 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", "VAR = val # escaped hash" ), 'makeescapedslash': ( pymake.parser.itermakefilechars, "VAR = val\\\\", "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", "echo boo # comment", ), 'commandcomment': ( pymake.parser.itercommandchars, "echo boo \# comment", "echo boo \# comment", ), 'commandcontinue': ( pymake.parser.itercommandchars, "echo boo # \\\n\t command 2", "echo boo # \\\n command 2" ), } def runSingle(self, ifunc, idata, expected): d = pymake.parser.Data.fromstring(idata, 'IterTest data') it = pymake.parser._alltokens.finditer(d.s, 0, d.lend) actual = ''.join( [c for c, t, o, oo in ifunc(d, 0, ('dummy-token',), it)] ) self.assertEqual(actual, expected) if ifunc == pymake.parser.itermakefilechars: print "testing %r" % expected self.assertEqual(pymake.parser.flattenmakesyntax(d, 0), expected) multitest(IterTest) # '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" # ), 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 %r" % (path, actual)) 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): stmts = pymake.parser.parsestring(self.testdata, 'VariableTest') 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): stmts = pymake.parser.parsestring(self.testdata, 'SimpleRuleTest') 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()