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 "components/metrics/metrics_reporting_scheduler.h" 6 7 #include "base/basictypes.h" 8 #include "base/bind.h" 9 #include "base/message_loop/message_loop.h" 10 #include "base/run_loop.h" 11 #include "testing/gtest/include/gtest/gtest.h" 12 13 namespace metrics { 14 15 class MetricsReportingSchedulerTest : public testing::Test { 16 public: 17 MetricsReportingSchedulerTest() : callback_call_count_(0) {} 18 virtual ~MetricsReportingSchedulerTest() {} 19 20 base::Closure GetCallback() { 21 return base::Bind(&MetricsReportingSchedulerTest::SchedulerCallback, 22 base::Unretained(this)); 23 } 24 25 int callback_call_count() const { return callback_call_count_; } 26 27 private: 28 void SchedulerCallback() { 29 ++callback_call_count_; 30 } 31 32 int callback_call_count_; 33 34 base::MessageLoopForUI message_loop_; 35 36 DISALLOW_COPY_AND_ASSIGN(MetricsReportingSchedulerTest); 37 }; 38 39 40 TEST_F(MetricsReportingSchedulerTest, InitTaskCompleteBeforeTimer) { 41 MetricsReportingScheduler scheduler(GetCallback()); 42 scheduler.SetUploadIntervalForTesting(base::TimeDelta()); 43 scheduler.InitTaskComplete(); 44 scheduler.Start(); 45 EXPECT_EQ(0, callback_call_count()); 46 47 base::RunLoop().RunUntilIdle(); 48 EXPECT_EQ(1, callback_call_count()); 49 } 50 51 TEST_F(MetricsReportingSchedulerTest, InitTaskCompleteAfterTimer) { 52 MetricsReportingScheduler scheduler(GetCallback()); 53 scheduler.SetUploadIntervalForTesting(base::TimeDelta()); 54 scheduler.Start(); 55 base::RunLoop().RunUntilIdle(); 56 EXPECT_EQ(0, callback_call_count()); 57 58 scheduler.InitTaskComplete(); 59 EXPECT_EQ(1, callback_call_count()); 60 } 61 62 } // namespace metrics 63
