Index: build/android/devil/utils/mock_calls.py |
diff --git a/build/android/devil/utils/mock_calls.py b/build/android/devil/utils/mock_calls.py |
index 59167bacbf1b047c9200e00490917bf02e2b67c4..28f8023ce795ff374fc611046003f66cd5870395 100644 |
--- a/build/android/devil/utils/mock_calls.py |
+++ b/build/android/devil/utils/mock_calls.py |
@@ -6,14 +6,13 @@ |
A test facility to assert call sequences while mocking their behavior. |
""" |
-import os |
import sys |
import unittest |
-from pylib import constants |
+from devil import devil_env |
+devil_env.InitializeForTests() |
-sys.path.append(os.path.join( |
- constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) |
+sys.path.append(devil_env.instance.pymock_path) |
import mock # pylint: disable=F0401 |