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 35 matching lines @@
     }
   }
 };
 
 bool WriteValueToFile(std::string value, const FilePath& path) {
   int retval = file_util::WriteFile(path, value.c_str(), value.length());
   return retval == static_cast<int>(value.length());
 }
 
 // To actually do anything useful, this test should have a url
 // passed on the command line, eg.

[jrt, 2011/06/17 19:50:46]

Why does this test use command line arguments?

 //
 // --url=http://foo.bar.com
 //
 // Additional arguments:
 //
 // --wait_cookie_name=<name>
 //   Waits for a cookie named <name> to be set before exiting successfully.
 //
 // --wait_cookie_value=<value>
 //   In conjunction with --wait_cookie_name, this waits for a specific value
@@ skipping 69 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