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
13namespace metrics {
14
15class 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
40TEST_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
51TEST_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