1 // Copyright 2019 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/task/post_job.h"
6
7 #include <atomic>
8 #include <iterator>
9 #include <numeric>
10
11 #include "base/task/test_task_traits_extension.h"
12 #include "base/test/bind.h"
13 #include "base/test/gtest_util.h"
14 #include "base/test/task_environment.h"
15 #include "testing/gmock/include/gmock/gmock.h"
16 #include "testing/gtest/include/gtest/gtest.h"
17
18 namespace base {
19
TEST(PostJobTest,PostJobSimple)20 TEST(PostJobTest, PostJobSimple) {
21 test::TaskEnvironment task_environment;
22 std::atomic_size_t num_tasks_to_run(4);
23 auto handle = PostJob(
24 FROM_HERE, {},
25 BindLambdaForTesting([&](JobDelegate* delegate) { --num_tasks_to_run; }),
26 BindLambdaForTesting(
27 [&](size_t /*worker_count*/) -> size_t { return num_tasks_to_run; }));
28 handle.Join();
29 DCHECK_EQ(num_tasks_to_run, 0U);
30 }
31
TEST(PostJobTest,PostJobExtension)32 TEST(PostJobTest, PostJobExtension) {
33 testing::FLAGS_gtest_death_test_style = "threadsafe";
34 EXPECT_DCHECK_DEATH({
35 auto handle = PostJob(
36 FROM_HERE, TestExtensionBoolTrait(),
37 BindRepeating([](JobDelegate* delegate) {}),
38 BindRepeating([](size_t /*worker_count*/) -> size_t { return 0; }));
39 });
40 }
41
42 // Verify that concurrent accesses with task_id as the only form of
43 // synchronisation doesn't trigger a race.
TEST(PostJobTest,TaskIds)44 TEST(PostJobTest, TaskIds) {
45 static constexpr size_t kNumConcurrentThreads = 2;
46 static constexpr size_t kNumTasksToRun = 1000;
47 base::test::TaskEnvironment task_environment;
48
49 size_t concurrent_array[kNumConcurrentThreads] = {0};
50 std::atomic_size_t remaining_tasks{kNumTasksToRun};
51 base::JobHandle handle = base::PostJob(
52 FROM_HERE, {base::ThreadPool()},
53 BindLambdaForTesting([&](base::JobDelegate* job) {
54 uint8_t id = job->GetTaskId();
55 size_t& slot = concurrent_array[id];
56 slot++;
57 --remaining_tasks;
58 }),
59 BindLambdaForTesting([&remaining_tasks](size_t) {
60 return std::min(remaining_tasks.load(), kNumConcurrentThreads);
61 }));
62 handle.Join();
63 EXPECT_EQ(kNumTasksToRun, std::accumulate(std::begin(concurrent_array),
64 std::end(concurrent_array), 0U));
65 }
66
67 } // namespace base
68