catapult/telemetry/telemetry/internal/util/binary_manager_unittest.py

# 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 os
import unittest

from telemetry.core import exceptions
from telemetry.internal.util import binary_manager
import mock


class BinaryManagerTest(unittest.TestCase):
  def setUp(self):
    # We need to preserve the real initialized dependecny_manager.
    self.actual_binary_manager = binary_manager._binary_manager
    binary_manager._binary_manager = None

  def tearDown(self):
    binary_manager._binary_manager = self.actual_binary_manager

  @mock.patch(
      'telemetry.internal.util.binary_manager.binary_manager.BinaryManager') # pylint: disable=line-too-long
  @mock.patch(
      'telemetry.internal.util.binary_manager.base_config.BaseConfig')
  def testInitializationNoEnvironmentConfig(
      self, base_config_mock, binary_manager_mock):
    base_config_mock.side_effect = ['base_config_object1',
                                    'base_config_object2',
                                    'base_config_object3']
    binary_manager.InitDependencyManager(None)
    base_config_mock.assert_has_calls([
        mock.call.base_config.BaseConfig(
            binary_manager.TELEMETRY_PROJECT_CONFIG),
        mock.call.base_config.BaseConfig(
            binary_manager.CHROME_BINARY_CONFIG)])
    self.assertEqual(2, base_config_mock.call_count)
    binary_manager_mock.assert_called_once_with(['base_config_object1',
                                              'base_config_object2'])

  @mock.patch(
      'telemetry.internal.util.binary_manager.binary_manager.BinaryManager') # pylint: disable=line-too-long
  @mock.patch(
      'telemetry.internal.util.binary_manager.base_config.BaseConfig')
  def testInitializationWithEnvironmentConfig(
      self, base_config_mock, binary_manager_mock):
    base_config_mock.side_effect = ['base_config_object1',  # TELEMETRY_PROJECT
                                    'base_config_object2',  # CHROME_BINARY
                                    'base_config_object3']  # ENVIRONMENT
    environment_config = os.path.join('some', 'config', 'path')
    binary_manager.InitDependencyManager(environment_config)
    expected_calls = [mock.call(binary_manager.TELEMETRY_PROJECT_CONFIG),
                      mock.call(binary_manager.CHROME_BINARY_CONFIG),
                      mock.call(environment_config)]
    self.assertEqual(expected_calls, base_config_mock.call_args_list)
    # Make sure the environment config is passed first.
    binary_manager_mock.assert_called_once_with(
        ['base_config_object3', 'base_config_object1', 'base_config_object2'])

  def testReinitialization(self):
    binary_manager.InitDependencyManager(None)
    self.assertRaises(exceptions.InitializationError,
                      binary_manager.InitDependencyManager, None)

  @mock.patch(
      'telemetry.internal.util.binary_manager.binary_manager.BinaryManager') # pylint: disable=line-too-long
  @mock.patch(
      'telemetry.internal.util.binary_manager.base_config.BaseConfig')
  def testFetchPathInitialized(self, base_config_mock, binary_manager_mock):
    base_config_mock.return_value = 'base_config_object'
    expected = [mock.call.binary_manager.BinaryManager(
                   ['base_config_object']),
                mock.call.binary_manager.BinaryManager().FetchPath(
                    'dep', 'plat_arch')]
    binary_manager.InitDependencyManager(None)
    binary_manager.FetchPath('dep', 'plat', 'arch')
    binary_manager_mock.assert_call_args(expected)
    base_config_mock.assert_has_calls([
        mock.call.base_config.BaseConfig(
            binary_manager.TELEMETRY_PROJECT_CONFIG),
        mock.call.base_config.BaseConfig(
            binary_manager.CHROME_BINARY_CONFIG)])
    self.assertEqual(2, base_config_mock.call_count)

  def testFetchPathUninitialized(self):
    self.assertRaises(exceptions.InitializationError,
                      binary_manager.FetchPath, 'dep', 'plat', 'arch')

  @mock.patch(
      'telemetry.internal.util.binary_manager.binary_manager.BinaryManager') # pylint: disable=line-too-long
  @mock.patch(
      'telemetry.internal.util.binary_manager.base_config.BaseConfig')
  def testLocalPathInitialized(self, base_config_mock, binary_manager_mock):
    base_config_mock.return_value = 'base_config_object'
    expected = [mock.call.binary_manager.BinaryManager(
                   ['base_config_object']),
                mock.call.binary_manager.BinaryManager().LocalPath(
                    'dep', 'plat_arch')]
    binary_manager.InitDependencyManager(None)
    binary_manager.LocalPath('dep', 'plat', 'arch')
    binary_manager_mock.assert_call_args(expected)
    base_config_mock.assert_has_calls([
        mock.call.base_config.BaseConfig(
            binary_manager.TELEMETRY_PROJECT_CONFIG),
        mock.call.base_config.BaseConfig(
            binary_manager.CHROME_BINARY_CONFIG)])
    self.assertEqual(2, base_config_mock.call_count)

  def testLocalPathUninitialized(self):
    self.assertRaises(exceptions.InitializationError,
                      binary_manager.LocalPath, 'dep', 'plat', 'arch')