1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
|
import unittest
# hack to save ourselves from having to use PYTHONPATH
import sys
import os
sys.path.append(os.path.join (os.getcwd(), '..'))
import bake
import re
class SrcTest:
_name=''
def __init__(self, name):
self._name = name
class DepInfo:
def __init__(self):
self._edges = []
self._optional_edges = []
def add(self,src,dst, optional = False):
self._optional_edges.append([src,dst, optional])
def edges(self):
return self._edges
class Test:
def __init__(self, deps_str, failure_targets):
# the syntax for deps_str encodes many things:
# B depends on A: "A -> B"
# When processing B, add new dependencies to dependency graph:
# B {
# C -> D
# D -> B
# }
# B depends optionally on A: "A ?> B"
#
self._deps = bake.Dependencies()
self._infos = self._parse(deps_str, self._deps)
self._failure = failure_targets
self._processed = []
def _parse(self,deps_str, deps):
infos = dict()
reg_edge = re.compile('([^ ]+) -> (.+)')
reg_opt_edge = re.compile('([^ ]+) \?> (.+)')
reg_open = re.compile('([^ ]+) {?')
reg_close = re.compile('}?')
current_info = None
current_info_dst = ''
for line in deps_str.splitlines():
match_edge = reg_edge.search(line)
match_opt_edge = reg_opt_edge.search(line)
match_open = reg_open.search(line)
match_close = reg_close.search(line)
if match_edge is not None:
src = match_edge.group(1)
dst = match_edge.group(2)
if current_info is None:
deps.add_dst(src)
deps.add_dst(dst)
deps.add_dep(src,dst)
else:
current_info.add(src,dst)
continue
if match_opt_edge is not None:
src = match_opt_edge.group(1)
dst = match_opt_edge.group(2)
if current_info is None:
deps.add_dst(src)
deps.add_dst(dst)
deps.add_dep(src,dst, optional = True)
else:
current_info.add(src,dst, optional = True)
continue
if match_open is not None:
dst = match_open.group(1)
current_info = DepInfo()
current_info_dst = dst;
continue
if match_close is not None:
infos[current_info_dst] = current_info
continue
return infos
def _dep_handler(self, target, context):
if target in self._failure:
return False
self._processed.append(target)
if self._infos.has_key(target):
for src,dst,optional in self._infos[target].edges ():
self._deps.add_dst(src)
self._deps.add_dst(dst)
self._deps.add_dep(src,dst, optional)
return True
def run(self, targets):
self._deps.resolve(targets, self._dep_handler, None)
return self._processed
class TestModuleDependencies(unittest.TestCase):
""" Tests for the dependencies between modules. """
def run_one_test(self, deps, targets, expected, failure = []):
test = Test (deps, failure)
got = test.run(targets)
self.assertEqual(got, expected)
def test_simple(self):
deps = """
foo.h -> foo.c
foo.c -> foo.o
bar.h -> bar.c
bar.c -> bar.o
foo.o -> main
bar.o -> main
lex.yy -> lex.yy.h
lex.yy.h -> bar.c
"""
self.run_one_test(deps, ['bar.o'],
['bar.h', 'lex.yy', 'lex.yy.h', 'bar.c', 'bar.o'])
self.run_one_test(deps, ['foo.o'],
['foo.h', 'foo.c', 'foo.o'])
self.run_one_test(deps, ['main'],
['bar.h', 'foo.h', 'lex.yy', 'foo.c', 'lex.yy.h',
'bar.c', 'foo.o', 'bar.o', 'main'])
self.run_one_test(deps, ['main', 'foo.o'],
['bar.h', 'foo.h', 'lex.yy', 'foo.c', 'lex.yy.h',
'bar.c', 'foo.o', 'bar.o', 'main'])
def Dtest_optional(self):
self.run_one_test("A ?> B", targets = [SrcTest('B')],
expected = [SrcTest('A'), SrcTest('B')])
self.run_one_test("A ?> B", targets = [SrcTest('B')],
expected = [SrcTest('B')],
failure = [SrcTest('A')])
self.assertRaises(bake.DependencyUnmet, self.run_one_test,
"A -> B", targets = [SrcTest('B')],
expected = [SrcTest('B')],
failure = [SrcTest('A')])
self.run_one_test("""A ?> B
B -> C
""", targets = ['C'],
expected = ['B', 'C'],
failure = ['A'])
self.run_one_test("""A ?> B
C ?> B
""", targets = ['B'],
expected = ['B'],
failure = ['A', 'C'])
if __name__ == '__main__':
unittest.main()
|