Index: components/scheduler/child/task_queue_manager_unittest.cc |
diff --git a/components/scheduler/child/task_queue_manager_unittest.cc b/components/scheduler/child/task_queue_manager_unittest.cc |
index c35861a1e5738a7953b7dc8d515ba232d0cefcf4..dac0447dafe5d41153e3711d5cfae2b9e1ba7772 100644 |
--- a/components/scheduler/child/task_queue_manager_unittest.cc |
+++ b/components/scheduler/child/task_queue_manager_unittest.cc |
@@ -1237,4 +1237,28 @@ TEST_F(TaskQueueManagerTest, UnregisterTaskQueue_WithDelayedTasks) { |
ASSERT_THAT(run_order, ElementsAre(1, 3)); |
} |
+namespace { |
+class MockObserver : public TaskQueueManager::Observer { |
+ public: |
+ MOCK_METHOD1(OnUnregisterTaskQueue, |
+ void(const scoped_refptr<internal::TaskQueueImpl>& queue)); |
+}; |
+} // namespace |
+ |
+TEST_F(TaskQueueManagerTest, OnUnregisterTaskQueue) { |
+ Initialize(0u); |
+ |
+ MockObserver observer; |
+ manager_->SetQueueObserver(&observer); |
+ |
+ scoped_refptr<internal::TaskQueueImpl> task_queue = |
+ manager_->NewTaskQueue(TaskQueue::Spec("test_queue")); |
+ |
+ EXPECT_CALL(observer, OnUnregisterTaskQueue(_)).Times(2); |
+ task_queue->UnregisterTaskQueue(); |
+ |
+ // Tidy up. |
+ manager_->SetQueueObserver(nullptr); |
+} |
+ |
} // namespace scheduler |