123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170 |
- // Copyright (c) 2012 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 "base/task/task_runner_util.h"
- #include <memory>
- #include <utility>
- #include "base/bind.h"
- #include "base/location.h"
- #include "base/run_loop.h"
- #include "base/test/task_environment.h"
- #include "base/threading/thread_task_runner_handle.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace base {
- namespace {
- int ReturnFourtyTwo() {
- return 42;
- }
- void StoreValue(int* destination, int value) {
- *destination = value;
- }
- void StoreDoubleValue(double* destination, double value) {
- *destination = value;
- }
- int g_foo_destruct_count = 0;
- int g_foo_free_count = 0;
- struct Foo {
- ~Foo() { ++g_foo_destruct_count; }
- };
- std::unique_ptr<Foo> CreateFoo() {
- return std::make_unique<Foo>();
- }
- void ExpectFoo(std::unique_ptr<Foo> foo) {
- EXPECT_TRUE(foo.get());
- std::unique_ptr<Foo> local_foo(std::move(foo));
- EXPECT_TRUE(local_foo.get());
- EXPECT_FALSE(foo.get());
- }
- struct FooDeleter {
- void operator()(Foo* foo) const {
- ++g_foo_free_count;
- delete foo;
- }
- };
- std::unique_ptr<Foo, FooDeleter> CreateScopedFoo() {
- return std::unique_ptr<Foo, FooDeleter>(new Foo);
- }
- void ExpectScopedFoo(std::unique_ptr<Foo, FooDeleter> foo) {
- EXPECT_TRUE(foo.get());
- std::unique_ptr<Foo, FooDeleter> local_foo(std::move(foo));
- EXPECT_TRUE(local_foo.get());
- EXPECT_FALSE(foo.get());
- }
- struct FooWithoutDefaultConstructor {
- explicit FooWithoutDefaultConstructor(int value) : value(value) {}
- int value;
- };
- FooWithoutDefaultConstructor CreateFooWithoutDefaultConstructor(int value) {
- return FooWithoutDefaultConstructor(value);
- }
- void SaveFooWithoutDefaultConstructor(int* output_value,
- FooWithoutDefaultConstructor input) {
- *output_value = input.value;
- }
- class TaskRunnerTest : public testing::Test {
- public:
- TaskRunnerTest() = default;
- void SetUp() override {
- g_foo_destruct_count = 0;
- g_foo_free_count = 0;
- }
- };
- } // namespace
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResult) {
- int result = 0;
- test::SingleThreadTaskEnvironment task_environment;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindOnce(&ReturnFourtyTwo), BindOnce(&StoreValue, &result));
- RunLoop().RunUntilIdle();
- EXPECT_EQ(42, result);
- }
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResultRepeatingCallbacks) {
- int result = 0;
- test::SingleThreadTaskEnvironment task_environment;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindRepeating(&ReturnFourtyTwo),
- BindRepeating(&StoreValue, &result));
- RunLoop().RunUntilIdle();
- EXPECT_EQ(42, result);
- }
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResultImplicitConvert) {
- double result = 0;
- test::SingleThreadTaskEnvironment task_environment;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindOnce(&ReturnFourtyTwo),
- BindOnce(&StoreDoubleValue, &result));
- RunLoop().RunUntilIdle();
- EXPECT_DOUBLE_EQ(42.0, result);
- }
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResultPassed) {
- test::SingleThreadTaskEnvironment task_environment;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindOnce(&CreateFoo), BindOnce(&ExpectFoo));
- RunLoop().RunUntilIdle();
- EXPECT_EQ(1, g_foo_destruct_count);
- EXPECT_EQ(0, g_foo_free_count);
- }
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResultPassedFreeProc) {
- test::SingleThreadTaskEnvironment task_environment;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindOnce(&CreateScopedFoo), BindOnce(&ExpectScopedFoo));
- RunLoop().RunUntilIdle();
- EXPECT_EQ(1, g_foo_destruct_count);
- EXPECT_EQ(1, g_foo_free_count);
- }
- TEST_F(TaskRunnerTest, PostTaskAndReplyWithResultWithoutDefaultConstructor) {
- const int kSomeVal = 17;
- test::SingleThreadTaskEnvironment task_environment;
- int actual = 0;
- ThreadTaskRunnerHandle::Get()->PostTaskAndReplyWithResult(
- FROM_HERE, BindOnce(&CreateFooWithoutDefaultConstructor, kSomeVal),
- BindOnce(&SaveFooWithoutDefaultConstructor, &actual));
- RunLoop().RunUntilIdle();
- EXPECT_EQ(kSomeVal, actual);
- }
- } // namespace base
|