You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
2.9 KiB
126 lines
2.9 KiB
// 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_runner_util.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "base/bind.h"
|
|
#include "base/location.h"
|
|
#include "base/message_loop/message_loop.h"
|
|
#include "base/run_loop.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::unique_ptr<Foo>(new 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());
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(TaskRunnerHelpersTest, PostTaskAndReplyWithResult) {
|
|
int result = 0;
|
|
|
|
MessageLoop message_loop;
|
|
PostTaskAndReplyWithResult(message_loop.task_runner().get(), FROM_HERE,
|
|
Bind(&ReturnFourtyTwo),
|
|
Bind(&StoreValue, &result));
|
|
|
|
RunLoop().RunUntilIdle();
|
|
|
|
EXPECT_EQ(42, result);
|
|
}
|
|
|
|
TEST(TaskRunnerHelpersTest, PostTaskAndReplyWithResultImplicitConvert) {
|
|
double result = 0;
|
|
|
|
MessageLoop message_loop;
|
|
PostTaskAndReplyWithResult(message_loop.task_runner().get(), FROM_HERE,
|
|
Bind(&ReturnFourtyTwo),
|
|
Bind(&StoreDoubleValue, &result));
|
|
|
|
RunLoop().RunUntilIdle();
|
|
|
|
EXPECT_DOUBLE_EQ(42.0, result);
|
|
}
|
|
|
|
TEST(TaskRunnerHelpersTest, PostTaskAndReplyWithResultPassed) {
|
|
g_foo_destruct_count = 0;
|
|
g_foo_free_count = 0;
|
|
|
|
MessageLoop message_loop;
|
|
PostTaskAndReplyWithResult(message_loop.task_runner().get(), FROM_HERE,
|
|
Bind(&CreateFoo), Bind(&ExpectFoo));
|
|
|
|
RunLoop().RunUntilIdle();
|
|
|
|
EXPECT_EQ(1, g_foo_destruct_count);
|
|
EXPECT_EQ(0, g_foo_free_count);
|
|
}
|
|
|
|
TEST(TaskRunnerHelpersTest, PostTaskAndReplyWithResultPassedFreeProc) {
|
|
g_foo_destruct_count = 0;
|
|
g_foo_free_count = 0;
|
|
|
|
MessageLoop message_loop;
|
|
PostTaskAndReplyWithResult(message_loop.task_runner().get(), FROM_HERE,
|
|
Bind(&CreateScopedFoo), Bind(&ExpectScopedFoo));
|
|
|
|
RunLoop().RunUntilIdle();
|
|
|
|
EXPECT_EQ(1, g_foo_destruct_count);
|
|
EXPECT_EQ(1, g_foo_free_count);
|
|
}
|
|
|
|
} // namespace base
|