Index: components/scheduler/child/scheduler_helper_unittest.cc |
diff --git a/components/scheduler/child/scheduler_helper_unittest.cc b/components/scheduler/child/scheduler_helper_unittest.cc |
index 2807186a853e3aa176fca7a4c5004b2c02cc19e4..5064c9d34e5f4565a25217395cad9a170f2ed2ea 100644 |
--- a/components/scheduler/child/scheduler_helper_unittest.cc |
+++ b/components/scheduler/child/scheduler_helper_unittest.cc |
@@ -183,4 +183,26 @@ TEST_F(SchedulerHelperTest, |
RunUntilIdle(); |
} |
+namespace { |
+class MockObserver : public SchedulerHelper::Observer { |
+ public: |
+ MOCK_METHOD1(OnUnregisterTaskQueue, |
+ void(const scoped_refptr<TaskQueue>& queue)); |
+}; |
+} // namespace |
+ |
+TEST_F(SchedulerHelperTest, OnUnregisterTaskQueue) { |
+ MockObserver observer; |
+ scheduler_helper_->SetQueueObserver(&observer); |
+ |
+ scoped_refptr<TaskQueue> task_queue = |
+ scheduler_helper_->NewTaskQueue(TaskQueue::Spec("test_queue")); |
+ |
+ EXPECT_CALL(observer, OnUnregisterTaskQueue(_)).Times(1); |
+ task_queue->UnregisterTaskQueue(); |
+ |
+ // Tidy up. |
+ scheduler_helper_->SetQueueObserver(nullptr); |
+} |
+ |
} // namespace scheduler |