Refactor the symbolize step of the orderfile generation.

tools/cygprofile/symbolize_unittest.py

Index: tools/cygprofile/symbolize_unittest.py
diff --git a/tools/cygprofile/symbolize_unittest.py b/tools/cygprofile/symbolize_unittest.py
new file mode 100755
index 0000000000000000000000000000000000000000..e3a15664cdeb8a83e5930de3d3b35cd0e7c88f0e
--- /dev/null
+++ b/tools/cygprofile/symbolize_unittest.py
@@ -0,0 +1,91 @@
+#!/usr/bin/python
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import unittest
+
+import symbol_extractor
+import symbolize
+
+
+class TestSymbolizeTest(unittest.TestCase):
+  def testParseLogLines(self):
+    lines = """5086e000-52e92000 r-xp 00000000 b3:02 51276      libchromeview.so
+secs       usecs      pid:threadid    func
+START
+1314897086 795828     3587:1074648168 0x509e105c
+1314897086 795874     3587:1074648168 0x509e0eb4
+END""".split('\n')
+    call_info = symbolize._ParseLogLines(lines)
+    self.assertEquals(len(call_info), 2)
+    self.assertEquals(
+        call_info[0], (1314897086, 795828, '3587:1074648168',
+                       0x509e105c - 0x5086e000))
+    self.assertEquals(
+        call_info[1], (1314897086, 795874, '3587:1074648168',
+                       0x509e0eb4 - 0x5086e000))
+
+  def testFindSymbolInfosAtOffsetExactMatch(self):
+    offset_map = {0x10: [symbol_extractor.SymbolInfo(
+        name='Symbol', offset=0x10, size=0x13, section='.text')]}
+    functions = symbolize._FindSymbolInfosAtOffset(offset_map, 0x10)
+    self.assertEquals(len(functions), 1)
+    self.assertEquals(functions[0], offset_map[0x10][0])
+
+  def testFindSymbolInfosAtOffsetInexactMatch(self):
+    offset_map = {0x10: [symbol_extractor.SymbolInfo(
+        name='Symbol', offset=0x10, size=0x13, section='.text')]}
+    functions = symbolize._FindSymbolInfosAtOffset(offset_map, 0x11)
+    self.assertEquals(len(functions), 1)
+    self.assertEquals(functions[0], offset_map[0x10][0])
+
+  def testFindSymbolInfosAtOffsetNoMatch(self):
+    offset_map = {0x10: [symbol_extractor.SymbolInfo(
+        name='Symbol', offset=0x10, size=0x13, section='.text')]}
+    self.assertRaises(symbolize.SymbolNotFoundException,
+                      symbolize._FindSymbolInfosAtOffset, offset_map, 0x12)
+
+  def testWarnAboutDuplicates(self):
+    offsets = [0x1, 0x2, 0x3]
+    self.assertTrue(symbolize._WarnAboutDuplicates(offsets))
+    offsets.append(0x1)
+    self.assertFalse(symbolize._WarnAboutDuplicates(offsets))
+
+  def testOutputOrderfile(self):
+    class FakeOutputFile(object):
+      def __init__(self):
+        self.writes = []
+      def write(self, data):
+        self.writes.append(data)
+
+    # One symbol not matched, one with an odd address, one regularly matched
+    # And two symbols aliased to the same address
+    offsets = [0x12, 0x17]
+    offset_to_symbol_infos = {
+        0x10:[symbol_extractor.SymbolInfo(
+                  name='Symbol', offset=0x10, size=0x13, section='dummy')],
+        0x12:[symbol_extractor.SymbolInfo(
+                  name='Symbol2', offset=0x12, size=0x13, section='dummy')],
+        0x16:[symbol_extractor.SymbolInfo(
+                  name='Symbol3', offset=0x16, size=0x13, section='dummy'),
+              symbol_extractor.SymbolInfo(
+                  name='Symbol32', offset=0x16, size=0x13, section='dummy'),]}
+    symbol_to_section_map = {
+        'Symbol': '.text.Symbol',
+        'Symbol2': '.text.Symbol2',
+        'Symbol3': '.text.Symbol3',
+        'Symbol32': '.text.Symbol32'}
+    fake_output = FakeOutputFile()
+    symbolize._OutputOrderfile(offsets, offset_to_symbol_infos,
+                               symbol_to_section_map,
+                               fake_output)
+    expected = """.text.Symbol2
+.text.Symbol3
+.text.Symbol32
+"""
+    self.assertEquals(expected, "".join(fake_output.writes))
+
+
+if __name__ == '__main__':
+  unittest.main()