1 // Copyright 2021 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/nanopb/fake_channel_output.h"
16
17 #include <array>
18 #include <cstddef>
19 #include <memory>
20
21 #include "pw_rpc/channel.h"
22 #include "pw_rpc/internal/packet.h"
23 #include "pw_rpc_test_protos/test.rpc.pb.h"
24 #include "pw_unit_test/framework.h"
25
26 namespace pw::rpc::internal::test {
27 namespace {
28
29 using rpc::test::pw_rpc::nanopb::TestService;
30 using Info = internal::MethodInfo<TestService::TestUnaryRpc>;
31
TEST(NanopbFakeChannelOutput,Requests)32 TEST(NanopbFakeChannelOutput, Requests) {
33 NanopbFakeChannelOutput<1> output;
34
35 std::byte payload_buffer[32] = {};
36 constexpr Info::Request request{.integer = -100, .status_code = 5};
37 const StatusWithSize payload =
38 Info::serde().request().Encode(&request, payload_buffer);
39 ASSERT_TRUE(payload.ok());
40
41 std::array<std::byte, 128> buffer;
42
43 auto packet = Packet(pwpb::PacketType::REQUEST,
44 1,
45 Info::kServiceId,
46 Info::kMethodId,
47 999,
48 span(payload_buffer, payload.size()))
49 .Encode(buffer);
50 ASSERT_TRUE(packet.ok());
51
52 ASSERT_EQ(OkStatus(), output.Send(span(buffer).first(packet->size())));
53
54 ASSERT_TRUE(output.responses<TestService::TestUnaryRpc>().empty());
55 ASSERT_EQ(output.requests<TestService::TestUnaryRpc>().size(), 1u);
56
57 Info::Request sent = output.requests<TestService::TestUnaryRpc>().front();
58 EXPECT_EQ(sent.integer, -100);
59 EXPECT_EQ(sent.status_code, 5u);
60 }
61
TEST(NanopbFakeChannelOutput,Responses)62 TEST(NanopbFakeChannelOutput, Responses) {
63 NanopbFakeChannelOutput<1> output;
64
65 std::byte payload_buffer[32] = {};
66 constexpr Info::Response response{.value = -9876, .repeated_field = {}};
67 const StatusWithSize payload =
68 Info::serde().response().Encode(&response, payload_buffer);
69 ASSERT_TRUE(payload.ok());
70
71 std::array<std::byte, 128> buffer;
72
73 auto packet = Packet(pwpb::PacketType::RESPONSE,
74 1,
75 Info::kServiceId,
76 Info::kMethodId,
77 999,
78 span(payload_buffer, payload.size()))
79 .Encode(buffer);
80 ASSERT_TRUE(packet.ok());
81
82 ASSERT_EQ(OkStatus(), output.Send(span(buffer).first(packet->size())));
83
84 ASSERT_EQ(output.responses<TestService::TestUnaryRpc>().size(), 1u);
85 ASSERT_TRUE(output.requests<TestService::TestUnaryRpc>().empty());
86
87 Info::Response sent = output.responses<TestService::TestUnaryRpc>().front();
88 EXPECT_EQ(sent.value, -9876);
89 }
90
91 } // namespace
92 } // namespace pw::rpc::internal::test
93