Plumb network stack information about existence of cached copy

content/test/content_browser_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 "content/test/content_browser_test.h"
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 37 matching lines @@
   CHECK(PathService::Get(DIR_TEST_DATA, &content_test_data_dir));
   embedded_test_server()->ServeFilesFromDirectory(content_test_data_dir);
 }
 
 ContentBrowserTest::~ContentBrowserTest() {
   CHECK(setup_called_) << "Overridden SetUp() did not call parent "
                           "implementation, so test not run.";
 }
 
 void ContentBrowserTest::SetUp() {
-  setup_called_ = true;
-
   shell_main_delegate_.reset(new ShellMainDelegate);
   shell_main_delegate_->PreSandboxStartup();
 
   CommandLine* command_line = CommandLine::ForCurrentProcess();
   command_line->AppendSwitch(switches::kContentBrowserTest);
 
   SetUpCommandLine(command_line);
 
-  // Single-process mode is not set in BrowserMain, so process it explicitly,
-  // and set up renderer.
+  // Single-process mode is not set in BrowserMain, so if a subclass hasn't
+  // modified it yet, process it explicitly, and set up renderer.
   if (command_line->HasSwitch(switches::kSingleProcess)) {
-    single_process_renderer_client_.reset(new ShellContentRendererClient);
+    if (!single_process_renderer_client_)
+      single_process_renderer_client_.reset(new ShellContentRendererClient());
     SetRendererClientForTesting(single_process_renderer_client_.get());
+  } else {
+    // Confirm no test has called SetContentRendererClient() without
+    // setting up single process mode.
+    DCHECK(!single_process_renderer_client_);
   }
 
 #if defined(OS_MACOSX)
   // See InProcessBrowserTest::PrepareTestCommandLine().
   base::FilePath subprocess_path;
   PathService::Get(base::FILE_EXE, &subprocess_path);
   subprocess_path = subprocess_path.DirName().DirName();
   DCHECK_EQ(subprocess_path.BaseName().value(), "Contents");
   subprocess_path = subprocess_path.Append(
       "Frameworks/Content Shell Helper.app/Contents/MacOS/Content Shell Helper");
   command_line->AppendSwitchPath(switches::kBrowserSubprocessPath,
                                  subprocess_path);
 #endif
 
   // LinuxInputMethodContextFactory has to be initialized.
 #if !defined(OS_CHROMEOS) && defined(USE_AURA) && defined(USE_X11)
   ui::InitializeInputMethodForTesting();
 #endif
 
+  setup_called_ = true;
+
   BrowserTestBase::SetUp();
 }
 
 void ContentBrowserTest::TearDown() {
   BrowserTestBase::TearDown();
 
   // LinuxInputMethodContextFactory has to be shutdown.
 #if !defined(OS_CHROMEOS) && defined(USE_AURA) && defined(USE_X11)
   ui::ShutdownInputMethodForTesting();
 #endif
@@ skipping 35 matching lines @@
 #endif
 
   for (RenderProcessHost::iterator i(RenderProcessHost::AllHostsIterator());
        !i.IsAtEnd(); i.Advance()) {
     i.GetCurrentValue()->FastShutdownIfPossible();
   }
 
   Shell::CloseAllWindows();
 }
 
+void ContentBrowserTest::SetContentRendererClient(
+    scoped_ptr<ContentRendererClient> renderer_client) {
+  // This routine must be called before SetUp().
+  DCHECK(!setup_called_);
+  DCHECK(!single_process_renderer_client_);
+  single_process_renderer_client_ = renderer_client.Pass();
+}
+
 Shell* ContentBrowserTest::CreateBrowser() {
   return Shell::CreateNewWindow(
       ShellContentBrowserClient::Get()->browser_context(),
       GURL(kAboutBlankURL),
       NULL,
       MSG_ROUTING_NONE,
       gfx::Size());
 }
 
 Shell* ContentBrowserTest::CreateOffTheRecordBrowser() {
   return Shell::CreateNewWindow(
       ShellContentBrowserClient::Get()->off_the_record_browser_context(),
       GURL(kAboutBlankURL),
       NULL,
       MSG_ROUTING_NONE,
       gfx::Size());
 }
 
 }  // namespace content