content/browser/browser_thread_unittest.cc

   1 // Copyright (c) 2012 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/location.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/sequenced_task_runner_helpers.h"
  10 #include "base/single_thread_task_runner.h"
  11 #include "content/browser/browser_thread_impl.h"
  12 #include "content/public/test/test_browser_thread.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14 #include "testing/platform_test.h"
  15
  16 namespace content {
  17
  18 class BrowserThreadTest : public testing::Test {
  19  public:
  20   void Release() const {
  21     CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  22     loop_.task_runner()->PostTask(FROM_HERE, base::MessageLoop::QuitClosure());
  23   }
  24
  25  protected:
  26   void SetUp() override {
  27     ui_thread_.reset(new BrowserThreadImpl(BrowserThread::UI));
  28     file_thread_.reset(new BrowserThreadImpl(BrowserThread::FILE));
  29     ui_thread_->Start();
  30     file_thread_->Start();
  31   }
  32
  33   void TearDown() override {
  34     ui_thread_->Stop();
  35     file_thread_->Stop();
  36   }
  37
  38   static void BasicFunction(base::MessageLoop* message_loop) {
  39     CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
  40     message_loop->task_runner()->PostTask(FROM_HERE,
  41                                           base::MessageLoop::QuitClosure());
  42   }
  43
  44   class DeletedOnFile
  45       : public base::RefCountedThreadSafe<
  46             DeletedOnFile, BrowserThread::DeleteOnFileThread> {
  47    public:
  48     explicit DeletedOnFile(base::MessageLoop* message_loop)
  49         : message_loop_(message_loop) {}
  50
  51    private:
  52     friend struct BrowserThread::DeleteOnThread<BrowserThread::FILE>;
  53     friend class base::DeleteHelper<DeletedOnFile>;
  54
  55     ~DeletedOnFile() {
  56       CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
  57       message_loop_->task_runner()->PostTask(FROM_HERE,
  58                                              base::MessageLoop::QuitClosure());
  59     }
  60
  61     base::MessageLoop* message_loop_;
  62   };
  63
  64  private:
  65   scoped_ptr<BrowserThreadImpl> ui_thread_;
  66   scoped_ptr<BrowserThreadImpl> file_thread_;
  67   // It's kind of ugly to make this mutable - solely so we can post the Quit
  68   // Task from Release(). This should be fixed.
  69   mutable base::MessageLoop loop_;
  70 };
  71
  72 TEST_F(BrowserThreadTest, PostTask) {
  73   BrowserThread::PostTask(
  74       BrowserThread::FILE,
  75       FROM_HERE,
  76       base::Bind(&BasicFunction, base::MessageLoop::current()));
  77   base::MessageLoop::current()->Run();
  78 }
  79
  80 TEST_F(BrowserThreadTest, Release) {
  81   BrowserThread::ReleaseSoon(BrowserThread::UI, FROM_HERE, this);
  82   base::MessageLoop::current()->Run();
  83 }
  84
  85 TEST_F(BrowserThreadTest, ReleasedOnCorrectThread) {
  86   {
  87     scoped_refptr<DeletedOnFile> test(
  88         new DeletedOnFile(base::MessageLoop::current()));
  89   }
  90   base::MessageLoop::current()->Run();
  91 }
  92
  93 TEST_F(BrowserThreadTest, PostTaskViaTaskRunner) {
  94   scoped_refptr<base::SingleThreadTaskRunner> task_runner =
  95       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE);
  96   task_runner->PostTask(
  97       FROM_HERE, base::Bind(&BasicFunction, base::MessageLoop::current()));
  98   base::MessageLoop::current()->Run();
  99 }
 100
 101 TEST_F(BrowserThreadTest, ReleaseViaTaskRunner) {
 102   scoped_refptr<base::SingleThreadTaskRunner> task_runner =
 103       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI);
 104   task_runner->ReleaseSoon(FROM_HERE, this);
 105   base::MessageLoop::current()->Run();
 106 }
 107
 108 TEST_F(BrowserThreadTest, PostTaskAndReply) {
 109   // Most of the heavy testing for PostTaskAndReply() is done inside the
 110   // task runner test.  This just makes sure we get piped through at all.
 111   ASSERT_TRUE(BrowserThread::PostTaskAndReply(
 112       BrowserThread::FILE,
 113       FROM_HERE,
 114       base::Bind(&base::DoNothing),
 115       base::Bind(&base::MessageLoop::Quit,
 116                  base::Unretained(base::MessageLoop::current()->current()))));
 117   base::MessageLoop::current()->Run();
 118 }
 119
 120 }  // namespace content