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 patch_orderfile | 6 import patch_orderfile |
7 import unittest | 7 import unittest |
8 | 8 |
9 | 9 |
10 class TestPatchOrderFile(unittest.TestCase): | 10 class TestPatchOrderFile(unittest.TestCase): |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
59 | 59 |
60 self.assertEquals(len(offset_to_symbol_infos[test_offset]), 1) | 60 self.assertEquals(len(offset_to_symbol_infos[test_offset]), 1) |
61 s = offset_to_symbol_infos[test_offset][0] | 61 s = offset_to_symbol_infos[test_offset][0] |
62 self.assertEquals(test_offset, s.offset) | 62 self.assertEquals(test_offset, s.offset) |
63 self.assertEquals(-1, s.size) | 63 self.assertEquals(-1, s.size) |
64 self.assertEquals(test_name, s.name) | 64 self.assertEquals(test_name, s.name) |
65 | 65 |
66 def testGetSymbolsFromStream(self): | 66 def testGetSymbolsFromStream(self): |
67 lines = [".text.startup.", | 67 lines = [".text.startup.", |
68 ".text.with.a.prefix", | 68 ".text.with.a.prefix", |
69 "_ZN2v88internal33HEnvironmentLivenessAnalysisPhase3RunEv"] | 69 "", |
| 70 "_ZN2v88internal33HEnvironmentLivenessAnalysisPhase3RunEv", |
| 71 ".text", |
| 72 ".text.*"] |
70 names = patch_orderfile._GetSymbolsFromStream(lines) | 73 names = patch_orderfile._GetSymbolsFromStream(lines) |
71 self.assertEquals(len(names), 2) | 74 self.assertEquals(len(names), 2) |
72 self.assertEquals( | 75 self.assertEquals( |
73 names[0], "with.a.prefix") | 76 names[0], "with.a.prefix") |
74 self.assertEquals( | 77 self.assertEquals( |
75 names[1], "_ZN2v88internal33HEnvironmentLivenessAnalysisPhase3RunEv") | 78 names[1], "_ZN2v88internal33HEnvironmentLivenessAnalysisPhase3RunEv") |
76 | 79 |
77 | 80 |
78 def testMatchProfiledSymbols(self): | 81 def testMatchProfiledSymbols(self): |
79 symbol_name = "dummySymbol" | 82 symbol_name = "dummySymbol" |
(...skipping 14 matching lines...) Expand all Loading... |
94 patch_orderfile.SymbolInfo(0x42, 0x12, symbol_name2)]} | 97 patch_orderfile.SymbolInfo(0x42, 0x12, symbol_name2)]} |
95 symbol_names = patch_orderfile._ExpandSymbolsWithDupsFromSameOffset( | 98 symbol_names = patch_orderfile._ExpandSymbolsWithDupsFromSameOffset( |
96 symbols, offset_to_symbol_infos) | 99 symbols, offset_to_symbol_infos) |
97 self.assertEquals(len(symbol_names), 2) | 100 self.assertEquals(len(symbol_names), 2) |
98 self.assertEquals(symbol_names[0], symbol_name) | 101 self.assertEquals(symbol_names[0], symbol_name) |
99 self.assertEquals(symbol_names[1], symbol_name2) | 102 self.assertEquals(symbol_names[1], symbol_name2) |
100 | 103 |
101 def testPrintSymbolWithPrefixes(self): | 104 def testPrintSymbolWithPrefixes(self): |
102 class FakeOutputFile(object): | 105 class FakeOutputFile(object): |
103 def __init__(self): | 106 def __init__(self): |
104 self.output = [] | 107 self.output = '' |
105 def write(self, s): | 108 def write(self, s): |
106 self.output.append(s) | 109 self.output = self.output + s |
107 test_symbol = "dummySymbol" | 110 test_symbol = "dummySymbol" |
108 symbol_names = [test_symbol] | 111 symbol_names = [test_symbol] |
109 fake_output = FakeOutputFile() | 112 fake_output = FakeOutputFile() |
110 patch_orderfile._PrintSymbolsWithPrefixes(symbol_names, fake_output) | 113 patch_orderfile._PrintSymbolsWithPrefixes(symbol_names, fake_output) |
111 expected_output = """.text.startup.dummySymbol | 114 expected_output = """.text.startup.dummySymbol |
112 .text.hot.dummySymbol | 115 .text.hot.dummySymbol |
113 .text.unlikely.dummySymbol | 116 .text.unlikely.dummySymbol |
114 .text.dummySymbol | 117 .text.dummySymbol |
115 """ | 118 """ |
116 self.assertEquals("\n".join(fake_output.output), expected_output) | 119 self.assertEquals(fake_output.output, expected_output) |
117 | 120 |
118 | 121 |
119 if __name__ == "__main__": | 122 if __name__ == "__main__": |
120 unittest.main() | 123 unittest.main() |
OLD | NEW |