OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/sync_driver/sync_stopped_reporter.h" | 5 #include "components/sync/driver/sync_stopped_reporter.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
11 #include "base/run_loop.h" | 11 #include "base/run_loop.h" |
12 #include "base/test/test_simple_task_runner.h" | 12 #include "base/test/test_simple_task_runner.h" |
13 #include "base/threading/non_thread_safe.h" | 13 #include "base/threading/non_thread_safe.h" |
14 #include "components/sync/protocol/sync.pb.h" | 14 #include "components/sync/protocol/sync.pb.h" |
15 #include "net/http/http_status_code.h" | 15 #include "net/http/http_status_code.h" |
(...skipping 14 matching lines...) Expand all Loading... |
30 const char kBirthday[] = "2263"; | 30 const char kBirthday[] = "2263"; |
31 | 31 |
32 const char kAuthHeaderPrefix[] = "Bearer "; | 32 const char kAuthHeaderPrefix[] = "Bearer "; |
33 | 33 |
34 class SyncStoppedReporterTest : public testing::Test { | 34 class SyncStoppedReporterTest : public testing::Test { |
35 public: | 35 public: |
36 SyncStoppedReporterTest() {} | 36 SyncStoppedReporterTest() {} |
37 ~SyncStoppedReporterTest() override {} | 37 ~SyncStoppedReporterTest() override {} |
38 | 38 |
39 void SetUp() override { | 39 void SetUp() override { |
40 request_context_ = new net::TestURLRequestContextGetter( | 40 request_context_ = |
41 message_loop_.task_runner()); | 41 new net::TestURLRequestContextGetter(message_loop_.task_runner()); |
42 } | 42 } |
43 | 43 |
44 void RequestFinishedCallback(const SyncStoppedReporter::Result& result) { | 44 void RequestFinishedCallback(const SyncStoppedReporter::Result& result) { |
45 request_result_ = result; | 45 request_result_ = result; |
46 } | 46 } |
47 | 47 |
48 GURL test_url() { | 48 GURL test_url() { return GURL(kTestURL); } |
49 return GURL(kTestURL); | |
50 } | |
51 | 49 |
52 std::string user_agent() const { | 50 std::string user_agent() const { return std::string(kTestUserAgent); } |
53 return std::string(kTestUserAgent); | |
54 } | |
55 | 51 |
56 SyncStoppedReporter::ResultCallback callback() { | 52 SyncStoppedReporter::ResultCallback callback() { |
57 return base::Bind(&SyncStoppedReporterTest::RequestFinishedCallback, | 53 return base::Bind(&SyncStoppedReporterTest::RequestFinishedCallback, |
58 base::Unretained(this)); | 54 base::Unretained(this)); |
59 } | 55 } |
60 | 56 |
61 const SyncStoppedReporter::Result& request_result() const { | 57 const SyncStoppedReporter::Result& request_result() const { |
62 return request_result_; | 58 return request_result_; |
63 } | 59 } |
64 | 60 |
65 net::URLRequestContextGetter* request_context() { | 61 net::URLRequestContextGetter* request_context() { |
66 return request_context_.get(); | 62 return request_context_.get(); |
67 } | 63 } |
68 | 64 |
69 private: | 65 private: |
70 base::MessageLoop message_loop_; | 66 base::MessageLoop message_loop_; |
71 scoped_refptr<net::URLRequestContextGetter> request_context_; | 67 scoped_refptr<net::URLRequestContextGetter> request_context_; |
72 SyncStoppedReporter::Result request_result_; | 68 SyncStoppedReporter::Result request_result_; |
73 | 69 |
74 DISALLOW_COPY_AND_ASSIGN(SyncStoppedReporterTest); | 70 DISALLOW_COPY_AND_ASSIGN(SyncStoppedReporterTest); |
75 }; | 71 }; |
76 | 72 |
77 // Test that the event URL gets constructed correctly. | 73 // Test that the event URL gets constructed correctly. |
78 TEST_F(SyncStoppedReporterTest, EventURL) { | 74 TEST_F(SyncStoppedReporterTest, EventURL) { |
79 net::TestURLFetcherFactory factory; | 75 net::TestURLFetcherFactory factory; |
80 SyncStoppedReporter ssr(GURL(kTestURL), user_agent(), | 76 SyncStoppedReporter ssr(GURL(kTestURL), user_agent(), request_context(), |
81 request_context(), callback()); | 77 callback()); |
82 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 78 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
83 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 79 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
84 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); | 80 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); |
85 } | 81 } |
86 | 82 |
87 // Test that the event URL gets constructed correctly with a trailing slash. | 83 // Test that the event URL gets constructed correctly with a trailing slash. |
88 TEST_F(SyncStoppedReporterTest, EventURLWithSlash) { | 84 TEST_F(SyncStoppedReporterTest, EventURLWithSlash) { |
89 net::TestURLFetcherFactory factory; | 85 net::TestURLFetcherFactory factory; |
90 SyncStoppedReporter ssr(GURL(kTestURLTrailingSlash), user_agent(), | 86 SyncStoppedReporter ssr(GURL(kTestURLTrailingSlash), user_agent(), |
91 request_context(), callback()); | 87 request_context(), callback()); |
92 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 88 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
93 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 89 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
94 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); | 90 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); |
95 } | 91 } |
96 | 92 |
97 // Test that the URLFetcher gets configured correctly. | 93 // Test that the URLFetcher gets configured correctly. |
98 TEST_F(SyncStoppedReporterTest, FetcherConfiguration) { | 94 TEST_F(SyncStoppedReporterTest, FetcherConfiguration) { |
99 net::TestURLFetcherFactory factory; | 95 net::TestURLFetcherFactory factory; |
100 SyncStoppedReporter ssr(test_url(), user_agent(), | 96 SyncStoppedReporter ssr(test_url(), user_agent(), request_context(), |
101 request_context(), callback()); | 97 callback()); |
102 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 98 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
103 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 99 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
104 | 100 |
105 // Ensure the headers are set correctly. | 101 // Ensure the headers are set correctly. |
106 net::HttpRequestHeaders headers; | 102 net::HttpRequestHeaders headers; |
107 std::string header; | 103 std::string header; |
108 fetcher->GetExtraRequestHeaders(&headers); | 104 fetcher->GetExtraRequestHeaders(&headers); |
109 headers.GetHeader(net::HttpRequestHeaders::kAuthorization, &header); | 105 headers.GetHeader(net::HttpRequestHeaders::kAuthorization, &header); |
110 std::string auth_header(kAuthHeaderPrefix); | 106 std::string auth_header(kAuthHeaderPrefix); |
111 auth_header.append(kAuthToken); | 107 auth_header.append(kAuthToken); |
112 EXPECT_EQ(auth_header, header); | 108 EXPECT_EQ(auth_header, header); |
113 headers.GetHeader(net::HttpRequestHeaders::kUserAgent, &header); | 109 headers.GetHeader(net::HttpRequestHeaders::kUserAgent, &header); |
114 EXPECT_EQ(user_agent(), header); | 110 EXPECT_EQ(user_agent(), header); |
115 | 111 |
116 sync_pb::EventRequest event_request; | 112 sync_pb::EventRequest event_request; |
117 event_request.ParseFromString(fetcher->upload_data()); | 113 event_request.ParseFromString(fetcher->upload_data()); |
118 | 114 |
119 EXPECT_EQ(kCacheGuid, event_request.sync_disabled().cache_guid()); | 115 EXPECT_EQ(kCacheGuid, event_request.sync_disabled().cache_guid()); |
120 EXPECT_EQ(kBirthday, event_request.sync_disabled().store_birthday()); | 116 EXPECT_EQ(kBirthday, event_request.sync_disabled().store_birthday()); |
121 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); | 117 EXPECT_EQ(kEventURL, fetcher->GetOriginalURL().spec()); |
122 } | 118 } |
123 | 119 |
124 TEST_F(SyncStoppedReporterTest, HappyCase) { | 120 TEST_F(SyncStoppedReporterTest, HappyCase) { |
125 net::TestURLFetcherFactory factory; | 121 net::TestURLFetcherFactory factory; |
126 SyncStoppedReporter ssr(test_url(), user_agent(), | 122 SyncStoppedReporter ssr(test_url(), user_agent(), request_context(), |
127 request_context(), callback()); | 123 callback()); |
128 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 124 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
129 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 125 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
130 fetcher->set_response_code(net::HTTP_OK); | 126 fetcher->set_response_code(net::HTTP_OK); |
131 ssr.OnURLFetchComplete(fetcher); | 127 ssr.OnURLFetchComplete(fetcher); |
132 base::RunLoop run_loop; | 128 base::RunLoop run_loop; |
133 run_loop.RunUntilIdle(); | 129 run_loop.RunUntilIdle(); |
134 EXPECT_EQ(SyncStoppedReporter::RESULT_SUCCESS, request_result()); | 130 EXPECT_EQ(SyncStoppedReporter::RESULT_SUCCESS, request_result()); |
135 } | 131 } |
136 | 132 |
137 TEST_F(SyncStoppedReporterTest, ServerNotFound) { | 133 TEST_F(SyncStoppedReporterTest, ServerNotFound) { |
138 net::TestURLFetcherFactory factory; | 134 net::TestURLFetcherFactory factory; |
139 SyncStoppedReporter ssr(test_url(), user_agent(), | 135 SyncStoppedReporter ssr(test_url(), user_agent(), request_context(), |
140 request_context(), callback()); | 136 callback()); |
141 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 137 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
142 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 138 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
143 fetcher->set_response_code(net::HTTP_NOT_FOUND); | 139 fetcher->set_response_code(net::HTTP_NOT_FOUND); |
144 ssr.OnURLFetchComplete(fetcher); | 140 ssr.OnURLFetchComplete(fetcher); |
145 base::RunLoop run_loop; | 141 base::RunLoop run_loop; |
146 run_loop.RunUntilIdle(); | 142 run_loop.RunUntilIdle(); |
147 EXPECT_EQ(SyncStoppedReporter::RESULT_ERROR, request_result()); | 143 EXPECT_EQ(SyncStoppedReporter::RESULT_ERROR, request_result()); |
148 } | 144 } |
149 | 145 |
150 TEST_F(SyncStoppedReporterTest, DestructionDuringRequestHandler) { | 146 TEST_F(SyncStoppedReporterTest, DestructionDuringRequestHandler) { |
151 net::TestURLFetcherFactory factory; | 147 net::TestURLFetcherFactory factory; |
152 factory.set_remove_fetcher_on_delete(true); | 148 factory.set_remove_fetcher_on_delete(true); |
153 { | 149 { |
154 SyncStoppedReporter ssr(test_url(), user_agent(), | 150 SyncStoppedReporter ssr(test_url(), user_agent(), request_context(), |
155 request_context(), callback()); | 151 callback()); |
156 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 152 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
157 EXPECT_FALSE(factory.GetFetcherByID(0) == nullptr); | 153 EXPECT_FALSE(factory.GetFetcherByID(0) == nullptr); |
158 } | 154 } |
159 EXPECT_TRUE(factory.GetFetcherByID(0) == nullptr); | 155 EXPECT_TRUE(factory.GetFetcherByID(0) == nullptr); |
160 } | 156 } |
161 | 157 |
162 TEST_F(SyncStoppedReporterTest, Timeout) { | 158 TEST_F(SyncStoppedReporterTest, Timeout) { |
163 SyncStoppedReporter ssr(test_url(), user_agent(), | 159 SyncStoppedReporter ssr(test_url(), user_agent(), request_context(), |
164 request_context(), callback()); | 160 callback()); |
165 | 161 |
166 // A task runner that can trigger the timeout immediately. | 162 // A task runner that can trigger the timeout immediately. |
167 scoped_refptr<base::TestSimpleTaskRunner> task_runner( | 163 scoped_refptr<base::TestSimpleTaskRunner> task_runner( |
168 new base::TestSimpleTaskRunner()); | 164 new base::TestSimpleTaskRunner()); |
169 ssr.SetTimerTaskRunnerForTest(task_runner); | 165 ssr.SetTimerTaskRunnerForTest(task_runner); |
170 | 166 |
171 // Begin request. | 167 // Begin request. |
172 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 168 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
173 | 169 |
174 // Trigger the timeout. | 170 // Trigger the timeout. |
(...skipping 24 matching lines...) Expand all Loading... |
199 new base::TestSimpleTaskRunner()); | 195 new base::TestSimpleTaskRunner()); |
200 ssr.SetTimerTaskRunnerForTest(task_runner); | 196 ssr.SetTimerTaskRunnerForTest(task_runner); |
201 | 197 |
202 // Begin request. | 198 // Begin request. |
203 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); | 199 ssr.ReportSyncStopped(kAuthToken, kCacheGuid, kBirthday); |
204 | 200 |
205 // Trigger the timeout. | 201 // Trigger the timeout. |
206 ASSERT_TRUE(task_runner->HasPendingTask()); | 202 ASSERT_TRUE(task_runner->HasPendingTask()); |
207 task_runner->RunPendingTasks(); | 203 task_runner->RunPendingTasks(); |
208 } | 204 } |
OLD | NEW |