1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// 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/macros.h"
#include "base/memory/scoped_ptr.h"
#include "base/run_loop.h"
#include "content/browser/webui/url_data_manager_backend.h"
#include "content/public/test/mock_resource_context.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "net/http/http_response_headers.h"
#include "net/http/http_response_info.h"
#include "net/url_request/url_request_context.h"
#include "net/url_request/url_request_job.h"
#include "net/url_request/url_request_job_factory_impl.h"
#include "net/url_request/url_request_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace content {
class CancelAfterFirstReadURLRequestDelegate : public net::TestDelegate {
public:
CancelAfterFirstReadURLRequestDelegate() {}
~CancelAfterFirstReadURLRequestDelegate() override {}
void OnResponseStarted(net::URLRequest* request) override {
// net::TestDelegate will start the first read.
TestDelegate::OnResponseStarted(request);
request->Cancel();
}
void OnReadCompleted(net::URLRequest* request, int bytes_read) override {
// Read should have been cancelled.
EXPECT_EQ(-1, bytes_read);
}
private:
DISALLOW_COPY_AND_ASSIGN(CancelAfterFirstReadURLRequestDelegate);
};
class UrlDataManagerBackendTest : public testing::Test {
public:
UrlDataManagerBackendTest() {
// URLRequestJobFactory takes ownership of the passed in ProtocolHandler.
url_request_job_factory_.SetProtocolHandler(
"chrome", URLDataManagerBackend::CreateProtocolHandler(
&resource_context_, false, nullptr));
url_request_context_.set_job_factory(&url_request_job_factory_);
}
scoped_ptr<net::URLRequest> CreateRequest(net::URLRequest::Delegate* delegate,
const char* origin) {
scoped_ptr<net::URLRequest> request = url_request_context_.CreateRequest(
GURL("chrome://resources/polymer/v1_0/polymer/polymer-extracted.js"),
net::HIGHEST, delegate);
request->SetExtraRequestHeaderByName("Origin", origin, true);
return request;
}
protected:
TestBrowserThreadBundle thread_bundle_;
MockResourceContext resource_context_;
net::URLRequestJobFactoryImpl url_request_job_factory_;
net::URLRequestContext url_request_context_;
net::TestDelegate delegate_;
};
TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginChromeUrl) {
scoped_ptr<net::URLRequest> request(
CreateRequest(&delegate_, "chrome://webui"));
request->Start();
base::RunLoop().RunUntilIdle();
EXPECT_TRUE(request->response_headers()->HasHeaderValue(
"Access-Control-Allow-Origin", "chrome://webui"));
}
TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginNonChromeUrl) {
scoped_ptr<net::URLRequest> request(
CreateRequest(&delegate_, "http://www.example.com"));
request->Start();
base::RunLoop().RunUntilIdle();
EXPECT_TRUE(request->response_headers()->HasHeaderValue(
"Access-Control-Allow-Origin", "null"));
}
// Check that the URLRequest isn't passed headers after cancellation.
TEST_F(UrlDataManagerBackendTest, CancelBeforeResponseStarts) {
scoped_ptr<net::URLRequest> request(
CreateRequest(&delegate_, "chrome://webui"));
request->Start();
request->Cancel();
base::RunLoop().RunUntilIdle();
EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status());
EXPECT_EQ(1, delegate_.response_started_count());
}
// Check that the URLRequest isn't passed data after cancellation.
TEST_F(UrlDataManagerBackendTest, CancelAfterFirstReadStarted) {
CancelAfterFirstReadURLRequestDelegate cancel_delegate;
scoped_ptr<net::URLRequest> request(
CreateRequest(&cancel_delegate, "chrome://webui"));
request->Start();
base::RunLoop().RunUntilIdle();
EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status());
EXPECT_EQ(1, cancel_delegate.response_started_count());
EXPECT_EQ("", cancel_delegate.data_received());
}
// Check for a network error page request via chrome://network-error/.
TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequest) {
scoped_ptr<net::URLRequest> error_request =
url_request_context_.CreateRequest(
GURL("chrome://network-error/-105"), net::HIGHEST, &delegate_);
error_request->Start();
base::RunLoop().Run();
EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status());
EXPECT_EQ(net::ERR_NAME_NOT_RESOLVED, error_request->status().error());
}
// Check for an invalid network error page request via chrome://network-error/.
TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequestFailed) {
scoped_ptr<net::URLRequest> error_request =
url_request_context_.CreateRequest(
GURL("chrome://network-error/-123456789"), net::HIGHEST, &delegate_);
error_request->Start();
base::RunLoop().Run();
EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status());
EXPECT_EQ(net::ERR_INVALID_URL, error_request->status().error());
}
} // namespace content
|
