Index: build/android/devil/android/md5sum_test.py |
diff --git a/build/android/devil/android/md5sum_test.py b/build/android/devil/android/md5sum_test.py |
index 440305c51c648e12fea49c7638c1d4c4a3bf3a34..9ca1150aef03185b39092dcba972a8e794da4511 100755 |
--- a/build/android/devil/android/md5sum_test.py |
+++ b/build/android/devil/android/md5sum_test.py |
@@ -7,12 +7,11 @@ import os |
import sys |
import unittest |
+from devil import env |
from devil.android import device_errors |
from devil.android import md5sum |
-from pylib import constants |
-sys.path.append( |
- os.path.join(constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) |
+sys.path.append(env.dependencies.pymock_path) |
import mock # pylint: disable=import-error |
TEST_OUT_DIR = os.path.join('test', 'out', 'directory') |
@@ -22,8 +21,8 @@ class Md5SumTest(unittest.TestCase): |
def setUp(self): |
self._patchers = [ |
- mock.patch('pylib.constants.GetOutDirectory', |
- new=mock.Mock(return_value=TEST_OUT_DIR)), |
+ mock.patch('devil.env.binary_config.BinaryConfig.binary_dir', |
+ new=mock.PropertyMock(return_value=TEST_OUT_DIR)), |
mock.patch('os.path.exists', |
new=mock.Mock(return_value=True)), |
] |