Index: cc/base/unique_notifier_unittest.cc |
diff --git a/cc/base/unique_notifier_unittest.cc b/cc/base/unique_notifier_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e5344ca1b40a7594540d7f0f7d8e0c9f40ae96c7 |
--- /dev/null |
+++ b/cc/base/unique_notifier_unittest.cc |
@@ -0,0 +1,60 @@ |
+// 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 "base/bind.h" |
+#include "base/bind_helpers.h" |
+#include "base/message_loop/message_loop_proxy.h" |
+#include "base/run_loop.h" |
+#include "base/single_thread_task_runner.h" |
+#include "cc/base/unique_notifier.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace cc { |
+namespace { |
+ |
+class UniqueNotifierTest : public testing::Test { |
+ public: |
+ UniqueNotifierTest() : notification_count_(0) {} |
+ |
+ virtual void SetUp() OVERRIDE { |
+ task_runner_ = base::MessageLoopProxy::current(); |
+ ResetNotificationCount(); |
+ } |
+ |
+ virtual void TearDown() OVERRIDE { task_runner_ = NULL; } |
+ |
+ void Notify() { ++notification_count_; } |
+ |
+ int NotificationCount() const { return notification_count_; } |
+ |
+ void ResetNotificationCount() { notification_count_ = 0; } |
+ |
+ protected: |
+ int notification_count_; |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
reveman
2014/05/21 19:42:45
nit: remove this and just use base::MessageLoopPro
|
+}; |
+ |
+TEST_F(UniqueNotifierTest, Schedule) { |
+ UniqueNotifier notifier( |
+ task_runner_.get(), |
+ base::Bind(&UniqueNotifierTest::Notify, base::Unretained(this))); |
+ |
+ EXPECT_EQ(0, NotificationCount()); |
+ |
+ // Basic schedule should result in a run. |
+ notifier.Schedule(); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, NotificationCount()); |
+ |
+ // Multiple schedules should only result in one run. |
+ for (int i = 0; i < 5; ++i) |
+ notifier.Schedule(); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(2, NotificationCount()); |
+} |
+ |
+} // namespace |
+} // namespace cc |