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

chrome/test/functional/media/pyauto_media.py

 #!/usr/bin/python
 
 # Copyright (c) 2011 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.
 
 """Setup for PyAuto media tests.
 
 Use the following in your scripts to run them standalone:
 
 # This should be at the top
 import pyauto_media
 
 if __name__ == '__main__':
   pyauto_media.Main()
 
 This script looks similar to pyauto_functional.py. However, unlike
 pyauto_functional, this script can NOT be used as an executable to
 fire off other media scripts since media tests require additional
 parameters to be set in the form of environment variables (unless you
 want these variables to be defaults, which is generally not a good
 idea).
 """
 
 import os
 import sys
+import tempfile
+
+from media_test_env_names import MediaTestEnvNames
 
 
 def _SetupPaths():
   """Setting path to find pyauto_functional.py."""
   media_dir = os.path.abspath(os.path.dirname(__file__))
   sys.path.append(media_dir)
   sys.path.append(os.path.normpath(os.path.join(media_dir, os.pardir)))
   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'))

[Nirnimesh, 2011/05/23 17:52:44]

why hardcode to chrome-linux?

[imasaki1, 2011/05/23 18:58:47]

This is just test directory, it will be change later. I am adding TODO.

[Nirnimesh, 2011/05/23 20:13:11]

Even if it's experimental or temporary, it seems like a bad idea to hardcode the
platform. How about chrome-media-test?

[imasaki1, 2011/05/23 20:26:23]

Done.

+    sys.path.insert(0, reference_build_dir)
 
 _SetupPaths()
 
 import pyauto_functional
 import pyauto
 
 
 class Main(pyauto_functional.Main):
   """Main program for running PyAuto media tests."""
 
   def __init__(self):
     pyauto_functional.Main.__init__(self)
 
 
 if __name__ == '__main__':
   Main()