[Android] Add a configurable environment for devil/.

build/android/devil/android/device_utils_test.py

 #!/usr/bin/env python
 # Copyright 2014 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.
 
 """
 Unit tests for the contents of device_utils.py (mostly DeviceUtils).
 """
 
 # pylint: disable=protected-access
 # pylint: disable=unused-argument
 
 import logging
-import os
 import sys
 import unittest
 
+from devil import devil_env
 from devil.android import device_errors
 from devil.android import device_signal
 from devil.android import device_utils
 from devil.android.sdk import adb_wrapper
 from devil.android.sdk import intent
 from devil.android.sdk import version_codes
 from devil.utils import cmd_helper
 from devil.utils import mock_calls
-from pylib import constants
 
-sys.path.append(os.path.join(
-    constants.DIR_SOURCE_ROOT, 'third_party', 'pymock'))
+devil_env.InitializeForTests()
+sys.path.append(devil_env.instance.pymock_path)
 import mock # pylint: disable=F0401
 
 
 class DeviceUtilsInitTest(unittest.TestCase):
 
   def testInitWithStr(self):
     serial_as_str = str('0123456789abcdef')
     d = device_utils.DeviceUtils('0123456789abcdef')
     self.assertEqual(serial_as_str, d.adb.GetDeviceSerial())
 
@@ skipping 10 matching lines @@
 
   def testInitWithMissing_fails(self):
     with self.assertRaises(ValueError):
       device_utils.DeviceUtils(None)
     with self.assertRaises(ValueError):
       device_utils.DeviceUtils('')
 
 
 class DeviceUtilsGetAVDsTest(mock_calls.TestCase):
 
+  @mock.patch('devil.devil_env._Environment.android_sdk_path',
+              mock.PropertyMock(return_value='/my/android/sdk'))
   def testGetAVDs(self):
     with self.assertCall(
         mock.call.devil.utils.cmd_helper.GetCmdOutput(
             [mock.ANY, 'list', 'avd']),
         'Available Android Virtual Devices:\n'
         '    Name: my_android5.0\n'
         '    Path: /some/path/to/.android/avd/my_android5.0.avd\n'
         '  Target: Android 5.0 (API level 21)\n'
         ' Tag/ABI: default/x86\n'
         '    Skin: WVGA800\n'):
-      self.assertEquals(['my_android5.0'],
-                        device_utils.GetAVDs())
+      self.assertEquals(['my_android5.0'], device_utils.GetAVDs())
 
 
 class DeviceUtilsRestartServerTest(mock_calls.TestCase):
 
   @mock.patch('time.sleep', mock.Mock())
   def testRestartServer_succeeds(self):
     with self.assertCalls(
         mock.call.devil.android.sdk.adb_wrapper.AdbWrapper.KillServer(),
         (mock.call.devil.utils.cmd_helper.GetCmdStatusAndOutput(
             ['pgrep', 'adb']),
@@ skipping 2011 matching lines @@
   def testGrantPermissions_BlackList(self):
     with self.patch_call(
         self.call.device.build_version_sdk, return_value=23):
         self.device.GrantPermissions(
             'package', ['android.permission.ACCESS_MOCK_LOCATION'])
 
 
 if __name__ == '__main__':
   logging.getLogger().setLevel(logging.DEBUG)
   unittest.main(verbosity=2)