Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 import os | |
| 2 import unittest | |
| 3 | |
| 4 from telemetry.core import util | |
| 5 from telemetry.internal.util import binary_manager | |
| 6 | |
| 7 util.AddDirToPythonPath(util.GetTelemetryDir(), 'third_party', 'mock') | |
|
eakuefner
2015/08/10 19:57:23
Note: as crrev.com/1276173004 has landed, this is
aiolos (Not reviewing)
2015/08/10 20:37:27
Done.
| |
| 8 import mock | |
| 9 | |
| 10 class BinaryManagerTest(unittest.TestCase): | |
| 11 | |
| 12 def testInitializationNoEnvironmentConfig(self): | |
| 13 manager = mock.MagicMock() | |
| 14 binary_manager.dependency_manager = manager.dependency_manager | |
| 15 expected = [mock.call.dependency_manager.DependencyManager( | |
| 16 [binary_manager.TELEMETRY_PROJECT_CONFIG])] | |
| 17 binary_manager.InitDependencyManager(None) | |
| 18 self.assertEqual(manager.mock_calls, expected) | |
| 19 | |
| 20 def testInitializationWithEnvironmentConfig(self): | |
| 21 environment_config = os.path.join('some', 'config', 'path') | |
| 22 manager = mock.MagicMock() | |
| 23 binary_manager.dependency_manager = manager.dependency_manager | |
| 24 expected = [mock.call.dependency_manager.DependencyManager( | |
| 25 [binary_manager.TELEMETRY_PROJECT_CONFIG, environment_config])] | |
| 26 binary_manager.InitDependencyManager(environment_config) | |
| 27 self.assertEqual(manager.mock_calls, expected) | |
| 28 | |
| 29 def testReinitialization(self): | |
| 30 manager = mock.MagicMock() | |
| 31 binary_manager.dependency_manager = manager.dependency_manager | |
| 32 binary_manager.InitDependencyManager(None) | |
| 33 self.assertRaises(binary_manager.InitializationError, | |
| 34 binary_manager.InitDependencyManager, None) | |
| 35 | |
| 36 def testFetchPathInitialized(self): | |
| 37 manager = mock.MagicMock() | |
| 38 binary_manager.dependency_manager = manager.dependency_manager | |
| 39 binary_manager.support_binaries = manager.support_binaries | |
| 40 expected = [mock.call.dependency_manager.DependencyManager( | |
| 41 [binary_manager.TELEMETRY_PROJECT_CONFIG]), | |
| 42 mock.call.support_binaries.FindPath('dep', 'plat', 'arch')] | |
| 43 binary_manager.InitDependencyManager(None) | |
| 44 binary_manager.FetchPath('dep', 'plat', 'arch') | |
| 45 self.assertEqual(manager.mock_calls, expected) | |
| 46 #TODO(aiolos): We should be switching over to using the dependency_manager | |
| 47 #insead of the support binaries, and update the tests at that time. | |
| 48 | |
| 49 def testFetchPathUninitialized(self): | |
| 50 manager = mock.MagicMock() | |
| 51 binary_manager.dependency_manager = manager.dependency_manager | |
| 52 binary_manager.support_binaries = manager.support_binaries | |
| 53 self.assertRaises(binary_manager.InitializationError, | |
| 54 binary_manager.FetchPath, 'dep', 'plat', 'arch') | |
| 55 | |
| 56 def testLocalPathInitialized(self): | |
| 57 manager = mock.MagicMock() | |
| 58 binary_manager.dependency_manager = manager.dependency_manager | |
| 59 binary_manager.support_binaries = manager.support_binaries | |
| 60 expected = [mock.call.dependency_manager.DependencyManager( | |
| 61 [binary_manager.TELEMETRY_PROJECT_CONFIG]), | |
| 62 mock.call.support_binaries.FindLocallyBuiltPath('dep')] | |
| 63 binary_manager.InitDependencyManager(None) | |
| 64 binary_manager.LocalPath('dep', 'plat', 'arch') | |
| 65 self.assertEqual(manager.mock_calls, expected) | |
| 66 #TODO(aiolos): We should be switching over to using the dependency_manager | |
| 67 #insead of the support binaries, and update the tests at that time. | |
| 68 | |
| 69 def testLocalPathUninitialized(self): | |
| 70 manager = mock.MagicMock() | |
| 71 binary_manager.dependency_manager = manager.dependency_manager | |
| 72 binary_manager.support_binaries = manager.support_binaries | |
| 73 self.assertRaises(binary_manager.InitializationError, | |
| 74 binary_manager.LocalPath, 'dep', 'plat', 'arch') | |
| 75 | |
| OLD | NEW |