chrome/browser/no_best_effort_tasks_during_startup_browsertest.cc

   1 // Copyright 2019 The Chromium Authors
   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/barrier_closure.h"
   6 #include "base/run_loop.h"
   7 #include "base/task/thread_pool.h"
   8 #include "base/test/bind.h"
   9 #include "chrome/browser/after_startup_task_utils.h"
  10 #include "chrome/test/base/in_process_browser_test.h"
  11 #include "content/public/browser/browser_task_traits.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "content/public/test/browser_test.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 namespace {
  17
  18 class NoBestEffortTasksDuringStartupTest : public InProcessBrowserTest {
  19  public:
  20   // InProcessBrowserTest:
  21   void PreRunTestOnMainThread() override {
  22     // This test must run before PreRunTestOnMainThread() sets startup as
  23     // complete.
  24     TestNoBestEffortTasksDuringStartup();
  25
  26     InProcessBrowserTest::PreRunTestOnMainThread();
  27   }
  28
  29   void TestNoBestEffortTasksDuringStartup() {
  30     EXPECT_FALSE(AfterStartupTaskUtils::IsBrowserStartupComplete());
  31
  32     base::RunLoop run_loop;
  33     auto barrier = base::BarrierClosure(2, run_loop.QuitClosure());
  34
  35     // Thread pool task.
  36     base::ThreadPool::PostTask(
  37         FROM_HERE, {base::TaskPriority::BEST_EFFORT},
  38         base::BindLambdaForTesting([&]() {
  39           EXPECT_TRUE(AfterStartupTaskUtils::IsBrowserStartupComplete());
  40           barrier.Run();
  41         }));
  42
  43     // UI thread task.
  44     content::GetUIThreadTaskRunner({base::TaskPriority::BEST_EFFORT})
  45         ->PostTask(
  46             FROM_HERE, base::BindLambdaForTesting([&]() {
  47               EXPECT_TRUE(AfterStartupTaskUtils::IsBrowserStartupComplete());
  48               barrier.Run();
  49             }));
  50
  51     run_loop.Run();
  52   }
  53 };
  54
  55 }  // namespace
  56
  57 // Verify that BEST_EFFORT tasks don't run until startup is complete.
  58 IN_PROC_BROWSER_TEST_F(NoBestEffortTasksDuringStartupTest,
  59                        NoBestEffortTasksDuringStartup) {
  60   // The body of the test is in the TestNoBestEffortTasksDuringStartup() method
  61   // called from PreRunTestOnMainThread().
  62 }