# Copyright 2016 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 json import os from pyfakefs import fake_filesystem_unittest from dependency_manager import exceptions from py_utils import binary_manager class BinaryManagerTest(fake_filesystem_unittest.TestCase): # TODO(aiolos): disable cloud storage use during this test. def setUp(self): self.setUpPyfakefs() # pylint: disable=bad-continuation self.expected_dependencies = { 'dep_1': { 'cloud_storage_base_folder': 'dependencies/fake_config', 'cloud_storage_bucket': 'chrome-tel', 'file_info': { 'linux_x86_64': { 'cloud_storage_hash': '661ce936b3276f7ec3d687ab62be05b96d796f21', 'download_path': 'bin/linux/x86_64/dep_1' }, 'mac_x86_64': { 'cloud_storage_hash': 'c7b1bfc6399dc683058e88dac1ef0f877edea74b', 'download_path': 'bin/mac/x86_64/dep_1' }, 'win_AMD64': { 'cloud_storage_hash': 'ac4fee89a51662b9d920bce443c19b9b2929b198', 'download_path': 'bin/win/AMD64/dep_1.exe' }, 'win_x86': { 'cloud_storage_hash': 'e246e183553ea26967d7b323ea269e3357b9c837', 'download_path': 'bin/win/x86/dep_1.exe' } } }, 'dep_2': { 'cloud_storage_base_folder': 'dependencies/fake_config', 'cloud_storage_bucket': 'chrome-tel', 'file_info': { 'linux_x86_64': { 'cloud_storage_hash': '13a57efae9a680ac0f160b3567e02e81f4ac493c', 'download_path': 'bin/linux/x86_64/dep_2', 'local_paths': [ '../../example/location/linux/dep_2', '../../example/location2/linux/dep_2' ] }, 'mac_x86_64': { 'cloud_storage_hash': 'd10c0ddaa8586b20449e951216bee852fa0f8850', 'download_path': 'bin/mac/x86_64/dep_2', 'local_paths': [ '../../example/location/mac/dep_2', '../../example/location2/mac/dep_2' ] }, 'win_AMD64': { 'cloud_storage_hash': 'fd5b417f78c7f7d9192a98967058709ded1d399d', 'download_path': 'bin/win/AMD64/dep_2.exe', 'local_paths': [ '../../example/location/win64/dep_2', '../../example/location2/win64/dep_2' ] }, 'win_x86': { 'cloud_storage_hash': 'cf5c8fe920378ce30d057e76591d57f63fd31c1a', 'download_path': 'bin/win/x86/dep_2.exe', 'local_paths': [ '../../example/location/win32/dep_2', '../../example/location2/win32/dep_2' ] }, 'android_k_x64': { 'cloud_storage_hash': '09177be2fed00b44df0e777932828425440b23b3', 'download_path': 'bin/android/x64/k/dep_2.apk', 'local_paths': [ '../../example/location/android_x64/k/dep_2', '../../example/location2/android_x64/k/dep_2' ] }, 'android_l_x64': { 'cloud_storage_hash': '09177be2fed00b44df0e777932828425440b23b3', 'download_path': 'bin/android/x64/l/dep_2.apk', 'local_paths': [ '../../example/location/android_x64/l/dep_2', '../../example/location2/android_x64/l/dep_2' ] }, 'android_k_x86': { 'cloud_storage_hash': 'bcf02af039713a48b69b89bd7f0f9c81ed8183a4', 'download_path': 'bin/android/x86/k/dep_2.apk', 'local_paths': [ '../../example/location/android_x86/k/dep_2', '../../example/location2/android_x86/k/dep_2' ] }, 'android_l_x86': { 'cloud_storage_hash': '12a74cec071017ba11655b5740b8a58e2f52a219', 'download_path': 'bin/android/x86/l/dep_2.apk', 'local_paths': [ '../../example/location/android_x86/l/dep_2', '../../example/location2/android_x86/l/dep_2' ] } } }, 'dep_3': { 'file_info': { 'linux_x86_64': { 'local_paths': [ '../../example/location/linux/dep_3', '../../example/location2/linux/dep_3' ] }, 'mac_x86_64': { 'local_paths': [ '../../example/location/mac/dep_3', '../../example/location2/mac/dep_3' ] }, 'win_AMD64': { 'local_paths': [ '../../example/location/win64/dep_3', '../../example/location2/win64/dep_3' ] }, 'win_x86': { 'local_paths': [ '../../example/location/win32/dep_3', '../../example/location2/win32/dep_3' ] } } } } # pylint: enable=bad-continuation fake_config = { 'config_type': 'BaseConfig', 'dependencies': self.expected_dependencies } self.base_config = os.path.join(os.path.dirname(__file__), 'example_config.json') self.fs.CreateFile(self.base_config, contents=json.dumps(fake_config)) linux_file = os.path.join( os.path.dirname(self.base_config), os.path.join('..', '..', 'example', 'location2', 'linux', 'dep_2')) android_file = os.path.join( os.path.dirname(self.base_config), '..', '..', 'example', 'location', 'android_x86', 'l', 'dep_2') self.expected_dep2_linux_file = os.path.abspath(linux_file) self.expected_dep2_android_file = os.path.abspath(android_file) self.fs.CreateFile(self.expected_dep2_linux_file) self.fs.CreateFile(self.expected_dep2_android_file) def tearDown(self): self.tearDownPyfakefs() def testInitializationNoConfig(self): with self.assertRaises(ValueError): binary_manager.BinaryManager(None) def testInitializationMissingConfig(self): with self.assertRaises(ValueError): binary_manager.BinaryManager(os.path.join('missing', 'path')) def testInitializationWithConfig(self): with self.assertRaises(ValueError): manager = binary_manager.BinaryManager(self.base_config) manager = binary_manager.BinaryManager([self.base_config]) self.assertItemsEqual(self.expected_dependencies, manager._dependency_manager._lookup_dict) def testSuccessfulFetchPathNoOsVersion(self): manager = binary_manager.BinaryManager([self.base_config]) found_path = manager.FetchPath('dep_2', 'linux', 'x86_64') self.assertEqual(self.expected_dep2_linux_file, found_path) def testSuccessfulFetchPathOsVersion(self): manager = binary_manager.BinaryManager([self.base_config]) found_path = manager.FetchPath('dep_2', 'android', 'x86', 'l') self.assertEqual(self.expected_dep2_android_file, found_path) def testSuccessfulFetchPathFallbackToNoOsVersion(self): manager = binary_manager.BinaryManager([self.base_config]) found_path = manager.FetchPath('dep_2', 'linux', 'x86_64', 'fake_version') self.assertEqual(self.expected_dep2_linux_file, found_path) def testFailedFetchPathMissingDep(self): manager = binary_manager.BinaryManager([self.base_config]) with self.assertRaises(exceptions.NoPathFoundError): manager.FetchPath('missing_dep', 'linux', 'x86_64') with self.assertRaises(exceptions.NoPathFoundError): manager.FetchPath('missing_dep', 'android', 'x86', 'l') with self.assertRaises(exceptions.NoPathFoundError): manager.FetchPath('dep_1', 'linux', 'bad_arch') with self.assertRaises(exceptions.NoPathFoundError): manager.FetchPath('dep_1', 'bad_os', 'x86') def testSuccessfulLocalPathNoOsVersion(self): manager = binary_manager.BinaryManager([self.base_config]) found_path = manager.LocalPath('dep_2', 'linux', 'x86_64') self.assertEqual(self.expected_dep2_linux_file, found_path) def testSuccessfulLocalPathOsVersion(self): manager = binary_manager.BinaryManager([self.base_config]) found_path = manager.LocalPath('dep_2', 'android', 'x86', 'l') self.assertEqual(self.expected_dep2_android_file, found_path)