src/third_party/WebKit/Tools/Scripts/webkitpy/common/system/profiler_unittest.py

   1 # Copyright (C) 2012 Google Inc. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or without
   4 # modification, are permitted provided that the following conditions are
   5 # met:
   6 #
   7 #     * Redistributions of source code must retain the above copyright
   8 # notice, this list of conditions and the following disclaimer.
   9 #     * Redistributions in binary form must reproduce the above
  10 # copyright notice, this list of conditions and the following disclaimer
  11 # in the documentation and/or other materials provided with the
  12 # distribution.
  13 #     * Neither the name of Google Inc. nor the names of its
  14 # contributors may be used to endorse or promote products derived from
  15 # this software without specific prior written permission.
  16 #
  17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28
  29 import unittest
  30
  31 from webkitpy.common.system.platforminfo_mock import MockPlatformInfo
  32 from webkitpy.common.system.systemhost_mock import MockSystemHost
  33
  34 from .profiler import ProfilerFactory, GooglePProf
  35
  36
  37 class ProfilerFactoryTest(unittest.TestCase):
  38     def _assert_default_profiler_name(self, os_name, expected_profiler_name):
  39         profiler_name = ProfilerFactory.default_profiler_name(MockPlatformInfo(os_name))
  40         self.assertEqual(profiler_name, expected_profiler_name)
  41
  42     def test_default_profilers(self):
  43         self._assert_default_profiler_name('mac', 'iprofiler')
  44         self._assert_default_profiler_name('linux', 'perf')
  45         self._assert_default_profiler_name('win32', None)
  46         self._assert_default_profiler_name('freebsd', None)
  47
  48     def test_default_profiler_output(self):
  49         host = MockSystemHost()
  50         self.assertFalse(host.filesystem.exists("/tmp/output"))
  51
  52         # Default mocks are Mac, so iprofile should be default.
  53         profiler = ProfilerFactory.create_profiler(host, '/bin/executable', '/tmp/output')
  54         self.assertTrue(host.filesystem.exists("/tmp/output"))
  55         self.assertEqual(profiler._output_path, "/tmp/output/test.dtps")
  56
  57         # Linux defaults to perf.
  58         host.platform.os_name = 'linux'
  59         profiler = ProfilerFactory.create_profiler(host, '/bin/executable', '/tmp/output')
  60         self.assertEqual(profiler._output_path, "/tmp/output/test.data")
  61
  62
  63 class GooglePProfTest(unittest.TestCase):
  64     def test_pprof_output_regexp(self):
  65         pprof_output = """
  66 sometimes
  67 there
  68 is
  69 junk before the total line
  70
  71
  72 Total: 3770 samples
  73       76   2.0%   2.0%      104   2.8% lookup (inline)
  74       60   1.6%   3.6%       60   1.6% FL_SetPrevious (inline)
  75       56   1.5%   5.1%       56   1.5% MaskPtr (inline)
  76       51   1.4%   6.4%      222   5.9% WebCore::HTMLTokenizer::nextToken
  77       42   1.1%   7.6%       47   1.2% WTF::Vector::shrinkCapacity
  78       35   0.9%   8.5%       35   0.9% WTF::RefPtr::get (inline)
  79       33   0.9%   9.4%       43   1.1% append (inline)
  80       29   0.8%  10.1%       67   1.8% WTF::StringImpl::deref (inline)
  81       29   0.8%  10.9%      100   2.7% add (inline)
  82       28   0.7%  11.6%       28   0.7% WebCore::QualifiedName::localName (inline)
  83       25   0.7%  12.3%       27   0.7% WebCore::Private::addChildNodesToDeletionQueue
  84       24   0.6%  12.9%       24   0.6% __memcpy_ssse3_back
  85       23   0.6%  13.6%       23   0.6% intHash (inline)
  86       23   0.6%  14.2%       76   2.0% tcmalloc::FL_Next
  87       23   0.6%  14.8%       95   2.5% tcmalloc::FL_Push
  88       22   0.6%  15.4%       22   0.6% WebCore::MarkupTokenizerBase::InputStreamPreprocessor::peek (inline)
  89 """
  90         expected_first_ten_lines = """      76   2.0%   2.0%      104   2.8% lookup (inline)
  91       60   1.6%   3.6%       60   1.6% FL_SetPrevious (inline)
  92       56   1.5%   5.1%       56   1.5% MaskPtr (inline)
  93       51   1.4%   6.4%      222   5.9% WebCore::HTMLTokenizer::nextToken
  94       42   1.1%   7.6%       47   1.2% WTF::Vector::shrinkCapacity
  95       35   0.9%   8.5%       35   0.9% WTF::RefPtr::get (inline)
  96       33   0.9%   9.4%       43   1.1% append (inline)
  97       29   0.8%  10.1%       67   1.8% WTF::StringImpl::deref (inline)
  98       29   0.8%  10.9%      100   2.7% add (inline)
  99       28   0.7%  11.6%       28   0.7% WebCore::QualifiedName::localName (inline)
 100 """
 101         host = MockSystemHost()
 102         profiler = GooglePProf(host, '/bin/executable', '/tmp/output')
 103         self.assertEqual(profiler._first_ten_lines_of_profile(pprof_output), expected_first_ten_lines)