OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # Copyright 2015 The Chromium Authors. All rights reserved. | 2 # Copyright 2015 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import unittest | 6 import unittest |
7 | 7 |
8 import patch_orderfile | 8 import patch_orderfile |
9 import symbol_extractor | 9 import symbol_extractor |
10 | 10 |
11 | 11 |
12 class TestPatchOrderFile(unittest.TestCase): | 12 class TestPatchOrderFile(unittest.TestCase): |
13 def testRemoveClone(self): | 13 def testRemoveClone(self): |
14 no_clone = "this.does.not.contain.clone" | 14 no_clone = "this.does.not.contain.clone" |
15 self.assertEquals(no_clone, patch_orderfile._RemoveClone(no_clone)) | 15 self.assertEquals(no_clone, patch_orderfile._RemoveClone(no_clone)) |
16 with_clone = "this.does.contain.clone." | 16 with_clone = "this.does.contain.clone." |
17 self.assertEquals( | 17 self.assertEquals( |
18 "this.does.contain", patch_orderfile._RemoveClone(with_clone)) | 18 "this.does.contain", patch_orderfile._RemoveClone(with_clone)) |
19 | 19 |
20 def testAliasClonedSymbols(self): | 20 def testAliasClonedSymbols(self): |
21 symbol_infos = [ | 21 symbol_infos = [ |
22 symbol_extractor.SymbolInfo(name='aSymbol', offset=0x42, size=0x12), | 22 symbol_extractor.SymbolInfo(name='aSymbol', offset=0x42, size=0x12, |
23 symbol_extractor.SymbolInfo(name='aSymbol.clone.', offset=8, size=1)] | 23 section='.text'), |
| 24 symbol_extractor.SymbolInfo(name='aSymbol.clone.', offset=8, size=1, |
| 25 section='.text')] |
24 (offset_to_symbol_infos, name_to_symbol_infos) = \ | 26 (offset_to_symbol_infos, name_to_symbol_infos) = \ |
25 patch_orderfile._GroupSymbolInfos(symbol_infos) | 27 patch_orderfile._GroupSymbolInfos(symbol_infos) |
26 self.assertEquals(len(offset_to_symbol_infos), 2) | 28 self.assertEquals(len(offset_to_symbol_infos), 2) |
27 for i in range(2): | 29 for i in range(2): |
28 s = symbol_infos[i] | 30 s = symbol_infos[i] |
29 matching = offset_to_symbol_infos[s.offset][0] | 31 matching = offset_to_symbol_infos[s.offset][0] |
30 self.assertEquals(matching.offset, s.offset) | 32 self.assertEquals(matching.offset, s.offset) |
31 self.assertEquals(matching.size, s.size) | 33 self.assertEquals(matching.size, s.size) |
32 self.assertEquals(len(name_to_symbol_infos), 1) | 34 self.assertEquals(len(name_to_symbol_infos), 1) |
33 self.assertEquals(len(name_to_symbol_infos['aSymbol']), 2) | 35 self.assertEquals(len(name_to_symbol_infos['aSymbol']), 2) |
34 | 36 |
35 def testGroupSymbolsByOffset(self): | 37 def testGroupSymbolsByOffset(self): |
36 symbol_infos = ( | 38 symbol_infos = ( |
37 symbol_extractor.SymbolInfo(name='aSymbol', offset=0x42, size=0x12), | 39 symbol_extractor.SymbolInfo(name='aSymbol', offset=0x42, size=0x12, |
38 symbol_extractor.SymbolInfo(name='anotherSymbol', offset=0x42, size=1)) | 40 section='.text'), |
| 41 symbol_extractor.SymbolInfo(name='anotherSymbol', offset=0x42, size=1, |
| 42 section='.text')) |
39 (offset_to_symbol_infos, _) = \ | 43 (offset_to_symbol_infos, _) = \ |
40 patch_orderfile._GroupSymbolInfos(symbol_infos) | 44 patch_orderfile._GroupSymbolInfos(symbol_infos) |
41 self.assertEquals(len(offset_to_symbol_infos), 1) | 45 self.assertEquals(len(offset_to_symbol_infos), 1) |
42 self.assertEquals(tuple(offset_to_symbol_infos[0x42]), symbol_infos) | 46 self.assertEquals(tuple(offset_to_symbol_infos[0x42]), symbol_infos) |
43 | 47 |
44 def testExpandSymbols(self): | 48 def testExpandSymbols(self): |
45 symbol_name = "dummySymbol" | 49 symbol_name = "dummySymbol" |
46 symbol_name2 = "other" | 50 symbol_name2 = "other" |
47 profiled_symbol_names = [symbol_name, "symbolThatShouldntMatch"] | 51 profiled_symbol_names = [symbol_name, "symbolThatShouldntMatch"] |
48 name_to_symbol_infos = {symbol_name: [ | 52 name_to_symbol_infos = {symbol_name: [ |
49 symbol_extractor.SymbolInfo(symbol_name, 0x42, 0x12)]} | 53 symbol_extractor.SymbolInfo(symbol_name, 0x42, 0x12, |
| 54 section='.text')]} |
50 offset_to_symbol_infos = { | 55 offset_to_symbol_infos = { |
51 0x42: [symbol_extractor.SymbolInfo(symbol_name, 0x42, 0x12), | 56 0x42: [symbol_extractor.SymbolInfo(symbol_name, 0x42, 0x12, |
52 symbol_extractor.SymbolInfo(symbol_name2, 0x42, 0x12)]} | 57 section='.text'), |
| 58 symbol_extractor.SymbolInfo(symbol_name2, 0x42, 0x12, |
| 59 section='.text')]} |
53 symbol_names = patch_orderfile._ExpandSymbols( | 60 symbol_names = patch_orderfile._ExpandSymbols( |
54 profiled_symbol_names, name_to_symbol_infos, offset_to_symbol_infos) | 61 profiled_symbol_names, name_to_symbol_infos, offset_to_symbol_infos) |
55 self.assertEquals(len(symbol_names), 3) | 62 self.assertEquals(len(symbol_names), 3) |
56 self.assertEquals(symbol_names[0], symbol_name) | 63 self.assertEquals(symbol_names[0], symbol_name) |
57 self.assertEquals(symbol_names[1], symbol_name2) | 64 self.assertEquals(symbol_names[1], symbol_name2) |
58 self.assertEquals(symbol_names[2], "symbolThatShouldntMatch") | 65 self.assertEquals(symbol_names[2], "symbolThatShouldntMatch") |
59 | 66 |
60 def testPrintSymbolWithPrefixes(self): | 67 def testPrintSymbolWithPrefixes(self): |
61 class FakeOutputFile(object): | 68 class FakeOutputFile(object): |
62 def __init__(self): | 69 def __init__(self): |
63 self.output = '' | 70 self.output = '' |
64 def write(self, s): | 71 def write(self, s): |
65 self.output = self.output + s | 72 self.output = self.output + s |
66 test_symbol = "dummySymbol" | 73 test_symbol = "dummySymbol" |
67 symbol_names = [test_symbol] | 74 symbol_names = [test_symbol] |
68 fake_output = FakeOutputFile() | 75 fake_output = FakeOutputFile() |
69 patch_orderfile._PrintSymbolsWithPrefixes(symbol_names, fake_output) | 76 patch_orderfile._PrintSymbolsWithPrefixes(symbol_names, fake_output) |
70 expected_output = """.text.startup.dummySymbol | 77 expected_output = """.text.startup.dummySymbol |
71 .text.hot.dummySymbol | 78 .text.hot.dummySymbol |
72 .text.unlikely.dummySymbol | 79 .text.unlikely.dummySymbol |
73 .text.dummySymbol | 80 .text.dummySymbol |
74 """ | 81 """ |
75 self.assertEquals(fake_output.output, expected_output) | 82 self.assertEquals(fake_output.output, expected_output) |
76 | 83 |
77 | 84 |
78 if __name__ == "__main__": | 85 if __name__ == "__main__": |
79 unittest.main() | 86 unittest.main() |
OLD | NEW |