#!/usr/bin/env vpython3
#
# Copyright 2019 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for run_simpleperf.py"""

import os
import unittest

import mock  # pylint: disable=import-error

from run_simpleperf import SimplePerfRunner
from run_simpleperf import StackAddressInterpreter

_EXAMPLE_WEBVIEW_PACKAGE_NAME = "com.google.android.webview"

_EXAMPLE_STACK_SCRIPT_INPUT = [
  (
    "11-15 00:00:00.000 11111 11111 E chromium: #00 0x0000001111111111 "
    "/data/app/com.google.android.webview--8E2vMMZTpLVeEKY7ZgoHQ=="
    "/lib/arm64/libwebviewchromium.so+0x00000000083a4db8"
  ),
  (
    "11-15 00:00:00.000 11111 11111 E chromium: #00 0x0000001111111111 "
    "/data/app/com.google.android.webview--8E2vMMZTpLVeEKY7ZgoHQ=="
    "/lib/arm64/libwebviewchromium.so+0x00000000083db114"
  ),
  (
    "11-15 00:00:00.000 11111 11111 E chromium: #00 0x0000001111111111 "
    "/data/app/com.google.android.webview--8E2vMMZTpLVeEKY7ZgoHQ=="
    "/lib/arm64/libwebviewchromium.so+0x000000000abcdef0"
  ),
]

_EXAMPLE_STACK_SCRIPT_OUTPUT = [
  "Stack Trace:",
  "  RELADDR   FUNCTION",
  (
    "  00000000083a4db8  mojo::core::ports::(anonymous namespace)::UpdateTLS("
    "mojo::core::ports::PortLocker*, mojo::core::ports::PortLocker*)  "
    "../../mojo/core/ports/port_locker.cc:26:3"
  ),
  "",
  "-----------------------------------------------------",
  "",
  "Stack Trace:",
  "  RELADDR   FUNCTION",
  (
    "  00000000083db114  viz::GLRenderer::SetUseProgram(viz::ProgramKey "
    "const&, gfx::ColorSpace const&, gfx::ColorSpace const&)  "
    "../../components/viz/service/display/gl_renderer.cc:3267:14"
  ),
  "",
  "-----------------------------------------------------",
]

_ADDRESSES = ['83a4db8', '83db114', 'abcdef0']  # 3rd one is ignored
_WEBVIEW_LIB_NAME = 'libwebviewchromium.so'
_WEBVIEW_LIB_PATH = (
  '/data/app/com.google.android.webview'
  '--8E2vMMZTpLVeEKY7ZgoHQ==/lib/arm64/libwebviewchromium.so'
)

_EXAMPLE_INTERPRETER_OUTPUT = [
  (
    '83a4db8',
    (
      'mojo::core::ports::(anonymous namespace)::UpdateTLS('
      'mojo::core::ports::PortLocker*, mojo::core::ports::PortLocker*)'
    ),
  ),
  (
    '83db114',
    (
      'viz::GLRenderer::SetUseProgram(viz::ProgramKey const&, '
      'gfx::ColorSpace const&, gfx::ColorSpace const&)'
    ),
  ),
]

_EXAMPLE_INTERPRETER_OUTPUT_WITH_FILE_NAME_LINE = [
  (
    '83a4db8',
    (
      'mojo::core::ports::(anonymous namespace)::UpdateTLS('
      'mojo::core::ports::PortLocker*, mojo::core::ports::PortLocker*)'
      ' | ../../mojo/core/ports/port_locker.cc:26:3'
    ),
  ),
  (
    '83db114',
    (
      'viz::GLRenderer::SetUseProgram(viz::ProgramKey const&, '
      'gfx::ColorSpace const&, gfx::ColorSpace const&)'
      ' | ../../components/viz/service/display/gl_renderer.cc:3267:14'
    ),
  ),
]

_MOCK_ORIGINAL_REPORT = [
  '"442": {"l": 28, "f": "libwebviewchromium.so[+3db7d84]"},',
  '"443": {"l": 28, "f": "libwebviewchromium.so[+3db7a5c]"},',
  '"444": {"l": 28, "f": "libwebviewchromium.so[+aaaaaaa]"},',
]

_MOCK_ADDRESSES = ['3db7d84', '3db7a5c', 'aaaaaaa']

_MOCK_ADDRESS_FUNCTION_NAME_PAIRS = [
  ('3db7d84', 'MyClass::FirstMethod(const char*)'),
  ('3db7a5c', 'MyClass::SecondMethod(int)'),
]

_MOCK_FINAL_REPORT = [
  (
    '"442": {"l": 28, "f": "libwebviewchromium.so[MyClass::'
    'FirstMethod(const char*)]"},'
  ),
  (
    '"443": {"l": 28, "f": "libwebviewchromium.so[MyClass::'
    'SecondMethod(int)]"},'
  ),
  ('"444": {"l": 28, "f": "libwebviewchromium.so[+aaaaaaa]"},'),
]


class _RunSimpleperfTest(unittest.TestCase):
  """Unit tests for the run_simpleperf module."""

  def _AssertFileLines(self, mock_open, expected_lines):
    handle = mock_open()
    # Get 'str1', 'str2', ... from the call to f.write(str_i + '\n') which is
    # saved as [(str1 + '\n'), (str2 + '\n'), ...].
    actual_lines = [args[0][:-1] for (args, _) in handle.write.call_args_list]
    self.assertEqual(expected_lines, actual_lines)

  def setUp(self):
    self.tmp_dir = '/tmp'  # the actual directory won't be used in this test.
    self.args = mock.Mock(
      report_path=os.path.join(self.tmp_dir, 'report.html'),
      show_file_line=False,
    )
    self.device = mock.Mock()

    self.stack_address_interpreter = StackAddressInterpreter(
      self.args, self.tmp_dir
    )
    self.simple_perf_runner = SimplePerfRunner(
      self.device, self.args, self.tmp_dir, self.stack_address_interpreter
    )

  @mock.patch('run_simpleperf.open', new_callable=mock.mock_open)
  def testStackAddressInterpreter(self, mock_open):
    StackAddressInterpreter.RunStackScript = mock.Mock(
      return_value=_EXAMPLE_STACK_SCRIPT_OUTPUT
    )
    self.assertEqual(
      _EXAMPLE_INTERPRETER_OUTPUT,
      self.stack_address_interpreter.Interpret(_ADDRESSES, _WEBVIEW_LIB_PATH),
    )
    self._AssertFileLines(mock_open, _EXAMPLE_STACK_SCRIPT_INPUT)

  @mock.patch('run_simpleperf.open', new_callable=mock.mock_open)
  def testStackAddressInterpreterWithFileNameLine(self, mock_open):
    self.args.show_file_line = True
    StackAddressInterpreter.RunStackScript = mock.Mock(
      return_value=_EXAMPLE_STACK_SCRIPT_OUTPUT
    )
    self.assertEqual(
      _EXAMPLE_INTERPRETER_OUTPUT_WITH_FILE_NAME_LINE,
      self.stack_address_interpreter.Interpret(_ADDRESSES, _WEBVIEW_LIB_PATH),
    )
    self._AssertFileLines(mock_open, _EXAMPLE_STACK_SCRIPT_INPUT)

  def testSimplePerfRunner_CollectAddresses(self):
    addresses = self.simple_perf_runner.CollectAddresses(
      _MOCK_ORIGINAL_REPORT, 'libwebviewchromium.so'
    )
    self.assertEqual(set(_MOCK_ADDRESSES), addresses)

  def testSimplePerfRunner_ReplaceAddresses(self):
    postprocessed_report = (
      self.simple_perf_runner.ReplaceAddressesWithFunctionInfos(
        _MOCK_ORIGINAL_REPORT,
        _MOCK_ADDRESS_FUNCTION_NAME_PAIRS,
        'libwebviewchromium.so',
      )
    )
    self.assertEqual(_MOCK_FINAL_REPORT, postprocessed_report)

  @mock.patch('run_simpleperf.open', new_callable=mock.mock_open)
  def testSimplePerfRunner_Run(self, mock_open):
    self.stack_address_interpreter.Interpret = mock.Mock(
      return_value=_MOCK_ADDRESS_FUNCTION_NAME_PAIRS
    )

    SimplePerfRunner.RunSimplePerf = mock.Mock()
    SimplePerfRunner.RunPackageCompile = mock.Mock()
    SimplePerfRunner.GetOriginalReportHtml = mock.Mock(
      return_value=_MOCK_ORIGINAL_REPORT
    )
    self.simple_perf_runner.GetCurrentWebViewProvider = mock.Mock(
      return_value=_EXAMPLE_WEBVIEW_PACKAGE_NAME
    )

    self.simple_perf_runner.GetWebViewLibraryNameAndPath = mock.Mock(
      return_value=(_WEBVIEW_LIB_NAME, _WEBVIEW_LIB_PATH)
    )

    self.simple_perf_runner.Run()

    self._AssertFileLines(mock_open, _MOCK_FINAL_REPORT)


if __name__ == '__main__':
  unittest.main()