third_party/pigweed/repo/pw_rpc/channel_test.cc

   1 // Copyright 2020 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/channel.h"
  16
  17 #include "gtest/gtest.h"
  18 #include "pw_rpc/internal/packet.h"
  19 #include "pw_rpc_private/internal_test_utils.h"
  20
  21 namespace pw::rpc::internal {
  22 namespace {
  23
  24 using std::byte;
  25
  26 TEST(ChannelOutput, Name) {
  27   class NameTester : public ChannelOutput {
  28    public:
  29     NameTester(const char* name) : ChannelOutput(name) {}
  30     std::span<std::byte> AcquireBuffer() override { return {}; }
  31     Status SendAndReleaseBuffer(std::span<const std::byte>) override {
  32       return OkStatus();
  33     }
  34   };
  35
  36   EXPECT_STREQ("hello_world", NameTester("hello_world").name());
  37   EXPECT_EQ(nullptr, NameTester(nullptr).name());
  38 }
  39
  40 constexpr Packet kTestPacket(PacketType::RESPONSE, 1, 42, 100);
  41 const size_t kReservedSize = 2 /* type */ + 2 /* channel */ + 5 /* service */ +
  42                              5 /* method */ + 2 /* payload key */ +
  43                              2 /* status */;
  44
  45 TEST(Channel, TestPacket_ReservedSizeMatchesMinEncodedSizeBytes) {
  46   EXPECT_EQ(kReservedSize, kTestPacket.MinEncodedSizeBytes());
  47 }
  48
  49 TEST(Channel, OutputBuffer_EmptyBuffer) {
  50   TestOutput<0> output;
  51   internal::Channel channel(100, &output);
  52
  53   Channel::OutputBuffer buffer = channel.AcquireBuffer();
  54   EXPECT_TRUE(buffer.payload(kTestPacket).empty());
  55 }
  56
  57 TEST(Channel, OutputBuffer_TooSmall) {
  58   TestOutput<kReservedSize - 1> output;
  59   internal::Channel channel(100, &output);
  60
  61   Channel::OutputBuffer output_buffer = channel.AcquireBuffer();
  62   EXPECT_TRUE(output_buffer.payload(kTestPacket).empty());
  63
  64   EXPECT_EQ(Status::Internal(), channel.Send(output_buffer, kTestPacket));
  65 }
  66
  67 TEST(Channel, OutputBuffer_ExactFit) {
  68   TestOutput<kReservedSize> output;
  69   internal::Channel channel(100, &output);
  70
  71   Channel::OutputBuffer output_buffer(channel.AcquireBuffer());
  72   const std::span payload = output_buffer.payload(kTestPacket);
  73
  74   EXPECT_EQ(payload.size(), output.buffer().size() - kReservedSize);
  75   EXPECT_EQ(output.buffer().data() + kReservedSize, payload.data());
  76
  77   EXPECT_EQ(OkStatus(), channel.Send(output_buffer, kTestPacket));
  78 }
  79
  80 TEST(Channel, OutputBuffer_PayloadDoesNotFit_ReportsError) {
  81   TestOutput<kReservedSize> output;
  82   internal::Channel channel(100, &output);
  83
  84   Packet packet = kTestPacket;
  85   byte data[1] = {};
  86   packet.set_payload(data);
  87
  88   EXPECT_EQ(Status::Internal(), channel.Send(packet));
  89 }
  90
  91 TEST(Channel, OutputBuffer_ExtraRoom) {
  92   TestOutput<kReservedSize * 3> output;
  93   internal::Channel channel(100, &output);
  94
  95   Channel::OutputBuffer output_buffer = channel.AcquireBuffer();
  96   const std::span payload = output_buffer.payload(kTestPacket);
  97
  98   EXPECT_EQ(payload.size(), output.buffer().size() - kReservedSize);
  99   EXPECT_EQ(output.buffer().data() + kReservedSize, payload.data());
 100
 101   EXPECT_EQ(OkStatus(), channel.Send(output_buffer, kTestPacket));
 102 }
 103
 104 TEST(Channel, OutputBuffer_ReturnsStatusFromChannelOutputSend) {
 105   TestOutput<kReservedSize * 3> output;
 106   internal::Channel channel(100, &output);
 107
 108   Channel::OutputBuffer output_buffer = channel.AcquireBuffer();
 109   output.set_send_status(Status::Aborted());
 110
 111   EXPECT_EQ(Status::Aborted(), channel.Send(output_buffer, kTestPacket));
 112 }
 113
 114 }  // namespace
 115 }  // namespace pw::rpc::internal