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/futures/test/TestExecutor.h> |
18 | #include <folly/portability/GTest.h> |
19 | |
20 | using namespace std; |
21 | using namespace std::chrono; |
22 | using namespace folly; |
23 | |
24 | TEST(TestExecutor, parallel_run) { |
25 | mutex m; |
26 | set<thread::id> ids; |
27 | auto executor = std::make_unique<TestExecutor>(4); |
28 | const auto numThreads = executor->numThreads(); |
29 | EXPECT_EQ(4, numThreads); |
30 | for (auto idx = 0U; idx < numThreads * 10; ++idx) { |
31 | executor->add([&m, &ids]() mutable { |
32 | /* sleep override */ this_thread::sleep_for(milliseconds(100)); |
33 | lock_guard<mutex> lg(m); |
34 | ids.insert(this_thread::get_id()); |
35 | }); |
36 | } |
37 | |
38 | executor = nullptr; |
39 | EXPECT_EQ(ids.size(), numThreads); |
40 | } |
41 |