1 // Copyright 2022 The Android Open Source Project
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 // http://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 <gtest/gtest.h>
16
17 #include "aemu/base/threads/WorkerThread.h"
18
19 #include <chrono>
20
21 namespace android {
22 namespace base {
23 namespace {
TEST(WorkerThread,TheReturnedFutureFromEnqueueShouldBeReadyWhenTheWorkerStops)24 TEST(WorkerThread, TheReturnedFutureFromEnqueueShouldBeReadyWhenTheWorkerStops) {
25 struct Item {};
26 WorkerThread<Item> worker([](Item&&) { return WorkerProcessingResult::Stop; });
27 worker.start();
28 worker.enqueue(Item{}).wait();
29 EXPECT_EQ(worker.enqueue(Item{}).wait_for(std::chrono::milliseconds(500)),
30 std::future_status::ready);
31 }
32
TEST(WorkerThread,TheReturnedFutureFromEnqueueShouldBeReadyBeforeTheWorkerStarts)33 TEST(WorkerThread, TheReturnedFutureFromEnqueueShouldBeReadyBeforeTheWorkerStarts) {
34 struct Item {};
35 WorkerThread<Item> worker([](Item&&) { return WorkerProcessingResult::Stop; });
36 EXPECT_EQ(worker.enqueue(Item{}).wait_for(std::chrono::milliseconds(0)),
37 std::future_status::ready);
38 }
39 } // namespace
40 } // namespace base
41 } // namespace android