| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 "base/message_loop/message_loop.h" | |
| 6 #include "base/run_loop.h" | |
| 7 #include "chrome/browser/chromeos/geolocation/simple_geolocation_provider.h" | |
| 8 #include "net/http/http_response_headers.h" | |
| 9 #include "net/http/http_status_code.h" | |
| 10 #include "net/url_request/test_url_fetcher_factory.h" | |
| 11 #include "net/url_request/url_fetcher_impl.h" | |
| 12 #include "net/url_request/url_request_status.h" | |
| 13 #include "testing/gtest/include/gtest/gtest.h" | |
| 14 | |
| 15 namespace { | |
| 16 | |
| 17 const int kRequestRetryIntervalMilliSeconds = 200; | |
| 18 | |
| 19 // This should be different from default to prevent SimpleGeolocationRequest | |
| 20 // from modifying it. | |
| 21 const char kTestGeolocationProviderUrl[] = | |
| 22 "https://localhost/geolocation/v1/geolocate?"; | |
| 23 | |
| 24 const char kSimpleResponseBody[] = | |
| 25 "{\n" | |
| 26 " \"location\": {\n" | |
| 27 " \"lat\": 51.0,\n" | |
| 28 " \"lng\": -0.1\n" | |
| 29 " },\n" | |
| 30 " \"accuracy\": 1200.4\n" | |
| 31 "}"; | |
| 32 } // anonymous namespace | |
| 33 | |
| 34 namespace chromeos { | |
| 35 | |
| 36 // This is helper class for net::FakeURLFetcherFactory. | |
| 37 class TestGeolocationAPIURLFetcherCallback { | |
| 38 public: | |
| 39 TestGeolocationAPIURLFetcherCallback(const GURL& url, | |
| 40 const size_t require_retries, | |
| 41 const std::string& response, | |
| 42 SimpleGeolocationProvider* provider) | |
| 43 : url_(url), | |
| 44 require_retries_(require_retries), | |
| 45 response_(response), | |
| 46 factory_(NULL), | |
| 47 attempts_(0), | |
| 48 provider_(provider) {} | |
| 49 | |
| 50 scoped_ptr<net::FakeURLFetcher> CreateURLFetcher( | |
| 51 const GURL& url, | |
| 52 net::URLFetcherDelegate* delegate, | |
| 53 const std::string& response_data, | |
| 54 net::HttpStatusCode response_code, | |
| 55 net::URLRequestStatus::Status status) { | |
| 56 EXPECT_EQ(provider_->requests_.size(), 1U); | |
| 57 | |
| 58 SimpleGeolocationRequest* geolocation_request = provider_->requests_[0]; | |
| 59 | |
| 60 const base::TimeDelta base_retry_interval = | |
| 61 base::TimeDelta::FromMilliseconds(kRequestRetryIntervalMilliSeconds); | |
| 62 geolocation_request->set_retry_sleep_on_server_error_for_testing( | |
| 63 base_retry_interval); | |
| 64 geolocation_request->set_retry_sleep_on_bad_response_for_testing( | |
| 65 base_retry_interval); | |
| 66 | |
| 67 ++attempts_; | |
| 68 if (attempts_ > require_retries_) { | |
| 69 response_code = net::HTTP_OK; | |
| 70 status = net::URLRequestStatus::SUCCESS; | |
| 71 factory_->SetFakeResponse(url, response_, response_code, status); | |
| 72 } | |
| 73 scoped_ptr<net::FakeURLFetcher> fetcher(new net::FakeURLFetcher( | |
| 74 url, delegate, response_, response_code, status)); | |
| 75 scoped_refptr<net::HttpResponseHeaders> download_headers = | |
| 76 new net::HttpResponseHeaders(std::string()); | |
| 77 download_headers->AddHeader("Content-Type: application/json"); | |
| 78 fetcher->set_response_headers(download_headers); | |
| 79 return fetcher.Pass(); | |
| 80 } | |
| 81 | |
| 82 void Initialize(net::FakeURLFetcherFactory* factory) { | |
| 83 factory_ = factory; | |
| 84 factory_->SetFakeResponse(url_, | |
| 85 std::string(), | |
| 86 net::HTTP_INTERNAL_SERVER_ERROR, | |
| 87 net::URLRequestStatus::FAILED); | |
| 88 } | |
| 89 | |
| 90 size_t attempts() const { return attempts_; } | |
| 91 | |
| 92 private: | |
| 93 const GURL url_; | |
| 94 // Respond with OK on required retry attempt. | |
| 95 const size_t require_retries_; | |
| 96 std::string response_; | |
| 97 net::FakeURLFetcherFactory* factory_; | |
| 98 size_t attempts_; | |
| 99 SimpleGeolocationProvider* provider_; | |
| 100 | |
| 101 DISALLOW_COPY_AND_ASSIGN(TestGeolocationAPIURLFetcherCallback); | |
| 102 }; | |
| 103 | |
| 104 // This implements fake Google MAPS Geolocation API remote endpoint. | |
| 105 // Response data is served to SimpleGeolocationProvider via | |
| 106 // net::FakeURLFetcher. | |
| 107 class GeolocationAPIFetcherFactory { | |
| 108 public: | |
| 109 GeolocationAPIFetcherFactory(const GURL& url, | |
| 110 const std::string& response, | |
| 111 const size_t require_retries, | |
| 112 SimpleGeolocationProvider* provider) { | |
| 113 url_callback_.reset(new TestGeolocationAPIURLFetcherCallback( | |
| 114 url, require_retries, response, provider)); | |
| 115 net::URLFetcherImpl::set_factory(NULL); | |
| 116 fetcher_factory_.reset(new net::FakeURLFetcherFactory( | |
| 117 NULL, | |
| 118 base::Bind(&TestGeolocationAPIURLFetcherCallback::CreateURLFetcher, | |
| 119 base::Unretained(url_callback_.get())))); | |
| 120 url_callback_->Initialize(fetcher_factory_.get()); | |
| 121 } | |
| 122 | |
| 123 size_t attempts() const { return url_callback_->attempts(); } | |
| 124 | |
| 125 private: | |
| 126 scoped_ptr<TestGeolocationAPIURLFetcherCallback> url_callback_; | |
| 127 scoped_ptr<net::FakeURLFetcherFactory> fetcher_factory_; | |
| 128 | |
| 129 DISALLOW_COPY_AND_ASSIGN(GeolocationAPIFetcherFactory); | |
| 130 }; | |
| 131 | |
| 132 class GeolocationReceiver { | |
| 133 public: | |
| 134 GeolocationReceiver() : server_error_(false) {} | |
| 135 | |
| 136 void OnRequestDone(const Geoposition& position, | |
| 137 bool server_error, | |
| 138 const base::TimeDelta elapsed) { | |
| 139 position_ = position; | |
| 140 server_error_ = server_error; | |
| 141 elapsed_ = elapsed; | |
| 142 | |
| 143 message_loop_runner_->Quit(); | |
| 144 } | |
| 145 | |
| 146 void WaitUntilRequestDone() { | |
| 147 message_loop_runner_.reset(new base::RunLoop); | |
| 148 message_loop_runner_->Run(); | |
| 149 } | |
| 150 | |
| 151 const Geoposition& position() const { return position_; } | |
| 152 bool server_error() const { return server_error_; } | |
| 153 base::TimeDelta elapsed() const { return elapsed_; } | |
| 154 | |
| 155 private: | |
| 156 Geoposition position_; | |
| 157 bool server_error_; | |
| 158 base::TimeDelta elapsed_; | |
| 159 scoped_ptr<base::RunLoop> message_loop_runner_; | |
| 160 }; | |
| 161 | |
| 162 class SimpleGeolocationTest : public testing::Test { | |
| 163 private: | |
| 164 base::MessageLoop message_loop_; | |
| 165 }; | |
| 166 | |
| 167 TEST_F(SimpleGeolocationTest, ResponseOK) { | |
| 168 SimpleGeolocationProvider provider(NULL, GURL(kTestGeolocationProviderUrl)); | |
| 169 | |
| 170 GeolocationAPIFetcherFactory url_factory(GURL(kTestGeolocationProviderUrl), | |
| 171 std::string(kSimpleResponseBody), | |
| 172 0 /* require_retries */, | |
| 173 &provider); | |
| 174 | |
| 175 GeolocationReceiver receiver; | |
| 176 provider.RequestGeolocation(base::TimeDelta::FromSeconds(1), | |
| 177 base::Bind(&GeolocationReceiver::OnRequestDone, | |
| 178 base::Unretained(&receiver))); | |
| 179 receiver.WaitUntilRequestDone(); | |
| 180 | |
| 181 EXPECT_EQ( | |
| 182 "latitude=51.000000, longitude=-0.100000, accuracy=1200.400000, " | |
| 183 "error_code=0, error_message='', status=1 (OK)", | |
| 184 receiver.position().ToString()); | |
| 185 EXPECT_FALSE(receiver.server_error()); | |
| 186 EXPECT_EQ(1U, url_factory.attempts()); | |
| 187 } | |
| 188 | |
| 189 TEST_F(SimpleGeolocationTest, ResponseOKWithRetries) { | |
| 190 SimpleGeolocationProvider provider(NULL, GURL(kTestGeolocationProviderUrl)); | |
| 191 | |
| 192 GeolocationAPIFetcherFactory url_factory(GURL(kTestGeolocationProviderUrl), | |
| 193 std::string(kSimpleResponseBody), | |
| 194 3 /* require_retries */, | |
| 195 &provider); | |
| 196 | |
| 197 GeolocationReceiver receiver; | |
| 198 provider.RequestGeolocation(base::TimeDelta::FromSeconds(1), | |
| 199 base::Bind(&GeolocationReceiver::OnRequestDone, | |
| 200 base::Unretained(&receiver))); | |
| 201 receiver.WaitUntilRequestDone(); | |
| 202 EXPECT_EQ( | |
| 203 "latitude=51.000000, longitude=-0.100000, accuracy=1200.400000, " | |
| 204 "error_code=0, error_message='', status=1 (OK)", | |
| 205 receiver.position().ToString()); | |
| 206 EXPECT_FALSE(receiver.server_error()); | |
| 207 EXPECT_EQ(4U, url_factory.attempts()); | |
| 208 } | |
| 209 | |
| 210 TEST_F(SimpleGeolocationTest, InvalidResponse) { | |
| 211 SimpleGeolocationProvider provider(NULL, GURL(kTestGeolocationProviderUrl)); | |
| 212 | |
| 213 GeolocationAPIFetcherFactory url_factory(GURL(kTestGeolocationProviderUrl), | |
| 214 "invalid JSON string", | |
| 215 0 /* require_retries */, | |
| 216 &provider); | |
| 217 | |
| 218 GeolocationReceiver receiver; | |
| 219 | |
| 220 const int timeout_seconds = 1; | |
| 221 size_t expected_retries = static_cast<size_t>( | |
| 222 timeout_seconds * 1000 / kRequestRetryIntervalMilliSeconds); | |
| 223 ASSERT_GE(expected_retries, 2U); | |
| 224 | |
| 225 provider.RequestGeolocation(base::TimeDelta::FromSeconds(timeout_seconds), | |
| 226 base::Bind(&GeolocationReceiver::OnRequestDone, | |
| 227 base::Unretained(&receiver))); | |
| 228 receiver.WaitUntilRequestDone(); | |
| 229 | |
| 230 EXPECT_EQ( | |
| 231 "latitude=200.000000, longitude=200.000000, accuracy=-1.000000, " | |
| 232 "error_code=0, error_message='SimpleGeolocation provider at " | |
| 233 "'https://localhost/' : JSONReader failed: Line: 1, column: 1, " | |
| 234 "Unexpected token..', status=4 (TIMEOUT)", | |
| 235 receiver.position().ToString()); | |
| 236 EXPECT_TRUE(receiver.server_error()); | |
| 237 EXPECT_GE(url_factory.attempts(), 2U); | |
| 238 if (url_factory.attempts() > expected_retries + 1) { | |
| 239 LOG(WARNING) | |
| 240 << "SimpleGeolocationTest::InvalidResponse: Too many attempts (" | |
| 241 << url_factory.attempts() << "), no more then " << expected_retries + 1 | |
| 242 << " expected."; | |
| 243 } | |
| 244 if (url_factory.attempts() < expected_retries - 1) { | |
| 245 LOG(WARNING) | |
| 246 << "SimpleGeolocationTest::InvalidResponse: Too little attempts (" | |
| 247 << url_factory.attempts() << "), greater then " << expected_retries - 1 | |
| 248 << " expected."; | |
| 249 } | |
| 250 } | |
| 251 | |
| 252 } // namespace chromeos | |
| OLD | NEW |