1 /*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
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 <gtest/gtest.h>
17 #include <thread>
18
19 #include "ring_buffer.h"
20
21 using namespace testing::ext;
22 using namespace Updater;
23
24 namespace OHOS {
25 constexpr uint32_t RING_MAX_LEN = 10240;
26 constexpr uint32_t BYTE_SIZE = 255;
27 uint32_t g_num;
28 bool g_result = true;
29 class RingBufferTest : public testing::Test {
30 public:
SetUpTestCase(void)31 static void SetUpTestCase(void) {}
TearDownTestCase(void)32 static void TearDownTestCase(void) {}
SetUp()33 void SetUp() {}
TearDown()34 void TearDown() {}
TestBody()35 void TestBody() {}
36 };
37
ProducerTask(RingBuffer * ringBuffer)38 void ProducerTask(RingBuffer *ringBuffer)
39 {
40 std::cout << "ring buffer ProducerTask start\n";
41 for (uint32_t i = 0; i < RING_MAX_LEN; i++) {
42 uint8_t buf[4] {}; // 4: test buffer size
43 buf[0] = i % BYTE_SIZE;
44 buf[1] = i / BYTE_SIZE;
45 ringBuffer->Push(buf, sizeof(buf));
46 }
47 }
48
ConsumerTask(RingBuffer * ringBuffer)49 void ConsumerTask(RingBuffer *ringBuffer)
50 {
51 std::cout << "ring buffer ConsumerTask start\n";
52 while (1) {
53 uint8_t buf[4] {}; // 4: test buffer size
54 uint32_t len = 0;
55 ringBuffer->Pop(buf, sizeof(buf), len);
56 if (buf[0] != g_num % BYTE_SIZE || buf[1] != g_num / BYTE_SIZE) {
57 g_result = false;
58 }
59 g_num++;
60 if (g_num == RING_MAX_LEN) {
61 break;
62 }
63 }
64 }
65
66 HWTEST_F(RingBufferTest, ringBufferTest01, TestSize.Level0)
67 {
68 std::cout << "ringBufferTest01 start\n";
69 g_num = 0;
70 RingBuffer ringBuffer;
71 bool ret = ringBuffer.Init(1024, 8);
72 EXPECT_TRUE(ret);
73 std::thread consumer(ConsumerTask, &ringBuffer);
74 std::thread producer(ProducerTask, &ringBuffer);
75 consumer.join();
76 producer.join();
77 EXPECT_EQ(g_num, RING_MAX_LEN);
78 std::cout << "ringBufferTest01 end\n";
79 }
80
81
82 HWTEST_F(RingBufferTest, ringBufferTest02, TestSize.Level0)
83 {
84 std::cout << "ringBufferTest01 start";
85 RingBuffer ringBuffer;
86 bool ret = ringBuffer.Init(1024, 3);
87 EXPECT_FALSE(ret);
88 EXPECT_TRUE(g_result);
89 std::cout << "ringBufferTest02 end";
90 }
91 } // namespace OHOS
92