diff options
Diffstat (limited to 'scripts/pybootchartgui/pybootchartgui/tests/process_tree_test.py')
-rw-r--r-- | scripts/pybootchartgui/pybootchartgui/tests/process_tree_test.py | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/scripts/pybootchartgui/pybootchartgui/tests/process_tree_test.py b/scripts/pybootchartgui/pybootchartgui/tests/process_tree_test.py new file mode 100644 index 0000000000..971e125eab --- /dev/null +++ b/scripts/pybootchartgui/pybootchartgui/tests/process_tree_test.py | |||
@@ -0,0 +1,78 @@ | |||
1 | import sys | ||
2 | import os | ||
3 | import unittest | ||
4 | |||
5 | sys.path.insert(0, os.getcwd()) | ||
6 | |||
7 | import parsing | ||
8 | import process_tree | ||
9 | |||
10 | class TestProcessTree(unittest.TestCase): | ||
11 | |||
12 | def setUp(self): | ||
13 | self.name = "Process tree unittest" | ||
14 | self.rootdir = '../examples/1' | ||
15 | self.ps_stats = parsing.parse_file(parsing.ParserState(), self.mk_fname('proc_ps.log')).ps_stats | ||
16 | self.processtree = process_tree.ProcessTree(self.ps_stats, None, False, for_testing = True) | ||
17 | |||
18 | def mk_fname(self,f): | ||
19 | return os.path.join(self.rootdir, f) | ||
20 | |||
21 | def flatten(self, process_tree): | ||
22 | flattened = [] | ||
23 | for p in process_tree: | ||
24 | flattened.append(p) | ||
25 | flattened.extend(self.flatten(p.child_list)) | ||
26 | return flattened | ||
27 | |||
28 | def checkAgainstJavaExtract(self, filename, process_tree): | ||
29 | for expected, actual in zip(open(filename), self.flatten(process_tree)): | ||
30 | tokens = expected.split('\t') | ||
31 | self.assertEqual(int(tokens[0]), actual.pid) | ||
32 | self.assertEqual(tokens[1], actual.cmd) | ||
33 | self.assertEqual(long(tokens[2]), 10 * actual.start_time) | ||
34 | self.assert_(long(tokens[3]) - 10 * actual.duration < 5, "duration") | ||
35 | self.assertEqual(int(tokens[4]), len(actual.child_list)) | ||
36 | self.assertEqual(int(tokens[5]), len(actual.samples)) | ||
37 | |||
38 | def testBuild(self): | ||
39 | process_tree = self.processtree.process_tree | ||
40 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.1.log'), process_tree) | ||
41 | |||
42 | def testMergeLogger(self): | ||
43 | self.processtree.merge_logger(self.processtree.process_tree, 'bootchartd', None, False) | ||
44 | process_tree = self.processtree.process_tree | ||
45 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.2.log'), process_tree) | ||
46 | |||
47 | def testPrune(self): | ||
48 | self.processtree.merge_logger(self.processtree.process_tree, 'bootchartd', None, False) | ||
49 | self.processtree.prune(self.processtree.process_tree, None) | ||
50 | process_tree = self.processtree.process_tree | ||
51 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.3b.log'), process_tree) | ||
52 | |||
53 | def testMergeExploders(self): | ||
54 | self.processtree.merge_logger(self.processtree.process_tree, 'bootchartd', None, False) | ||
55 | self.processtree.prune(self.processtree.process_tree, None) | ||
56 | self.processtree.merge_exploders(self.processtree.process_tree, set(['hwup'])) | ||
57 | process_tree = self.processtree.process_tree | ||
58 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.3c.log'), process_tree) | ||
59 | |||
60 | def testMergeSiblings(self): | ||
61 | self.processtree.merge_logger(self.processtree.process_tree, 'bootchartd', None, False) | ||
62 | self.processtree.prune(self.processtree.process_tree, None) | ||
63 | self.processtree.merge_exploders(self.processtree.process_tree, set(['hwup'])) | ||
64 | self.processtree.merge_siblings(self.processtree.process_tree) | ||
65 | process_tree = self.processtree.process_tree | ||
66 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.3d.log'), process_tree) | ||
67 | |||
68 | def testMergeRuns(self): | ||
69 | self.processtree.merge_logger(self.processtree.process_tree, 'bootchartd', None, False) | ||
70 | self.processtree.prune(self.processtree.process_tree, None) | ||
71 | self.processtree.merge_exploders(self.processtree.process_tree, set(['hwup'])) | ||
72 | self.processtree.merge_siblings(self.processtree.process_tree) | ||
73 | self.processtree.merge_runs(self.processtree.process_tree) | ||
74 | process_tree = self.processtree.process_tree | ||
75 | self.checkAgainstJavaExtract(self.mk_fname('extract.processtree.3e.log'), process_tree) | ||
76 | |||
77 | if __name__ == '__main__': | ||
78 | unittest.main() | ||