Add a test to cover service worker update codepath when extension is updated

content/browser/service_worker/service_worker_context_wrapper.cc

 // Copyright 2013 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/browser/service_worker/service_worker_context_wrapper.h"
 
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
@@ skipping 314 matching lines @@
     usage_info.scopes.push_back(registration_info.pattern);
     usage_info.total_size_bytes += registration_info.stored_version_size_bytes;
   }
 
   for (const auto& origin_info_pair : origins) {
     usage_infos.push_back(origin_info_pair.second);
   }
   callback.Run(usage_infos);
 }
 
+void ServiceWorkerContextWrapper::DidGetRegistrationsForForceUpdateForOrigin(
+    const std::vector<scoped_refptr<ServiceWorkerRegistration>>&
+        registrations) {
+  for (const auto& registration : registrations)
+    registration->set_force_update_on_page_load(true);
+}
+
 void ServiceWorkerContextWrapper::DidCheckHasServiceWorker(
     const CheckHasServiceWorkerCallback& callback,
     bool has_service_worker) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
                           base::Bind(callback, has_service_worker));
 }
 
 void ServiceWorkerContextWrapper::StopAllServiceWorkersForOrigin(
@@ skipping 59 matching lines @@
     BrowserThread::PostTask(
         BrowserThread::IO,
         FROM_HERE,
         base::Bind(result, false));
     return;
   }
   context()->UnregisterServiceWorkers(
       origin.GetOrigin(), base::Bind(&StatusCodeToBoolCallbackAdapter, result));
 }
 
+void ServiceWorkerContextWrapper::SetForceUpdateOnPageLoadForOrigin(
+    const GURL& origin) {
+  if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) {
+    BrowserThread::PostTask(
+        BrowserThread::IO, FROM_HERE,
+        base::Bind(
+            &ServiceWorkerContextWrapper::SetForceUpdateOnPageLoadForOrigin,
+            this, origin));
+    return;
+  }
+
+  if (!context_core_)
+    return;
+
+  context_core_->storage()->GetRegistrationsForOrigin(
+      origin.GetOrigin(),
+      base::Bind(&ServiceWorkerContextWrapper::
+                     DidGetRegistrationsForForceUpdateForOrigin,
+                 this));
+}
+
 void ServiceWorkerContextWrapper::CheckHasServiceWorker(
     const GURL& url,
     const GURL& other_url,
     const CheckHasServiceWorkerCallback& callback) {
   if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) {
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
         base::Bind(&ServiceWorkerContextWrapper::CheckHasServiceWorker, this,
                    url, other_url, callback));
     return;
@@ skipping 268 matching lines @@
   observer_list_->Notify(FROM_HERE,
                          &ServiceWorkerContextObserver::OnStorageWiped);
 }
 
 ServiceWorkerContextCore* ServiceWorkerContextWrapper::context() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   return context_core_.get();
 }
 
 }  // namespace content