Service Worker: Change last update check location and HTTP cache bypass rule (2/2)

content/browser/service_worker/service_worker_write_to_cache_job_unittest.cc

 // Copyright 2014 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/location.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/thread_task_runner_handle.h"
 #include "content/browser/fileapi/mock_url_request_delegate.h"
 #include "content/browser/service_worker/embedded_worker_test_helper.h"
@@ skipping 42 matching lines @@
 net::URLRequestJob* CreateNormalURLRequestJob(
     net::URLRequest* request,
     net::NetworkDelegate* network_delegate) {
   return new net::URLRequestTestJob(request,
                                     network_delegate,
                                     std::string(kHeaders, arraysize(kHeaders)),
                                     kScriptCode,
                                     true);
 }
 
-net::URLRequestJob* CreateResponseJob(const std::string& response_data,
-                                      net::URLRequest* request,
-                                      net::NetworkDelegate* network_delegate) {
+net::URLRequestJob* CreateCachedResponseJob(
+    const std::string& response_data,
+    net::URLRequest* request,
+    net::NetworkDelegate* network_delegate) {
   return new net::URLRequestTestJob(request, network_delegate,
                                     std::string(kHeaders, arraysize(kHeaders)),
                                     response_data, true);
 }
 
+class NetworkResponseJob : public net::URLRequestTestJob {
+ public:
+  NetworkResponseJob(net::URLRequest* request,
+                     net::NetworkDelegate* network_delegate,
+                     const std::string& response_headers,
+                     const std::string& response_data,
+                     bool auto_advance)
+      : net::URLRequestTestJob(request,
+                               network_delegate,
+                               response_headers,
+                               response_data,
+                               auto_advance) {}
+  void GetResponseInfo(net::HttpResponseInfo* info) override {
+    URLRequestTestJob::GetResponseInfo(info);
+    info->network_accessed = true;
+  }
+
+ protected:
+  ~NetworkResponseJob() override {}
+};
+
+net::URLRequestJob* CreateNetworkResponseJob(
+    const std::string& response_data,
+    net::URLRequest* request,
+    net::NetworkDelegate* network_delegate) {
+  return new NetworkResponseJob(request, network_delegate,
+                                std::string(kHeaders, arraysize(kHeaders)),
+                                response_data, true);
+}
+
 net::URLRequestJob* CreateInvalidMimeTypeJob(
     net::URLRequest* request,
     net::NetworkDelegate* network_delegate) {
   const char kPlainTextHeaders[] =
       "HTTP/1.1 200 OK\0"
       "Content-Type: text/plain\0"
       "Expires: Thu, 1 Jan 2100 20:00:00 GMT\0"
       "\0";
   return new net::URLRequestTestJob(
       request,
       network_delegate,
       std::string(kPlainTextHeaders, arraysize(kPlainTextHeaders)),
       kScriptCode,
       true);
 }
 
+net::URLRequestJob* CreateErrorReponseJob(
+    net::URLRequest* request,
+    net::NetworkDelegate* network_delegate) {
+  const char kErrorHeaders[] = "HTTP/1.1 404 Not Found\0\0";
+  return new net::URLRequestTestJob(
+      request, network_delegate,
+      std::string(kErrorHeaders, arraysize(kErrorHeaders)), kScriptCode, true);
+}
+
 class SSLCertificateErrorJob : public net::URLRequestTestJob {
  public:
   SSLCertificateErrorJob(net::URLRequest* request,
                          net::NetworkDelegate* network_delegate,
                          const std::string& response_headers,
                          const std::string& response_data,
                          bool auto_advance)
       : net::URLRequestTestJob(request,
                                network_delegate,
                                response_headers,
@@ skipping 229 matching lines @@
     mock_protocol_handler_ = nullptr;
     version_ = nullptr;
     registration_ = nullptr;
     helper_.reset();
     // URLRequestJobs may post clean-up tasks on destruction.
     base::RunLoop().RunUntilIdle();
   }
 
   int CreateIncumbent(const std::string& response) {
     mock_protocol_handler_->SetCreateJobCallback(
-        base::Bind(&CreateResponseJob, response));
+        base::Bind(&CreateNetworkResponseJob, response));
     request_->Start();
     base::RunLoop().RunUntilIdle();
     EXPECT_EQ(net::URLRequestStatus::SUCCESS, request_->status().status());
     int incumbent_resource_id =
         version_->script_cache_map()->LookupResourceId(script_url_);
     EXPECT_NE(kInvalidServiceWorkerResponseId, incumbent_resource_id);
 
     registration_->SetActiveVersion(version_);
 
     // Teardown the request.
     request_.reset();
     url_request_context_.reset();
     url_request_job_factory_.reset();
     mock_protocol_handler_ = nullptr;
     base::RunLoop().RunUntilIdle();
 
     return incumbent_resource_id;
   }
 
   int GetResourceId(ServiceWorkerVersion* version) {
     return version->script_cache_map()->LookupResourceId(script_url_);
   }
 
-  // Performs the net request for an update of |registration_|'s incumbent
-  // to the script |response|. Returns the new version.
+  // Performs the net request for an update of |registration_|'s incumbent to
+  // the script |response|. Returns the new version.
+  // The types of the callbacks used for |callback|:
+  //  - CreateNetworkResponseJob: Network accessed, 200 OK
+  //  - CreateCachedResponseJob:  Served from HTTP cache
+  //  - CreateErrorReponseJob:    Network accessed, Non-2xx
   scoped_refptr<ServiceWorkerVersion> UpdateScript(
-      const std::string& response) {
+      const MockHttpProtocolHandler::JobCallback& callback) {
     int render_process_id = NextRenderProcessId();
     int provider_id = NextProviderId();
     scoped_refptr<ServiceWorkerVersion> new_version =
         new ServiceWorkerVersion(registration_.get(), script_url_,
                                  NextVersionId(), context()->AsWeakPtr());
     CreateHostForVersion(render_process_id, provider_id, new_version);
 
     SetUpScriptRequest(render_process_id, provider_id);
-    mock_protocol_handler_->SetCreateJobCallback(
-        base::Bind(&CreateResponseJob, response));
+    mock_protocol_handler_->SetCreateJobCallback(callback);
     request_->Start();
     base::RunLoop().RunUntilIdle();
     return new_version;
   }
 
   void VerifyResource(int id, const std::string& expected) {
     ASSERT_NE(kInvalidServiceWorkerResourceId, id);
     bool is_equal = false;
     scoped_ptr<ServiceWorkerResponseReader> reader =
         context()->storage()->CreateResponseReader(id);
@@ skipping 71 matching lines @@
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(net::URLRequestStatus::FAILED, request_->status().status());
   EXPECT_EQ(net::ERR_INSECURE_RESPONSE, request_->status().error());
   EXPECT_EQ(kInvalidServiceWorkerResponseId,
             version_->script_cache_map()->LookupResourceId(script_url_));
 }
 
 TEST_F(ServiceWorkerWriteToCacheJobTest, Update_SameScript) {
   std::string response = GenerateLongResponse();
   CreateIncumbent(response);
-  scoped_refptr<ServiceWorkerVersion> version = UpdateScript(response);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   EXPECT_EQ(kInvalidServiceWorkerResponseId, GetResourceId(version.get()));
 }
 
 TEST_F(ServiceWorkerWriteToCacheJobTest, Update_SameSizeScript) {
   std::string response = GenerateLongResponse();
   CreateIncumbent(response);
 
   // Change the first byte.
   response[0] = 'x';
-  scoped_refptr<ServiceWorkerVersion> version = UpdateScript(response);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Change something within the first block.
   response[5555] = 'x';
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Change something in a middle block.
   response[kMiddleBlock * kBlockSize + 111] = 'x';
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Change something within the last block.
   response[(kNumBlocks - 1) * kBlockSize] = 'x';
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Change the last byte.
   response[(kNumBlocks * kBlockSize) - 1] = 'x';
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 }
 
 TEST_F(ServiceWorkerWriteToCacheJobTest, Update_TruncatedScript) {
   std::string response = GenerateLongResponse();
   CreateIncumbent(response);
 
   // Truncate a single byte.
   response.resize(response.size() - 1);
-  scoped_refptr<ServiceWorkerVersion> version = UpdateScript(response);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Truncate to a middle block.
   response.resize((kMiddleBlock + 1) * kBlockSize + 111);
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Truncate to a block boundary.
   response.resize((kMiddleBlock - 1) * kBlockSize);
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Truncate to a single byte.
   response.resize(1);
-  version = UpdateScript(response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 }
 
 TEST_F(ServiceWorkerWriteToCacheJobTest, Update_ElongatedScript) {
   std::string original_response = GenerateLongResponse();
   CreateIncumbent(original_response);
 
   // Extend a single byte.
   std::string new_response = original_response + 'a';
-  scoped_refptr<ServiceWorkerVersion> version = UpdateScript(new_response);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateNetworkResponseJob, new_response));
   VerifyResource(GetResourceId(version.get()), new_response);
   registration_->SetWaitingVersion(version);
 
   // Extend multiple blocks.
   new_response = original_response + std::string(3 * kBlockSize, 'a');
-  version = UpdateScript(new_response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, new_response));
   VerifyResource(GetResourceId(version.get()), new_response);
   registration_->SetWaitingVersion(version);
 
   // Extend multiple blocks and bytes.
   new_response = original_response + std::string(7 * kBlockSize + 777, 'a');
-  version = UpdateScript(new_response);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, new_response));
   VerifyResource(GetResourceId(version.get()), new_response);
   registration_->SetWaitingVersion(version);
 }
 
 TEST_F(ServiceWorkerWriteToCacheJobTest, Update_EmptyScript) {
   // Create empty incumbent.
   CreateIncumbent(std::string());
 
   // Update from empty to non-empty.
   std::string response = GenerateLongResponse();
-  scoped_refptr<ServiceWorkerVersion> version = UpdateScript(response);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
   VerifyResource(GetResourceId(version.get()), response);
   registration_->SetWaitingVersion(version);
 
   // Update from non-empty to empty.
-  version = UpdateScript(std::string());
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, std::string()));
   VerifyResource(GetResourceId(version.get()), std::string());
   registration_->SetWaitingVersion(version);
 
   // Update from empty to empty.
-  version = UpdateScript(std::string());
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, std::string()));
   EXPECT_EQ(kInvalidServiceWorkerResponseId, GetResourceId(version.get()));
 }
 
+TEST_F(ServiceWorkerWriteToCacheJobTest, Update_BumpLastUpdateCheckTime) {
+  std::string response = GenerateLongResponse();
+  CreateIncumbent(response);
+
+  // Script was served from HTTPCache.
+  base::Time current = base::Time::Now();
+  registration_->set_last_update_check(current);
+  scoped_refptr<ServiceWorkerVersion> version =
+      UpdateScript(base::Bind(&CreateCachedResponseJob, response));
+  EXPECT_EQ(current, registration_->last_update_check());
+
+  // Same script but the request accessed the network.
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
+  EXPECT_LT(current, registration_->last_update_check());
+
+  // Changed the first byte and the request accessed the network.
+  response[0] = 'x';
+  current = base::Time::Now();
+  registration_->set_last_update_check(current);
+  version = UpdateScript(base::Bind(&CreateNetworkResponseJob, response));
+  VerifyResource(GetResourceId(version.get()), response);
+  registration_->SetWaitingVersion(version);
+  EXPECT_LT(current, registration_->last_update_check());
+
+  // The request accessed the network but received a non-2xx error.
+  current = base::Time::Now();
+  registration_->set_last_update_check(current);
+  version = UpdateScript(base::Bind(&CreateErrorReponseJob));
+  EXPECT_EQ(current, registration_->last_update_check());
+}
+
 }  // namespace content