1 // Copyright 2023 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://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, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_rpc_transport/internal/packet_buffer_queue.h"
16
17 #include <array>
18
19 #include "pw_bytes/span.h"
20 #include "pw_result/result.h"
21 #include "pw_status/status.h"
22 #include "pw_unit_test/framework.h"
23
24 namespace pw::rpc::internal {
25 namespace {
26
27 constexpr size_t kMaxPacketSize = 256;
28
TEST(PacketBufferQueueTest,CopyAndGetPacket)29 TEST(PacketBufferQueueTest, CopyAndGetPacket) {
30 PacketBufferQueue<kMaxPacketSize>::PacketBuffer packet_buffer = {};
31 std::array<std::byte, 42> input{};
32 std::fill(input.begin(), input.end(), std::byte{0x42});
33
34 auto packet = packet_buffer.GetPacket();
35 ASSERT_EQ(packet.status(), OkStatus());
36 EXPECT_EQ(packet->size(), 0ul);
37
38 ASSERT_EQ(packet_buffer.CopyPacket(input), OkStatus());
39
40 packet = packet_buffer.GetPacket();
41 ASSERT_EQ(packet.status(), OkStatus());
42 EXPECT_EQ(packet->size(), 42ul);
43
44 EXPECT_TRUE(std::equal(input.begin(), input.end(), packet->begin()));
45
46 std::array<std::byte, 300> long_input{};
47 EXPECT_EQ(packet_buffer.CopyPacket(long_input), Status::ResourceExhausted());
48 }
49
TEST(PacketBufferQueueTest,PopWhenEmptyFails)50 TEST(PacketBufferQueueTest, PopWhenEmptyFails) {
51 PacketBufferQueue<kMaxPacketSize> queue;
52 EXPECT_EQ(queue.Pop().status(), Status::ResourceExhausted());
53 }
54
TEST(PacketBufferQueueTest,PopAllSucceeds)55 TEST(PacketBufferQueueTest, PopAllSucceeds) {
56 constexpr auto kPacketQueueSize = 3;
57
58 std::array<PacketBufferQueue<kMaxPacketSize>::PacketBuffer, kPacketQueueSize>
59 packets;
60 PacketBufferQueue<kMaxPacketSize> queue(packets);
61
62 for (size_t i = 0; i < kPacketQueueSize; ++i) {
63 EXPECT_EQ(queue.Pop().status(), OkStatus());
64 }
65
66 EXPECT_EQ(queue.Pop().status(), Status::ResourceExhausted());
67 }
68
TEST(PacketQueueTest,PushPopSucceeds)69 TEST(PacketQueueTest, PushPopSucceeds) {
70 PacketBufferQueue<kMaxPacketSize>::PacketBuffer packet_buffer;
71 PacketBufferQueue<kMaxPacketSize> queue;
72
73 pw::Result<PacketBufferQueue<kMaxPacketSize>::PacketBuffer*>
74 popped_packet_buffer = queue.Pop();
75 EXPECT_EQ(popped_packet_buffer.status(), Status::ResourceExhausted());
76
77 queue.Push(packet_buffer);
78 popped_packet_buffer = queue.Pop();
79 EXPECT_EQ(popped_packet_buffer.status(), OkStatus());
80 }
81
82 } // namespace
83 } // namespace pw::rpc::internal
84