93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
// Copyright 2021 The Pigweed Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
// use this file except in compliance with the License. You may obtain a copy of
|
|
// the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
// License for the specific language governing permissions and limitations under
|
|
// the License.
|
|
|
|
#include "pw_rpc/nanopb/fake_channel_output.h"
|
|
|
|
#include <array>
|
|
#include <cstddef>
|
|
#include <memory>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "pw_rpc/internal/channel.h"
|
|
#include "pw_rpc/internal/packet.h"
|
|
#include "pw_rpc_test_protos/test.rpc.pb.h"
|
|
|
|
namespace pw::rpc::internal::test {
|
|
namespace {
|
|
|
|
using rpc::test::pw_rpc::nanopb::TestService;
|
|
using Info = internal::MethodInfo<TestService::TestUnaryRpc>;
|
|
|
|
TEST(NanopbFakeChannelOutput, Requests) {
|
|
NanopbFakeChannelOutput<1> output;
|
|
|
|
std::byte payload_buffer[32] = {};
|
|
constexpr Info::Request request{.integer = -100, .status_code = 5};
|
|
const StatusWithSize payload =
|
|
Info::serde().EncodeRequest(&request, payload_buffer);
|
|
ASSERT_TRUE(payload.ok());
|
|
|
|
std::array<std::byte, 128> buffer;
|
|
|
|
auto packet = Packet(PacketType::REQUEST,
|
|
1,
|
|
Info::kServiceId,
|
|
Info::kMethodId,
|
|
999,
|
|
std::span(payload_buffer, payload.size()))
|
|
.Encode(buffer);
|
|
ASSERT_TRUE(packet.ok());
|
|
|
|
ASSERT_EQ(OkStatus(), output.Send(std::span(buffer).first(packet->size())));
|
|
|
|
ASSERT_TRUE(output.responses<TestService::TestUnaryRpc>().empty());
|
|
ASSERT_EQ(output.requests<TestService::TestUnaryRpc>().size(), 1u);
|
|
|
|
Info::Request sent = output.requests<TestService::TestUnaryRpc>().front();
|
|
EXPECT_EQ(sent.integer, -100);
|
|
EXPECT_EQ(sent.status_code, 5u);
|
|
}
|
|
|
|
TEST(NanopbFakeChannelOutput, Responses) {
|
|
NanopbFakeChannelOutput<1> output;
|
|
|
|
std::byte payload_buffer[32] = {};
|
|
constexpr Info::Response response{.value = -9876, .repeated_field = {}};
|
|
const StatusWithSize payload =
|
|
Info::serde().EncodeResponse(&response, payload_buffer);
|
|
ASSERT_TRUE(payload.ok());
|
|
|
|
std::array<std::byte, 128> buffer;
|
|
|
|
auto packet = Packet(PacketType::RESPONSE,
|
|
1,
|
|
Info::kServiceId,
|
|
Info::kMethodId,
|
|
999,
|
|
std::span(payload_buffer, payload.size()))
|
|
.Encode(buffer);
|
|
ASSERT_TRUE(packet.ok());
|
|
|
|
ASSERT_EQ(OkStatus(), output.Send(std::span(buffer).first(packet->size())));
|
|
|
|
ASSERT_EQ(output.responses<TestService::TestUnaryRpc>().size(), 1u);
|
|
ASSERT_TRUE(output.requests<TestService::TestUnaryRpc>().empty());
|
|
|
|
Info::Response sent = output.responses<TestService::TestUnaryRpc>().front();
|
|
EXPECT_EQ(sent.value, -9876);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace pw::rpc::internal::test
|