Remove hard dependency on bundled python_26 in tests

net/test/spawned_test_server/local_test_server_win.cc

 // 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.
 
 #include "net/test/spawned_test_server/local_test_server.h"
 
 #include <windows.h>
 #include <wincrypt.h>
 
 #include "base/base_paths.h"
 #include "base/bind.h"
 #include "base/command_line.h"
+#include "base/environment.h"
 #include "base/files/file_path.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/process/launch.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/thread.h"
 #include "base/win/scoped_handle.h"
@@ skipping 51 matching lines @@
   thread.Stop();
   // If the timeout kicked in, abort.
   if (unblocked) {
     LOG(ERROR) << "Timeout exceeded for ReadData";
     return false;
   }
 
   return true;
 }
 
+// Class that sets up a temporary path that includes the supplied path
+// at the end.
+//
+// TODO(bratell): By making this more generic we can possibly reuse
+//                it at other places such as
+//                chrome/common/multi_process_lock_unittest.cc.
+class ScopedPath {
+ public:
+  // Constructor which sets up the environment to include the path to
+  // |path_to_add|.
+  explicit ScopedPath(const base::FilePath& path_to_add);
+
+  // Destructor that restores the path that were active when the
+  // object was constructed.
+  ~ScopedPath();
+
+ private:
+  // The PATH environment variable before it was changed or an empty
+  // string if there was no PATH environment variable.
+  std::string old_path_;
+
+  // The helper object that allows us to read and set environment
+  // variables more easily.
+  scoped_ptr<base::Environment> environment_;
+
+  // A flag saying if we have actually modified the environment.
+  bool path_modified_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScopedPath);
+};
+
+ScopedPath::ScopedPath(const base::FilePath& path_to_add)
+    : environment_(base::Environment::Create()),
+      path_modified_(false) {
+  environment_->GetVar("PATH", &old_path_);
+
+  std::string new_value = old_path_;
+  if (!new_value.empty())
+    new_value += ";";
+
+  new_value += WideToUTF8(path_to_add.value());
+
+  path_modified_ = environment_->SetVar("PATH", new_value);
+}
+
+ScopedPath::~ScopedPath() {
+  if (!path_modified_)
+    return;
+  if (old_path_.empty())
+    environment_->UnSetVar("PATH");
+  else
+    environment_->SetVar("PATH", old_path_);
+}
+
 }  // namespace
 
 namespace net {
 
 bool LocalTestServer::LaunchPython(const base::FilePath& testserver_path) {
   CommandLine python_command(CommandLine::NO_PROGRAM);
   if (!GetPythonCommand(&python_command))
     return false;
 
   python_command.AppendArgPath(testserver_path);
@@ skipping 33 matching lines @@
   if (!job_handle_.IsValid()) {
     LOG(ERROR) << "Could not create JobObject.";
     return false;
   }
 
   if (!base::SetJobObjectAsKillOnJobClose(job_handle_.Get())) {
     LOG(ERROR) << "Could not SetInformationJobObject.";
     return false;
   }
 
+  // Add our internal python to the path so it can be used if there is
+  // no system python.
+  base::FilePath python_dir;
+  if (!PathService::Get(base::DIR_SOURCE_ROOT, &python_dir)) {
+    LOG(ERROR) << "Could not locate source root directory.";
+    return false;
+  }
+  python_dir = python_dir.AppendASCII("third_party").AppendASCII("python_26");
+  ScopedPath python_path(python_dir);
   base::LaunchOptions launch_options;
   launch_options.inherit_handles = true;
   launch_options.job_handle = job_handle_.Get();
   if (!base::LaunchProcess(python_command, launch_options, &process_handle_)) {
     LOG(ERROR) << "Failed to launch " << python_command.GetCommandLineString();
     return false;
   }
 
   return true;
 }
@@ skipping 19 matching lines @@
   if (!ParseServerData(server_data)) {
     LOG(ERROR) << "Could not parse server_data: " << server_data;
     return false;
   }
 
   return true;
 }
 
 }  // namespace net