1 | /* |
2 | * Copyright 2017-present Facebook, Inc. |
3 | * |
4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | * you may not use this file except in compliance with the License. |
6 | * You may obtain a copy of the License at |
7 | * |
8 | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | * |
10 | * Unless required by applicable law or agreed to in writing, software |
11 | * distributed under the License is distributed on an "AS IS" BASIS, |
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
13 | * See the License for the specific language governing permissions and |
14 | * limitations under the License. |
15 | */ |
16 | |
17 | #include <folly/executors/GlobalExecutor.h> |
18 | #include <folly/executors/IOExecutor.h> |
19 | #include <folly/portability/GTest.h> |
20 | |
21 | using namespace folly; |
22 | |
23 | TEST(GlobalExecutorTest, GlobalCPUExecutor) { |
24 | class DummyExecutor : public folly::Executor { |
25 | public: |
26 | void add(Func f) override { |
27 | f(); |
28 | count++; |
29 | } |
30 | int count{0}; |
31 | }; |
32 | |
33 | // The default CPU executor is a synchronous inline executor, lets verify |
34 | // that work we add is executed |
35 | auto count = 0; |
36 | auto f = [&]() { count++; }; |
37 | |
38 | // Don't explode, we should create the default global CPUExecutor lazily here. |
39 | getCPUExecutor()->add(f); |
40 | EXPECT_EQ(1, count); |
41 | |
42 | { |
43 | auto dummy = std::make_shared<DummyExecutor>(); |
44 | setCPUExecutor(dummy); |
45 | getCPUExecutor()->add(f); |
46 | // Make sure we were properly installed. |
47 | EXPECT_EQ(1, dummy->count); |
48 | EXPECT_EQ(2, count); |
49 | } |
50 | |
51 | // Don't explode, we should restore the default global CPUExecutor because our |
52 | // weak reference to dummy has expired |
53 | getCPUExecutor()->add(f); |
54 | EXPECT_EQ(3, count); |
55 | } |
56 | |
57 | TEST(GlobalExecutorTest, GlobalIOExecutor) { |
58 | class DummyExecutor : public IOExecutor { |
59 | public: |
60 | void add(Func) override { |
61 | count++; |
62 | } |
63 | folly::EventBase* getEventBase() override { |
64 | return nullptr; |
65 | } |
66 | int count{0}; |
67 | }; |
68 | |
69 | auto f = []() {}; |
70 | |
71 | // Don't explode, we should create the default global IOExecutor lazily here. |
72 | getIOExecutor()->add(f); |
73 | |
74 | { |
75 | auto dummy = std::make_shared<DummyExecutor>(); |
76 | setIOExecutor(dummy); |
77 | getIOExecutor()->add(f); |
78 | // Make sure we were properly installed. |
79 | EXPECT_EQ(1, dummy->count); |
80 | } |
81 | |
82 | // Don't explode, we should restore the default global IOExecutor because our |
83 | // weak reference to dummy has expired |
84 | getIOExecutor()->add(f); |
85 | } |
86 | |