Add support for mocking failed loads to WebURLLoaderMock.

webkit/support/weburl_loader_mock.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 "webkit/support/weburl_loader_mock.h"
 
 #include "base/logging.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebData.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURLError.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURLLoaderClient.h"
 #include "webkit/support/weburl_loader_mock_factory.h"
 
 WebURLLoaderMock::WebURLLoaderMock(WebURLLoaderMockFactory* factory,
                                    WebKit::WebURLLoader* default_loader)
     : factory_(factory),
       client_(NULL),
       default_loader_(default_loader),
       using_default_loader_(false),
       is_deferred_(false) {
 }
 
 WebURLLoaderMock::~WebURLLoaderMock() {
 }
 
 void WebURLLoaderMock::ServeAsynchronousRequest(
     const WebKit::WebURLResponse& response,
     const WebKit::WebData& data,
     const WebKit::WebURLError& error) {
   DCHECK(!using_default_loader_);
   if (!client_)
     return;
 
   client_->didReceiveResponse(this, response);
+
+  if (error.reason) {
+    client_->didFail(this, error);
+    return;
+  }
   client_->didReceiveData(this, data.data(), data.size(), data.size());
   client_->didFinishLoading(this, 0);
 }
 
 WebKit::WebURLRequest WebURLLoaderMock::ServeRedirect(
     const WebKit::WebURLResponse& redirectResponse) {
   WebKit::WebURLRequest newRequest;
   newRequest.initialize();
   GURL redirectURL(redirectResponse.httpHeaderField("Location"));
   newRequest.setURL(redirectURL);
@@ skipping 34 matching lines @@
 }
 
 void WebURLLoaderMock::setDefersLoading(bool deferred) {
   is_deferred_ = deferred;
   if (using_default_loader_) {
     default_loader_->setDefersLoading(deferred);
     return;
   }
   NOTIMPLEMENTED();
 }