weburl_loader_mock.cc revision d0247b1b59f9c528cb6df88b4f2b9afaf80d181e 
1// Copyright 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "content/test/weburl_loader_mock.h"
6
7#include "base/logging.h"
8#include "content/test/weburl_loader_mock_factory.h"
9#include "third_party/WebKit/public/platform/WebData.h"
10#include "third_party/WebKit/public/platform/WebURLError.h"
11#include "third_party/WebKit/public/platform/WebURLLoaderClient.h"
12
13WebURLLoaderMock::WebURLLoaderMock(WebURLLoaderMockFactory* factory,
14                                   WebKit::WebURLLoader* default_loader)
15    : factory_(factory),
16      client_(NULL),
17      default_loader_(default_loader),
18      using_default_loader_(false),
19      is_deferred_(false) {
20}
21
22WebURLLoaderMock::~WebURLLoaderMock() {
23}
24
25void WebURLLoaderMock::ServeAsynchronousRequest(
26    const WebKit::WebURLResponse& response,
27    const WebKit::WebData& data,
28    const WebKit::WebURLError& error) {
29  DCHECK(!using_default_loader_);
30  if (!client_)
31    return;
32
33  client_->didReceiveResponse(this, response);
34
35  if (error.reason) {
36    client_->didFail(this, error);
37    return;
38  }
39  client_->didReceiveData(this, data.data(), data.size(), data.size());
40  client_->didFinishLoading(this, 0);
41}
42
43WebKit::WebURLRequest WebURLLoaderMock::ServeRedirect(
44    const WebKit::WebURLResponse& redirectResponse) {
45  WebKit::WebURLRequest newRequest;
46  newRequest.initialize();
47  GURL redirectURL(redirectResponse.httpHeaderField("Location"));
48  newRequest.setURL(redirectURL);
49  client_->willSendRequest(this, newRequest, redirectResponse);
50  return newRequest;
51}
52
53void WebURLLoaderMock::loadSynchronously(const WebKit::WebURLRequest& request,
54                                         WebKit::WebURLResponse& response,
55                                         WebKit::WebURLError& error,
56                                         WebKit::WebData& data) {
57  if (factory_->IsMockedURL(request.url())) {
58    factory_->LoadSynchronously(request, &response, &error, &data);
59    return;
60  }
61  DCHECK(static_cast<const GURL&>(request.url()).SchemeIs("data"))
62      << "loadSynchronously shouldn't be falling back: "
63      << request.url().spec().data();
64  using_default_loader_ = true;
65  default_loader_->loadSynchronously(request, response, error, data);
66}
67
68void WebURLLoaderMock::loadAsynchronously(const WebKit::WebURLRequest& request,
69                                          WebKit::WebURLLoaderClient* client) {
70  if (factory_->IsMockedURL(request.url())) {
71    client_ = client;
72    factory_->LoadAsynchronouly(request, this);
73    return;
74  }
75  DCHECK(static_cast<const GURL&>(request.url()).SchemeIs("data"))
76      << "loadAsynchronously shouldn't be falling back: "
77      << request.url().spec().data();
78  using_default_loader_ = true;
79  default_loader_->loadAsynchronously(request, client);
80}
81
82void WebURLLoaderMock::cancel() {
83  if (using_default_loader_) {
84    default_loader_->cancel();
85    return;
86  }
87  client_ = NULL;
88  factory_->CancelLoad(this);
89}
90
91void WebURLLoaderMock::setDefersLoading(bool deferred) {
92  is_deferred_ = deferred;
93  if (using_default_loader_) {
94    default_loader_->setDefersLoading(deferred);
95    return;
96  }
97  NOTIMPLEMENTED();
98}
99