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 "base/bind.h" 6 #include "base/bind_helpers.h" 7 #include "base/message_loop/message_loop_proxy.h" 8 #include "base/run_loop.h" 9 #include "base/single_thread_task_runner.h" 10 #include "cc/base/unique_notifier.h" 11 #include "testing/gtest/include/gtest/gtest.h" 12 13 namespace cc { 14 namespace { 15 16 class UniqueNotifierTest : public testing::Test { 17 public: 18 UniqueNotifierTest() : notification_count_(0) {} 19 20 virtual void SetUp() OVERRIDE { ResetNotificationCount(); } 21 22 void Notify() { ++notification_count_; } 23 24 int NotificationCount() const { return notification_count_; } 25 26 void ResetNotificationCount() { notification_count_ = 0; } 27 28 protected: 29 int notification_count_; 30 }; 31 32 TEST_F(UniqueNotifierTest, Schedule) { 33 { 34 UniqueNotifier notifier( 35 base::MessageLoopProxy::current().get(), 36 base::Bind(&UniqueNotifierTest::Notify, base::Unretained(this))); 37 38 EXPECT_EQ(0, NotificationCount()); 39 40 // Basic schedule should result in a run. 41 notifier.Schedule(); 42 43 base::RunLoop().RunUntilIdle(); 44 EXPECT_EQ(1, NotificationCount()); 45 46 // Multiple schedules should only result in one run. 47 for (int i = 0; i < 5; ++i) 48 notifier.Schedule(); 49 50 base::RunLoop().RunUntilIdle(); 51 EXPECT_EQ(2, NotificationCount()); 52 53 notifier.Schedule(); 54 } 55 56 // Notifier went out of scope, so we don't expect to get a notification. 57 base::RunLoop().RunUntilIdle(); 58 EXPECT_EQ(2, NotificationCount()); 59 } 60 61 } // namespace 62 } // namespace cc 63
