Chromium Code Reviews| 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')] | |
|
pasko
2015/01/28 18:55:37
nit: some more whitespace here and below
Benoit L
2015/01/29 10:02:01
Done.
| |
| 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 |