Index: components/rappor/log_uploader_unittest.cc |
diff --git a/components/rappor/log_uploader_unittest.cc b/components/rappor/log_uploader_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..fb8c51fc9133a48a8fda766cf35d029b26a7cc1e |
--- /dev/null |
+++ b/components/rappor/log_uploader_unittest.cc |
@@ -0,0 +1,138 @@ |
+// 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/rappor/log_uploader.h" |
+ |
+#include "base/message_loop/message_loop_proxy.h" |
+#include "net/url_request/test_url_fetcher_factory.h" |
+#include "net/url_request/url_request_test_util.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace rappor { |
+ |
+const GURL kTestServerURL = GURL("http://a.com/"); |
+const std::string kTestMimeType = "text/plain"; |
+ |
+class LogUploaderTest : public testing::Test { |
+ public: |
+ LogUploaderTest() |
+ : request_context_(new net::TestURLRequestContextGetter( |
+ base::MessageLoopProxy::current())), |
+ factory_(NULL) {} |
+ |
+ protected: |
+ base::MessageLoopForUI loop_; |
+ scoped_refptr<net::TestURLRequestContextGetter> request_context_; |
+ net::FakeURLFetcherFactory factory_; |
+}; |
+ |
+class TestLogUploader : public LogUploader { |
+ public: |
+ TestLogUploader(net::URLRequestContextGetter* request_context) : |
+ LogUploader(kTestServerURL, kTestMimeType, request_context) { |
+ } |
+ |
+ base::TimeDelta last_interval_set; |
+ |
+ void StartUpload() { |
+ last_interval_set = base::TimeDelta(); |
+ StartScheduledUpload(); |
+ } |
+ |
+ protected: |
+ virtual bool IsUploadScheduled() const OVERRIDE { |
+ return last_interval_set != base::TimeDelta(); |
+ } |
+ |
+ // Schedules a future call to StartScheduledUpload if one isn't already |
+ // pending. |
+ virtual void ScheduleNextUpload(base::TimeDelta interval) OVERRIDE { |
+ EXPECT_EQ(last_interval_set, base::TimeDelta()); |
+ last_interval_set = interval; |
+ } |
+}; |
+ |
+TEST_F(LogUploaderTest, Success) { |
+ TestLogUploader uploader(request_context_); |
+ |
+ factory_.SetFakeResponse(kTestServerURL, |
+ "", |
+ net::HTTP_OK, |
+ net::URLRequestStatus::SUCCESS); |
+ |
+ uploader.QueueLog("log1"); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ // Log should be discarded instead of retransmitted |
+ EXPECT_EQ(uploader.last_interval_set, base::TimeDelta()); |
+} |
+ |
+TEST_F(LogUploaderTest, Rejection) { |
+ TestLogUploader uploader(request_context_); |
+ |
+ factory_.SetFakeResponse(kTestServerURL, |
+ "", |
+ net::HTTP_BAD_REQUEST, |
+ net::URLRequestStatus::SUCCESS); |
+ |
+ uploader.QueueLog("log1"); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ // Log should be discarded instead of retransmitted |
+ EXPECT_EQ(uploader.last_interval_set, base::TimeDelta()); |
+} |
+ |
+TEST_F(LogUploaderTest, Failure) { |
+ TestLogUploader uploader(request_context_); |
+ |
+ factory_.SetFakeResponse(kTestServerURL, |
+ "", |
+ net::HTTP_INTERNAL_SERVER_ERROR, |
+ net::URLRequestStatus::SUCCESS); |
+ |
+ uploader.QueueLog("log1"); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ // Log should be scheduled for retransmission |
+ base::TimeDelta error_interval = uploader.last_interval_set; |
+ EXPECT_GT(error_interval, base::TimeDelta()); |
+ |
+ for (int i = 0; i < 10; i++) { |
+ uploader.QueueLog("logX"); |
+ } |
+ |
+ // A second failure should lead to a longer interval, and the log should |
+ // be discarded due to full queue. |
+ uploader.StartUpload(); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ EXPECT_GT(uploader.last_interval_set, error_interval); |
+ |
+ factory_.SetFakeResponse(kTestServerURL, |
+ "", |
+ net::HTTP_OK, |
+ net::URLRequestStatus::SUCCESS); |
+ |
+ // A success should revert to base interval while queue is not empty. |
+ for (int i = 0; i < 9; i++) { |
+ uploader.StartUpload(); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ EXPECT_LT(uploader.last_interval_set, error_interval); |
+ } |
+ |
+ // Queue should be empty; |
+ uploader.StartUpload(); |
+ base::MessageLoop::current()->RunUntilIdle(); |
+ EXPECT_EQ(uploader.last_interval_set, base::TimeDelta()); |
+} |
+ |
+TEST_F(LogUploaderTest, Backoff) { |
+ base::TimeDelta current = base::TimeDelta(); |
+ base::TimeDelta next = base::TimeDelta::FromSeconds(1); |
+ // Backoff until the maximum is reached. |
+ while(next > current) { |
Alexei Svitkine (slow)
2014/02/07 18:24:22
Nit: Space after while
Steven Holte
2014/02/07 21:08:16
Done.
|
+ current = next; |
+ next = LogUploader::BackOffUploadInterval(current); |
+ } |
+ // Maximum backoff should have been reached. |
+ EXPECT_EQ(next, current); |
+} |
+ |
+} // namespace rappor |