Introduce AfterStartupTaskUtils::Runner for callsites that want to use AfterStartupTaskUtils but ne…

chrome/browser/after_startup_task_utils_unittest.cc

 // Copyright 2015 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 "chrome/browser/after_startup_task_utils.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 180 matching lines @@
                                    base::Bind(&base::DoNothing));
   PostAfterStartupTaskFromDBThread(FROM_HERE, db_thread_,
                                    base::Bind(&base::DoNothing));
   EXPECT_EQ(2, db_thread_->posted_task_count());
   EXPECT_EQ(2, ui_thread_->posted_task_count());
   FlushDBThread();
   RunLoop().RunUntilIdle();
   EXPECT_EQ(2, db_thread_->ran_task_count());
   EXPECT_EQ(2, ui_thread_->ran_task_count());
 }
+
+// Verify that posting to an AfterStartupTaskUtils::Runner bound to |db_thread_|
+// results in the same behavior as posting via
+// AfterStartupTaskUtils::PostTask(..., db_thread_, ...).
+TEST_F(AfterStartupTaskTest, AfterStartupTaskUtilsRunner) {
+  scoped_refptr<base::TaskRunner> after_startup_runner =
+      make_scoped_refptr(new AfterStartupTaskUtils::Runner(db_thread_));
+
+  EXPECT_FALSE(AfterStartupTaskUtils::IsBrowserStartupComplete());
+  after_startup_runner->PostTask(
+      FROM_HERE, base::Bind(&AfterStartupTaskTest::VerifyExpectedThread,
+                            BrowserThread::DB));
+
+  RunLoop().RunUntilIdle();
+  EXPECT_FALSE(AfterStartupTaskUtils::IsBrowserStartupComplete());
+  EXPECT_EQ(0, db_thread_->total_task_count());
+
+  AfterStartupTaskUtils::SetBrowserStartupIsCompleteForTesting();
+  EXPECT_EQ(1, db_thread_->posted_task_count());
+
+  FlushDBThread();
+  RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, db_thread_->ran_task_count());
+
+  EXPECT_EQ(0, ui_thread_->total_task_count());
+}