1/*
2 * Copyright (c) Huawei Technologies Co., Ltd. 2021. All rights reserved.
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
16#include <atomic>
17#include <chrono>
18#include <gtest/gtest.h>
19#include <thread>
20
21#include "i_semaphore.h"
22
23namespace {
24using namespace testing::ext;
25
26class SemaphoreTest : public testing::Test {
27protected:
28    static void SetUpTestCase() {}
29    static void TearDownTestCase() {}
30};
31
32/**
33 * @tc.name: SemaphoreTest
34 * @tc.desc: CtorDtor.
35 * @tc.type: FUNC
36 */
37HWTEST_F(SemaphoreTest, CtorDtor, TestSize.Level1)
38{
39    auto semaphore = GetSemaphoreFactory().Create(1);
40    EXPECT_NE(semaphore, nullptr);
41    EXPECT_TRUE(semaphore->TryWait());
42    EXPECT_FALSE(semaphore->TimedWait(1, 0));
43}
44
45/**
46 * @tc.name: SemaphoreTest
47 * @tc.desc: Wait.
48 * @tc.type: FUNC
49 */
50HWTEST_F(SemaphoreTest, Wait, TestSize.Level1)
51{
52    auto semaphore = GetSemaphoreFactory().Create(1);
53    ASSERT_NE(semaphore, nullptr);
54    EXPECT_TRUE(semaphore->Wait());
55    EXPECT_FALSE(semaphore->TryWait());
56}
57
58/**
59 * @tc.name: SemaphoreTest
60 * @tc.desc: Post.
61 * @tc.type: FUNC
62 */
63HWTEST_F(SemaphoreTest, Post, TestSize.Level1)
64{
65    auto semaphore = GetSemaphoreFactory().Create(0);
66    ASSERT_NE(semaphore, nullptr);
67    EXPECT_TRUE(semaphore->Post());
68}
69
70/**
71 * @tc.name: SemaphoreTest
72 * @tc.desc: Post.
73 * @tc.type: FUNC
74 */
75HWTEST_F(SemaphoreTest, WaitPost, TestSize.Level1)
76{
77    auto readySem = GetSemaphoreFactory().Create(0);
78    auto finishSem = GetSemaphoreFactory().Create(0);
79    ASSERT_NE(readySem, nullptr);
80    ASSERT_NE(finishSem, nullptr);
81
82    auto done = std::make_shared<bool>(false);
83    ASSERT_NE(done, nullptr);
84
85    std::thread bgThread([=]() {
86        readySem->Wait();
87        *done = true;
88        finishSem->Post();
89    });
90
91    EXPECT_TRUE(readySem->Post());
92    EXPECT_TRUE(finishSem->Wait());
93    EXPECT_TRUE(*done);
94
95    bgThread.join();
96}
97
98/**
99 * @tc.name: SemaphoreTest
100 * @tc.desc: CreatePosixSemaphoreFactory.
101 * @tc.type: FUNC
102 */
103HWTEST_F(SemaphoreTest, CreatePosixSemaphoreFactory, TestSize.Level1)
104{
105    auto semaphore = GetSemaphoreFactory(POSIX_SEMAPHORE_FACTORY).Create(0);
106    EXPECT_NE(semaphore, nullptr);
107}
108
109/**
110 * @tc.name: SemaphoreTest
111 * @tc.desc: CreatePtheadSemaphoreFactory.
112 * @tc.type: FUNC
113 */
114HWTEST_F(SemaphoreTest, CreatePtheadSemaphoreFactory, TestSize.Level1)
115{
116    auto semaphore = GetSemaphoreFactory(PTHREAD_SEMAPHORE_FACTORY).Create(0);
117    EXPECT_NE(semaphore, nullptr);
118}
119} // namespace