| // Copyright 2014 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 "chrome/browser/sync_file_system/drive_backend/callback_helper.h" |
| |
| #include "base/location.h" |
| #include "base/run_loop.h" |
| #include "base/single_thread_task_runner.h" |
| #include "base/test/scoped_task_environment.h" |
| #include "base/threading/thread.h" |
| #include "base/threading/thread_task_runner_handle.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace sync_file_system { |
| namespace drive_backend { |
| |
| namespace { |
| |
| void SimpleCallback(bool* called, int) { |
| ASSERT_TRUE(called); |
| EXPECT_FALSE(*called); |
| *called = true; |
| } |
| |
| void CallbackWithPassed(bool* called, std::unique_ptr<int>) { |
| ASSERT_TRUE(called); |
| EXPECT_FALSE(*called); |
| *called = true; |
| } |
| |
| void VerifyCalledOnTaskRunner(base::TaskRunner* task_runner, |
| bool* called) { |
| ASSERT_TRUE(called); |
| ASSERT_TRUE(task_runner); |
| |
| EXPECT_TRUE(task_runner->RunsTasksInCurrentSequence()); |
| EXPECT_FALSE(*called); |
| *called = true; |
| } |
| |
| } // namespace |
| |
| TEST(DriveBackendCallbackHelperTest, BasicTest) { |
| base::test::ScopedTaskEnvironment scoped_task_environment; |
| |
| bool called = false; |
| RelayCallbackToCurrentThread( |
| FROM_HERE, |
| base::Bind(&SimpleCallback, &called)).Run(0); |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| |
| called = false; |
| RelayCallbackToCurrentThread(FROM_HERE, |
| base::Bind(&CallbackWithPassed, &called)) |
| .Run(std::unique_ptr<int>(new int)); |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| } |
| |
| TEST(DriveBackendCallbackHelperTest, RunOnOtherThreadTest) { |
| base::test::ScopedTaskEnvironment scoped_task_environment; |
| base::Thread thread("WorkerThread"); |
| thread.Start(); |
| |
| scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner = |
| base::ThreadTaskRunnerHandle::Get(); |
| scoped_refptr<base::SequencedTaskRunner> worker_task_runner = |
| thread.task_runner(); |
| |
| bool called = false; |
| base::RunLoop run_loop; |
| worker_task_runner->PostTask( |
| FROM_HERE, RelayCallbackToTaskRunner( |
| ui_task_runner.get(), FROM_HERE, |
| base::Bind(&VerifyCalledOnTaskRunner, |
| base::RetainedRef(ui_task_runner), &called))); |
| worker_task_runner->PostTask( |
| FROM_HERE, |
| RelayCallbackToTaskRunner( |
| ui_task_runner.get(), FROM_HERE, run_loop.QuitClosure())); |
| run_loop.Run(); |
| EXPECT_TRUE(called); |
| |
| thread.Stop(); |
| base::RunLoop().RunUntilIdle(); |
| } |
| |
| TEST(DriveBackendCallbackHelperTest, PassNullFunctionTest) { |
| base::test::ScopedTaskEnvironment scoped_task_environment; |
| base::Closure closure = RelayCallbackToCurrentThread( |
| FROM_HERE, |
| base::Closure()); |
| EXPECT_TRUE(closure.is_null()); |
| } |
| |
| } // namespace drive_backend |
| } // namespace sync_file_system |