Prerender: Confirm ServiceWorkers are invoked for NoState Prefetch

chrome/browser/prerender/prerender_nostate_prefetch_browsertest.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 "base/command_line.h"
 #include "base/strings/string16.h"
 #include "base/strings/string_split.h"
+#include "base/strings/utf_string_conversions.h"
 #include "base/task_scheduler/post_task.h"
 #include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/history/history_test_utils.h"
 #include "chrome/browser/prerender/prerender_handle.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_manager_factory.h"
 #include "chrome/browser/prerender/prerender_test_utils.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/task_manager/task_manager_browsertest_util.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/grit/generated_resources.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/service_worker_context.h"
+#include "content/public/browser/storage_partition.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/test/browser_test_utils.h"
 #include "net/base/escape.h"
 #include "net/base/load_flags.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/request_handler_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using prerender::test_utils::DestructionWaiter;
 using prerender::test_utils::RequestCounter;
 using prerender::test_utils::TestPrerender;
 
+namespace {
+
+void ClosureOnUiThread(const base::Closure& callback) {
+  content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE,
+                                   callback);
+}
+
+}  // namespace
+
 namespace prerender {
 
 // These URLs used for test resources must be relative with the exception of
 // |kPrefetchLoaderPath|.
 const char kPrefetchImagePage[] = "prerender/prefetch_image.html";
 const char kPrefetchJpeg[] = "prerender/image.jpeg";
 const char kPrefetchLoaderPath[] = "/prerender/prefetch_loader.html";
 const char kPrefetchLoopPage[] = "prerender/prefetch_loop.html";
 const char kPrefetchMetaCSP[] = "prerender/prefetch_meta_csp.html";
 const char kPrefetchPage[] = "prerender/prefetch_page.html";
 const char kPrefetchPage2[] = "prerender/prefetch_page2.html";
 const char kPrefetchPng[] = "prerender/image.png";
 const char kPrefetchResponseHeaderCSP[] =
     "prerender/prefetch_response_csp.html";
 const char kPrefetchScript[] = "prerender/prefetch.js";
 const char kPrefetchScript2[] = "prerender/prefetch2.js";
+const char kServiceWorkerLoader[] = "prerender/service_worker.html";
 const char kPrefetchSubresourceRedirectPage[] =
     "prerender/prefetch_subresource_redirect.html";
 
 class NoStatePrefetchBrowserTest
     : public test_utils::PrerenderInProcessBrowserTest {
  public:
   NoStatePrefetchBrowserTest() {}
 
   void SetUpOnMainThread() override {
     test_utils::PrerenderInProcessBrowserTest::SetUpOnMainThread();
@@ skipping 467 matching lines @@
         constexpr net::RequestPriority kExpectedPriority = net::HIGHEST;
 #else
         constexpr net::RequestPriority kExpectedPriority = net::IDLE;
 #endif
         EXPECT_EQ(kExpectedPriority, request->priority());
       }));
   PrefetchFromFile(kPrefetchPage, FINAL_STATUS_NOSTATE_PREFETCH_FINISHED);
   script_counter.WaitForCount(1);
 }
 
+// Checks that a registered ServiceWorker (SW) without a renderer intercepts a

[falken, 2017/01/06 16:17:55]

nit: Does "without a renderer" mean the same thing as "not currently running"?
That'd be more common service worker terminology.

[mattcary, 2017/01/09 14:45:42]

Done.

+// prefetch request.
+IN_PROC_BROWSER_TEST_F(NoStatePrefetchBrowserTest, ServiceWorkerIntercept) {
+  // Register and launch a SW.
+  base::string16 expected_title = base::ASCIIToUTF16("SW READY");
+  content::TitleWatcher title_watcher(GetActiveWebContents(), expected_title);
+  ui_test_utils::NavigateToURL(
+      current_browser(),
+      src_server()->GetURL(MakeAbsolute(kServiceWorkerLoader)));
+  EXPECT_EQ(expected_title, title_watcher.WaitAndGetTitle());
+
+  // Stop the SW renderer process.

[falken, 2017/01/06 16:17:55]

Ditto, I think rather than "stop the renderer process" this is saying "Stop the
SW."

+  base::RunLoop run_loop;
+  content::StoragePartition* storage_partation =

[falken, 2017/01/06 16:17:55]

nit: typo for "partition"

[mattcary, 2017/01/09 14:45:42]

Done.

+      content::BrowserContext::GetStoragePartitionForSite(
+          GetActiveWebContents()->GetBrowserContext(),
+          src_server()->GetURL(MakeAbsolute(kPrefetchPng)));
+  storage_partation->GetServiceWorkerContext()
+      ->StopAllServiceWorkersForOriginWithCallback(
+          src_server()->GetURL("/"),
+          base::Bind(&ClosureOnUiThread, run_loop.QuitClosure()));
+  run_loop.Run();
+
+  // The SW intercepts kPrefetchPage and replaces it with a body that contains
+  // an <img> tage for kPrefetchPng. This verifies that the SW ran correctly by
+  // observing the fetch of the image.
+  RequestCounter image_counter;
+  CountRequestFor(kPrefetchPng, &image_counter);
+  PrefetchFromFile(kPrefetchPage, FINAL_STATUS_NOSTATE_PREFETCH_FINISHED);
+  image_counter.WaitForCount(1);
+}
+
 }  // namespace prerender