text: // Copyright (c) 2013 The Chromium Embedded Framework 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 "include/base/cef_bind.h"
  #include "include/cef_task.h"
  #include "include/wrapper/cef_closure_task.h"
  #include "tests/ceftests/test_handler.h"
  #include "tests/gtest/include/gtest/gtest.h"
  
  namespace {
  
  void GetForCurrentThread(bool* ran_test) {
    // Currently on the FILE thread.
    CefRefPtr<CefTaskRunner> runner = CefTaskRunner::GetForCurrentThread();
    EXPECT_TRUE(runner.get());
    EXPECT_TRUE(runner->BelongsToCurrentThread());
    EXPECT_TRUE(runner->BelongsToThread(TID_FILE));
    EXPECT_FALSE(runner->BelongsToThread(TID_DB));
    EXPECT_TRUE(runner->IsSame(runner));
  
    CefRefPtr<CefTaskRunner> runner2 = CefTaskRunner::GetForCurrentThread();
    EXPECT_TRUE(runner2.get());
    EXPECT_TRUE(runner->IsSame(runner2));
    EXPECT_TRUE(runner2->IsSame(runner));
  
    // Not on the DB thread.
    CefRefPtr<CefTaskRunner> runner3 = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner3.get());
    EXPECT_FALSE(runner->IsSame(runner3));
    EXPECT_FALSE(runner3->IsSame(runner));
  
    *ran_test = true;
  }
  
  void GetForThread(bool* ran_test) {
    // Currently on the FILE thread.
    CefRefPtr<CefTaskRunner> runner = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner.get());
    EXPECT_FALSE(runner->BelongsToCurrentThread());
    EXPECT_TRUE(runner->BelongsToThread(TID_DB));
    EXPECT_FALSE(runner->BelongsToThread(TID_FILE));
    EXPECT_TRUE(runner->IsSame(runner));
  
    CefRefPtr<CefTaskRunner> runner2 = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner2.get());
    EXPECT_TRUE(runner->IsSame(runner2));
    EXPECT_TRUE(runner2->IsSame(runner));
  
    CefRefPtr<CefTaskRunner> runner3 = CefTaskRunner::GetForThread(TID_FILE);
    EXPECT_TRUE(runner3.get());
    EXPECT_FALSE(runner->IsSame(runner3));
    EXPECT_FALSE(runner3->IsSame(runner));
  
    *ran_test = true;
  }
  
  void PostTaskEvent1(bool* got_it, CefRefPtr<CefTaskRunner> runner) {
    // Currently on the DB thread.
    EXPECT_TRUE(runner->BelongsToCurrentThread());
    EXPECT_TRUE(runner->BelongsToThread(TID_DB));
    EXPECT_FALSE(runner->BelongsToThread(TID_FILE));
  
    // Current thread should be the DB thread.
    CefRefPtr<CefTaskRunner> runner2 = CefTaskRunner::GetForCurrentThread();
    EXPECT_TRUE(runner2.get());
    EXPECT_TRUE(runner2->BelongsToCurrentThread());
    EXPECT_TRUE(runner2->BelongsToThread(TID_DB));
    EXPECT_FALSE(runner2->BelongsToThread(TID_FILE));
    EXPECT_TRUE(runner->IsSame(runner2));
    EXPECT_TRUE(runner2->IsSame(runner));
  
    // Current thread should be the DB thread.
    CefRefPtr<CefTaskRunner> runner3 = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner3.get());
    EXPECT_TRUE(runner3->BelongsToCurrentThread());
    EXPECT_TRUE(runner3->BelongsToThread(TID_DB));
    EXPECT_FALSE(runner3->BelongsToThread(TID_FILE));
    EXPECT_TRUE(runner->IsSame(runner3));
    EXPECT_TRUE(runner3->IsSame(runner));
  
    // Current thread should not be the FILE thread.
    CefRefPtr<CefTaskRunner> runner4 = CefTaskRunner::GetForThread(TID_FILE);
    EXPECT_TRUE(runner4.get());
    EXPECT_FALSE(runner4->BelongsToCurrentThread());
    EXPECT_FALSE(runner4->BelongsToThread(TID_DB));
    EXPECT_TRUE(runner4->BelongsToThread(TID_FILE));
    EXPECT_FALSE(runner->IsSame(runner4));
    EXPECT_FALSE(runner4->IsSame(runner));
  
    *got_it = true;
  }
  
  void PostTask1(bool* ran_test) {
    // Currently on the FILE thread.
    CefRefPtr<CefTaskRunner> runner = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner.get());
    EXPECT_FALSE(runner->BelongsToCurrentThread());
    EXPECT_TRUE(runner->BelongsToThread(TID_DB));
  
    bool got_it = false;
    runner->PostTask(
        CefCreateClosureTask(base::Bind(&PostTaskEvent1, &got_it, runner)));
  
    WaitForThread(runner);
    EXPECT_TRUE(got_it);
  
    *ran_test = true;
  }
  
  void PostDelayedTask1(bool* ran_test) {
    // Currently on the FILE thread.
    CefRefPtr<CefTaskRunner> runner = CefTaskRunner::GetForThread(TID_DB);
    EXPECT_TRUE(runner.get());
    EXPECT_FALSE(runner->BelongsToCurrentThread());
    EXPECT_TRUE(runner->BelongsToThread(TID_DB));
  
    bool got_it = false;
    runner->PostDelayedTask(
        CefCreateClosureTask(base::Bind(&PostTaskEvent1, &got_it, runner)), 0);
  
    WaitForThread(runner);
    EXPECT_TRUE(got_it);
  
    *ran_test = true;
  }
  
  void PostTaskEvent2(bool* got_it) {
    EXPECT_TRUE(CefCurrentlyOn(TID_DB));
    EXPECT_FALSE(CefCurrentlyOn(TID_FILE));
  
    *got_it = true;
  }
  
  void PostTask2(bool* ran_test) {
    // Currently on the FILE thread.
    EXPECT_FALSE(CefCurrentlyOn(TID_DB));
  
    bool got_it = false;
    CefPostTask(TID_DB,
                CefCreateClosureTask(base::Bind(&PostTaskEvent2, &got_it)));
  
    WaitForThread(TID_DB);
    EXPECT_TRUE(got_it);
  
    *ran_test = true;
  }
  
  void PostDelayedTask2(bool* ran_test) {
    // Currently on the FILE thread.
    EXPECT_FALSE(CefCurrentlyOn(TID_DB));
  
    bool got_it = false;
    CefPostDelayedTask(
        TID_DB, CefCreateClosureTask(base::Bind(&PostTaskEvent2, &got_it)), 0);
  
    WaitForThread(TID_DB);
    EXPECT_TRUE(got_it);
  
    *ran_test = true;
  }
  
  }  // namespace
  
  TEST(TaskTest, GetForCurrentThread) {
    bool ran_test = false;
    CefPostTask(TID_FILE, CefCreateClosureTask(
                              base::Bind(&GetForCurrentThread, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  
  TEST(TaskTest, GetForThread) {
    bool ran_test = false;
    CefPostTask(TID_FILE,
                CefCreateClosureTask(base::Bind(&GetForThread, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  
  TEST(TaskTest, PostTask1) {
    bool ran_test = false;
    CefPostTask(TID_FILE,
                CefCreateClosureTask(base::Bind(&PostTask1, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  
  TEST(TaskTest, PostDelayedTask1) {
    bool ran_test = false;
    CefPostTask(TID_FILE,
                CefCreateClosureTask(base::Bind(&PostDelayedTask1, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  
  TEST(TaskTest, PostTask2) {
    bool ran_test = false;
    CefPostTask(TID_FILE,
                CefCreateClosureTask(base::Bind(&PostTask2, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  
  TEST(TaskTest, PostDelayedTask2) {
    bool ran_test = false;
    CefPostTask(TID_FILE,
                CefCreateClosureTask(base::Bind(&PostDelayedTask2, &ran_test)));
    WaitForThread(TID_FILE);
    EXPECT_TRUE(ran_test);
  }
  