| // 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 "components/domain_reliability/uploader.h" |
| |
| #include <algorithm> |
| #include <cstring> |
| #include <map> |
| |
| #include "base/bind.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/test/test_simple_task_runner.h" |
| #include "components/domain_reliability/test_util.h" |
| #include "net/base/load_flags.h" |
| #include "net/base/net_errors.h" |
| #include "net/http/http_response_headers.h" |
| #include "net/url_request/test_url_fetcher_factory.h" |
| #include "net/url_request/url_fetcher_delegate.h" |
| #include "net/url_request/url_request_context_getter.h" |
| #include "net/url_request/url_request_test_util.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace domain_reliability { |
| namespace { |
| |
| const char *kReportJSON = "{}"; |
| const char *kUploadURL = "https://example/upload"; |
| |
| class DomainReliabilityUploaderTest : public testing::Test { |
| protected: |
| DomainReliabilityUploaderTest() |
| : network_task_runner_(new base::TestSimpleTaskRunner()), |
| url_request_context_getter_(new net::TestURLRequestContextGetter( |
| network_task_runner_)), |
| uploader_(DomainReliabilityUploader::Create( |
| &time_, url_request_context_getter_)) { |
| uploader_->set_discard_uploads(false); |
| } |
| |
| DomainReliabilityUploader::UploadCallback MakeUploadCallback(size_t index) { |
| return base::Bind(&DomainReliabilityUploaderTest::OnUploadComplete, |
| base::Unretained(this), |
| index); |
| } |
| |
| void OnUploadComplete(size_t index, |
| const DomainReliabilityUploader::UploadResult& result) { |
| EXPECT_FALSE(upload_complete_[index]); |
| upload_complete_[index] = true; |
| upload_result_[index] = result; |
| } |
| |
| void SimulateUpload() { |
| uploader_->UploadReport(kReportJSON, |
| GURL(kUploadURL), |
| MakeUploadCallback(0)); |
| } |
| |
| void SimulateUploadRequest() { |
| net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); |
| EXPECT_TRUE(fetcher); |
| |
| EXPECT_EQ(kReportJSON, fetcher->upload_data()); |
| EXPECT_EQ(GURL(kUploadURL), fetcher->GetOriginalURL()); |
| EXPECT_TRUE(fetcher->GetLoadFlags() & net::LOAD_DO_NOT_SAVE_COOKIES); |
| EXPECT_TRUE(fetcher->GetLoadFlags() & net::LOAD_DO_NOT_SEND_COOKIES); |
| |
| fetcher->set_url(GURL(kUploadURL)); |
| } |
| |
| void SimulateNoUploadRequest() { |
| net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); |
| EXPECT_FALSE(fetcher); |
| } |
| |
| void SimulateUploadResponse(int response_code, const char* headers) { |
| net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); |
| EXPECT_TRUE(fetcher); |
| |
| fetcher->set_status(net::URLRequestStatus()); |
| fetcher->set_response_code(response_code); |
| fetcher->set_response_headers(new net::HttpResponseHeaders( |
| net::HttpUtil::AssembleRawHeaders(headers, strlen(headers)))); |
| fetcher->SetResponseString(""); |
| fetcher->delegate()->OnURLFetchComplete(fetcher); |
| } |
| |
| void SimulateUploadError(int error) { |
| net::TestURLFetcher* fetcher = url_fetcher_factory_.GetFetcherByID(0); |
| EXPECT_TRUE(fetcher); |
| |
| net::URLRequestStatus status; |
| fetcher->set_status(net::URLRequestStatus::FromError(error)); |
| fetcher->set_response_code(-1); |
| fetcher->delegate()->OnURLFetchComplete(fetcher); |
| } |
| |
| scoped_refptr<base::TestSimpleTaskRunner> network_task_runner_; |
| net::TestURLFetcherFactory url_fetcher_factory_; |
| scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_; |
| MockTime time_; |
| scoped_ptr<DomainReliabilityUploader> uploader_; |
| |
| // Whether the upload callback was called for a particular collector index. |
| std::map<size_t, bool> upload_complete_; |
| // The result of the last completed upload for a particular collector index. |
| std::map<size_t, DomainReliabilityUploader::UploadResult> upload_result_; |
| }; |
| |
| TEST_F(DomainReliabilityUploaderTest, Create) { |
| SimulateNoUploadRequest(); |
| } |
| |
| TEST_F(DomainReliabilityUploaderTest, SuccessfulUpload) { |
| SimulateUpload(); |
| SimulateUploadRequest(); |
| EXPECT_FALSE(upload_complete_[0]); |
| SimulateUploadResponse(200, "HTTP/1.1 200 OK\n\n"); |
| EXPECT_TRUE(upload_complete_[0]); |
| EXPECT_TRUE(upload_result_[0].is_success()); |
| } |
| |
| TEST_F(DomainReliabilityUploaderTest, NetworkFailedUpload) { |
| SimulateUpload(); |
| SimulateUploadRequest(); |
| EXPECT_FALSE(upload_complete_[0]); |
| SimulateUploadError(net::ERR_CONNECTION_REFUSED); |
| EXPECT_TRUE(upload_complete_[0]); |
| EXPECT_TRUE(upload_result_[0].is_failure()); |
| } |
| |
| TEST_F(DomainReliabilityUploaderTest, ServerFailedUpload) { |
| SimulateUpload(); |
| SimulateUploadRequest(); |
| EXPECT_FALSE(upload_complete_[0]); |
| SimulateUploadResponse(500, "HTTP/1.1 500 Blargh\n\n"); |
| EXPECT_TRUE(upload_complete_[0]); |
| EXPECT_TRUE(upload_result_[0].is_failure()); |
| } |
| |
| TEST_F(DomainReliabilityUploaderTest, RetryAfterUpload) { |
| SimulateUpload(); |
| SimulateUploadRequest(); |
| EXPECT_FALSE(upload_complete_[0]); |
| SimulateUploadResponse(503, "HTTP/1.1 503 Ugh\nRetry-After: 3600\n\n"); |
| EXPECT_TRUE(upload_complete_[0]); |
| EXPECT_TRUE(upload_result_[0].is_retry_after()); |
| EXPECT_EQ(base::TimeDelta::FromSeconds(3600), upload_result_[0].retry_after); |
| } |
| |
| TEST_F(DomainReliabilityUploaderTest, DiscardedUpload) { |
| uploader_->set_discard_uploads(true); |
| SimulateUpload(); |
| SimulateNoUploadRequest(); |
| } |
| |
| } // namespace |
| } // namespace domain_reliability |