Support Chromium-branded builds and cleanup reference build usage.

chrome/test/url_fetch_test/url_fetch_test.cc

 // 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.
 
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/test/test_timeouts.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/test/automation/tab_proxy.h"
-#include "chrome/test/ui/ui_test.h"
+#include "chrome/test/ui/ui_perf_test.h"
 
 namespace {
 
 // Provides a UI Test that lets us take the browser to a url, and
 // wait for a cookie value to be set or a JavaScript expression to evaluate
 // true before closing the page. It is undefined what happens if you specify
 // both a cookie and a JS expression.
-class UrlFetchTest : public UITest {
+class UrlFetchTest : public UIPerfTest {
  public:
   UrlFetchTest() {
     show_window_ = true;
     dom_automation_enabled_ = true;
   }
   struct UrlFetchTestResult {
     std::string cookie_value;
     std::string javascript_variable;
   };
 
   void SetUp() {
     const CommandLine* cmd_line = CommandLine::ForCurrentProcess();
     if (cmd_line->HasSwitch("reference_build")) {
-      FilePath dir;
-      PathService::Get(chrome::DIR_TEST_TOOLS, &dir);
-      dir = dir.AppendASCII("reference_build");
-#if defined(OS_WIN)
-      dir = dir.AppendASCII("chrome");
-#elif defined(OS_LINUX)
-      dir = dir.AppendASCII("chrome_linux");
-#elif defined(OS_MACOSX)
-      dir = dir.AppendASCII("chrome_mac");
-#endif
-      browser_directory_ = dir;
+      UseReferenceBuild();
     }
     UITest::SetUp();
   }
 
   void RunTest(const GURL& url,
                const char* wait_cookie_name,
                const char* wait_cookie_value,
                const char* var_to_fetch,
                const std::string& wait_js_expr,
                const std::string& wait_js_frame_xpath,
@@ skipping 125 matching lines @@
 
   // Write out the JS Variable if requested
   FilePath jsvar_output_path = cmd_line->GetSwitchValuePath("jsvar_output");
   if (!jsvar_output_path.value().empty()) {
     ASSERT_TRUE(WriteValueToFile(result.javascript_variable,
                                  jsvar_output_path));
   }
 }
 
 }  // namespace