Update pdfium_tests.py to run pdfium tests with Dr. Memory on buildbots

tools/drmemory/scripts/pdfium_tests.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 ''' Runs various chrome tests through valgrind_test.py.'''
 
 import glob
 import logging
 import optparse
@@ skipping 37 matching lines @@
                                           "and --test %s" % test)
 
     self._options = options
     self._args = args
 
     # Compute the top of the tree (the "source dir") from the script dir
     # (where this script lives).  We assume that the script dir is in
     # tools/drmemory/scripts relative to the top of the tree.
     script_dir = os.path.dirname(path_utils.ScriptDir())
     self._source_dir = os.path.dirname(os.path.dirname(script_dir))
+    # Setup Dr. Memory if it's not set up yet.
+    drmem_cmd = os.getenv("DRMEMORY_COMMAND")
+    if not drmem_cmd:
+      drmem_sfx = os.path.join(script_dir, "drmemory-windows-sfx.exe")
+      if not os.path.isfile(drmem_sfx):
+        raise RuntimeError, "Cannot find drmemory-windows-sfx.exe"
+      drmem_dir = os.path.join(script_dir, "unpacked")
+      subprocess.call([drmem_sfx, "-o" + drmem_dir, "-y"], 0)
+      drmem_cmd = os.path.join(drmem_dir, "bin", "drmemory.exe")
+      os.environ["DRMEMORY_COMMAND"] = drmem_cmd
     # since this path is used for string matching, make sure it's always
     # an absolute Unix-style path
     self._source_dir = os.path.abspath(self._source_dir).replace('\\', '/')
-    valgrind_test_script = os.path.join(script_dir, "valgrind_test.py")
     self._command_preamble = ["--source-dir=%s" % (self._source_dir)]
 
     if not self._options.build_dir:
       dirs = [
         os.path.join(self._source_dir, "xcodebuild", "Debug"),
         os.path.join(self._source_dir, "out", "Debug"),
         os.path.join(self._source_dir, "build", "Debug"),
       ]
       build_dir = [d for d in dirs if os.path.isdir(d)]
       if len(build_dir) > 1:
@@ skipping 314 matching lines @@
 
   for t in options.test:
     tests = ChromeTests(options, args, t)
     ret = tests.Run()
     if ret: return ret
   return 0
 
 
 if __name__ == "__main__":
   sys.exit(_main())