Fallback override

webkit/appcache/appcache_request_handler_unittest.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 <stack>
 #include <string>
 #include <vector>
 
 #include "base/message_loop.h"
 #include "base/threading/thread.h"
 #include "base/synchronization/waitable_event.h"
 #include "net/base/net_errors.h"
+#include "net/http/http_response_headers.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_error_job.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/appcache/appcache.h"
 #include "webkit/appcache/appcache_backend_impl.h"
 #include "webkit/appcache/appcache_request_handler.h"
 #include "webkit/appcache/appcache_url_request_job.h"
 #include "webkit/appcache/mock_appcache_service.h"
 
 namespace appcache {
@@ skipping 38 matching lines @@
     virtual void Run() {
       test_->SetUpTest();
       (test_->*method_)();
     }
 
    private:
     AppCacheRequestHandlerTest* test_;
     Method method_;
   };
 
-  // Subclasses to simulate particular response codes so test cases can
+  // Subclasses to simulate particular responses so test cases can
   // exercise fallback code paths.
 
+  class MockURLRequestDelegate : public net::URLRequest::Delegate {
+    virtual void OnResponseStarted(net::URLRequest* request) {}
+    virtual void OnReadCompleted(net::URLRequest* request, int bytes_read) {}
+  };
+
   class MockURLRequestJob : public net::URLRequestJob {
    public:
-    MockURLRequestJob(net::URLRequest* request, int response_code)
-        : net::URLRequestJob(request), response_code_(response_code) {}
-    virtual void Start() {}
-    virtual int GetResponseCode() const { return response_code_; }
+    MockURLRequestJob(
+        net::URLRequest* request, int response_code)
+        : net::URLRequestJob(request),
+          response_code_(response_code),
+          has_response_info_(false) {}
+    MockURLRequestJob(
+        net::URLRequest* request, const net::HttpResponseInfo& info)
+        : net::URLRequestJob(request),
+          response_code_(0),
+          has_response_info_(true),
+          response_info_(info) {}
+    virtual void Start() {
+      NotifyHeadersComplete();
+    }
+    virtual int GetResponseCode() const {
+      if (!has_response_info_)

[jennb, 2011/05/10 00:13:53]

You could initialize response_code_ to info.headers->response_code() to keep
this method simpler.

[michaeln, 2011/05/10 00:41:58]

Done.

+        return response_code_;
+      return response_info_.headers->response_code();
+    }
+    virtual void GetResponseInfo(net::HttpResponseInfo* info) {
+      if (!has_response_info_)
+        return;
+      *info = response_info_;
+    }
     int response_code_;
+    bool has_response_info_;
+    net::HttpResponseInfo response_info_;
   };
 
   class MockURLRequest : public net::URLRequest {
    public:
     explicit MockURLRequest(const GURL& url) : net::URLRequest(url, NULL) {}
 
     void SimulateResponseCode(int http_response_code) {
       mock_factory_job_ = new MockURLRequestJob(this, http_response_code);
       Start();
       DCHECK(!mock_factory_job_);
-      // All our simulation need to do satisfy are the following two DCHECKs
+      // All our simulation needs  to satisfy are the following two DCHECKs
       DCHECK(status().is_success());
       DCHECK_EQ(http_response_code, GetResponseCode());
     }
+
+    void SimulateResponseInfo(const net::HttpResponseInfo& info) {
+      mock_factory_job_ = new MockURLRequestJob(this, info);
+      set_delegate(&delegate_);  // needed to get the info back out
+      Start();
+      DCHECK(!mock_factory_job_);
+    }
+
+    MockURLRequestDelegate delegate_;
   };
 
   static net::URLRequestJob* MockHttpJobFactory(net::URLRequest* request,
                                                 const std::string& scheme) {
     if (mock_factory_job_) {
       net::URLRequestJob* temp = mock_factory_job_;
       mock_factory_job_ = NULL;
       return temp;
     } else {
       // Some of these tests trigger UpdateJobs which start URLRequests.
@@ skipping 216 matching lines @@
     EXPECT_EQ(GURL("http://blah/manifest/"), manifest_url);
     EXPECT_TRUE(host_->main_resource_was_fallback_);
     EXPECT_EQ(GURL("http://blah/fallbackurl"), host_->fallback_url_);
 
     EXPECT_EQ(GURL("http://blah/manifest/"),
               host_->preferred_manifest_url());
 
     TestFinished();
   }
 
+  // MainResource_FallbackOverride --------------------------------------------
+
+  void MainResource_FallbackOverride() {
+    PushNextTask(NewRunnableMethod(
+       this,
+       &AppCacheRequestHandlerTest::Verify_MainResource_FallbackOverride));
+
+    request_.reset(new MockURLRequest(GURL("http://blah/fallback-override")));
+    handler_.reset(host_->CreateRequestHandler(request_.get(),
+                                               ResourceType::MAIN_FRAME));
+    EXPECT_TRUE(handler_.get());
+
+    mock_storage()->SimulateFindMainResource(
+        AppCacheEntry(),
+        GURL("http://blah/fallbackurl"),
+        AppCacheEntry(AppCacheEntry::EXPLICIT, 1),
+        1, GURL("http://blah/manifest/"));
+
+    job_ = handler_->MaybeLoadResource(request_.get());
+    EXPECT_TRUE(job_.get());
+    EXPECT_TRUE(job_->is_waiting());
+
+    // We have to wait for completion of storage->FindResponseForMainRequest.
+    ScheduleNextTask();
+  }
+
+  void Verify_MainResource_FallbackOverride() {
+    EXPECT_FALSE(job_->is_waiting());
+    EXPECT_TRUE(job_->is_delivering_network_response());
+
+    // When the request is restarted, the existing job is dropped so a
+    // real network job gets created. We expect NULL here which will cause
+    // the net library to create a real job.
+    job_ = handler_->MaybeLoadResource(request_.get());
+    EXPECT_FALSE(job_);
+
+    // Simulate an http error of the real network job, but with custom
+    // headers that override the fallback behavior.
+    const char kOverrideHeaders[] =
+        "HTTP/1.1 404 BOO HOO\0"
+        "x-chromium-appcache-fallback-override: disallow-fallback\0"
+        "\0";
+    net::HttpResponseInfo info;
+    info.headers = new net::HttpResponseHeaders(
+        std::string(kOverrideHeaders, arraysize(kOverrideHeaders)));
+    request_->SimulateResponseInfo(info);
+
+    job_ = handler_->MaybeLoadFallbackForResponse(request_.get());
+    EXPECT_FALSE(job_);
+
+    int64 cache_id = kNoCacheId;

[jennb, 2011/05/10 00:13:53]

These checks seem unnecessary as we've already asserted that !job_ in line 416.

[michaeln, 2011/05/10 00:41:58]

removed

+    GURL manifest_url;
+    handler_->GetExtraResponseInfo(&cache_id, &manifest_url);
+    EXPECT_EQ(kNoCacheId, cache_id);
+    EXPECT_TRUE(manifest_url.is_empty());
+
+    TestFinished();
+  }
+
   // SubResource_Miss_WithNoCacheSelected ----------------------------------
 
   void SubResource_Miss_WithNoCacheSelected() {
     request_.reset(new MockURLRequest(GURL("http://blah/")));
     handler_.reset(host_->CreateRequestHandler(request_.get(),
                                                ResourceType::SUB_RESOURCE));
 
     // We avoid creating handler when possible, sub-resource requests are not
     // subject to retrieval from an appcache when there's no associated cache.
     EXPECT_FALSE(handler_.get());
@@ skipping 349 matching lines @@
 }
 
 TEST_F(AppCacheRequestHandlerTest, MainResource_Hit) {
   RunTestOnIOThread(&AppCacheRequestHandlerTest::MainResource_Hit);
 }
 
 TEST_F(AppCacheRequestHandlerTest, MainResource_Fallback) {
   RunTestOnIOThread(&AppCacheRequestHandlerTest::MainResource_Fallback);
 }
 
+TEST_F(AppCacheRequestHandlerTest, MainResource_FallbackOverride) {
+  RunTestOnIOThread(
+      &AppCacheRequestHandlerTest::MainResource_FallbackOverride);
+}
+
 TEST_F(AppCacheRequestHandlerTest, SubResource_Miss_WithNoCacheSelected) {
   RunTestOnIOThread(
       &AppCacheRequestHandlerTest::SubResource_Miss_WithNoCacheSelected);
 }
 
 TEST_F(AppCacheRequestHandlerTest, SubResource_Miss_WithCacheSelected) {
   RunTestOnIOThread(
       &AppCacheRequestHandlerTest::SubResource_Miss_WithCacheSelected);
 }
 
@@ skipping 38 matching lines @@
 
 TEST_F(AppCacheRequestHandlerTest, WorkerRequest) {
   RunTestOnIOThread(&AppCacheRequestHandlerTest::WorkerRequest);
 }
 
 }  // namespace appcache
 
 // AppCacheRequestHandlerTest is expected to always live longer than the
 // runnable methods.  This lets us call NewRunnableMethod on its instances.
 DISABLE_RUNNABLE_METHOD_REFCOUNT(appcache::AppCacheRequestHandlerTest);