1 | // Copyright 2019 The Marl Authors. |
2 | // |
3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
4 | // you may not use this file except in compliance with the License. |
5 | // You may obtain a copy of the License at |
6 | // |
7 | // https://www.apache.org/licenses/LICENSE-2.0 |
8 | // |
9 | // Unless required by applicable law or agreed to in writing, software |
10 | // distributed under the License is distributed on an "AS IS" BASIS, |
11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | // See the License for the specific language governing permissions and |
13 | // limitations under the License. |
14 | |
15 | #include "gmock/gmock.h" |
16 | #include "gtest/gtest.h" |
17 | |
18 | #include "marl/scheduler.h" |
19 | |
20 | // SchedulerParams holds Scheduler construction parameters for testing. |
21 | struct SchedulerParams { |
22 | int numWorkerThreads; |
23 | |
24 | friend std::ostream& operator<<(std::ostream& os, |
25 | const SchedulerParams& params) { |
26 | return os << "SchedulerParams{" |
27 | << "numWorkerThreads: " << params.numWorkerThreads << "}" ; |
28 | } |
29 | }; |
30 | |
31 | // WithoutBoundScheduler is a test fixture that does not bind a scheduler. |
32 | class WithoutBoundScheduler : public testing::Test { |
33 | public: |
34 | void SetUp() override { |
35 | allocator = new marl::TrackedAllocator(marl::Allocator::Default); |
36 | } |
37 | |
38 | void TearDown() override { |
39 | auto stats = allocator->stats(); |
40 | ASSERT_EQ(stats.numAllocations(), 0U); |
41 | ASSERT_EQ(stats.bytesAllocated(), 0U); |
42 | delete allocator; |
43 | } |
44 | |
45 | marl::TrackedAllocator* allocator = nullptr; |
46 | }; |
47 | |
48 | // WithBoundScheduler is a parameterized test fixture that performs tests with |
49 | // a bound scheduler using a number of different configurations. |
50 | class WithBoundScheduler : public testing::TestWithParam<SchedulerParams> { |
51 | public: |
52 | void SetUp() override { |
53 | allocator = new marl::TrackedAllocator(marl::Allocator::Default); |
54 | |
55 | auto& params = GetParam(); |
56 | |
57 | marl::Scheduler::Config cfg; |
58 | cfg.setAllocator(allocator); |
59 | cfg.setWorkerThreadCount(params.numWorkerThreads); |
60 | cfg.setFiberStackSize(0x10000); |
61 | |
62 | auto scheduler = new marl::Scheduler(cfg); |
63 | scheduler->bind(); |
64 | } |
65 | |
66 | void TearDown() override { |
67 | auto scheduler = marl::Scheduler::get(); |
68 | scheduler->unbind(); |
69 | delete scheduler; |
70 | |
71 | auto stats = allocator->stats(); |
72 | ASSERT_EQ(stats.numAllocations(), 0U); |
73 | ASSERT_EQ(stats.bytesAllocated(), 0U); |
74 | delete allocator; |
75 | } |
76 | |
77 | marl::TrackedAllocator* allocator = nullptr; |
78 | }; |
79 | |