Change for preparation for reference-build on perfbot (media)

chrome/test/functional/media/pyauto_media.py

Index: chrome/test/functional/media/pyauto_media.py
diff --git a/chrome/test/functional/media/pyauto_media.py b/chrome/test/functional/media/pyauto_media.py
index c40598df88fdbe0c72d37bc7b1eb6cc275843261..27d73c9436eca300d0278f015deb7a802ccc4d21 100644
--- a/chrome/test/functional/media/pyauto_media.py
+++ b/chrome/test/functional/media/pyauto_media.py
@@ -24,6 +24,9 @@ idea).
 
 import os
 import sys
+import tempfile
+
+from media_test_env_names import MediaTestEnvNames
 
 
 def _SetupPaths():
@@ -34,6 +37,14 @@ def _SetupPaths():
   sys.path.append(os.path.normpath(os.path.join(
       media_dir, os.pardir, os.pardir, os.pardir, os.pardir,
       'third_party', 'psutil')))
+  # Setting PYTHONPATH for reference build.
+  if os.getenv(MediaTestEnvNames.REFERENCE_BUILD_ENV_NAME):
+    reference_build_dir = os.getenv(
+        MediaTestEnvNames.REFERENCE_BUILD_DIR_ENV_NAME,
+        # Default dir is just for testing so this dir must
+        # be set in the build script.
+        os.path.join(tempfile.gettempdir(), 'chrome-linux'))
+    sys.path.insert(0, reference_build_dir)
 
 _SetupPaths()