blob: 7d45879fd6383e1003ea4d99cc918823cc76eb09 [file] [log] [blame]
// Copyright 2020 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 "gtest/gtest.h"
#include "pw_rpc/nanopb/test_method_context.h"
#include "pw_rpc/raw/test_method_context.h"
#include "pw_rpc_test_protos/test.rpc.pb.h"
namespace pw::rpc {
namespace {
class MixedService1
: public test::pw_rpc::nanopb::TestService::Service<MixedService1> {
public:
void TestUnaryRpc(ConstByteSpan, RawUnaryResponder& responder) {
std::byte response[5] = {};
ASSERT_EQ(OkStatus(), responder.Finish(response, OkStatus()));
}
void TestAnotherUnaryRpc(const pw_rpc_test_TestRequest&,
NanopbUnaryResponder<pw_rpc_test_TestResponse>&) {
called_async_unary_method = true;
}
void TestServerStreamRpc(const pw_rpc_test_TestRequest&,
ServerWriter<pw_rpc_test_TestStreamResponse>&) {
called_server_streaming_method = true;
}
void TestClientStreamRpc(RawServerReader&) {
called_client_streaming_method = true;
}
void TestBidirectionalStreamRpc(
ServerReaderWriter<pw_rpc_test_TestRequest,
pw_rpc_test_TestStreamResponse>&) {
called_bidirectional_streaming_method = true;
}
bool called_async_unary_method = false;
bool called_server_streaming_method = false;
bool called_client_streaming_method = false;
bool called_bidirectional_streaming_method = false;
};
class MixedService2
: public test::pw_rpc::nanopb::TestService::Service<MixedService2> {
public:
Status TestUnaryRpc(const pw_rpc_test_TestRequest&,
pw_rpc_test_TestResponse&) {
return Status::Unauthenticated();
}
void TestAnotherUnaryRpc(ConstByteSpan, RawUnaryResponder&) {
called_async_unary_method = true;
}
void TestServerStreamRpc(ConstByteSpan, RawServerWriter&) {
called_server_streaming_method = true;
}
void TestClientStreamRpc(
ServerReader<pw_rpc_test_TestRequest, pw_rpc_test_TestStreamResponse>&) {
called_client_streaming_method = true;
}
void TestBidirectionalStreamRpc(RawServerReaderWriter&) {
called_bidirectional_streaming_method = true;
}
bool called_async_unary_method = false;
bool called_server_streaming_method = false;
bool called_client_streaming_method = false;
bool called_bidirectional_streaming_method = false;
};
TEST(MixedService1, CallRawMethod_SyncUnary) {
PW_RAW_TEST_METHOD_CONTEXT(MixedService1, TestUnaryRpc) context;
context.call({});
EXPECT_EQ(OkStatus(), context.status());
EXPECT_EQ(5u, context.response().size());
}
TEST(MixedService1, CallNanopbMethod_AsyncUnary) {
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService1, TestAnotherUnaryRpc) context;
ASSERT_FALSE(context.service().called_async_unary_method);
context.call({});
EXPECT_TRUE(context.service().called_async_unary_method);
}
TEST(MixedService1, CallNanopbMethod_ServerStreaming) {
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService1, TestServerStreamRpc) context;
ASSERT_FALSE(context.service().called_server_streaming_method);
context.call({});
EXPECT_TRUE(context.service().called_server_streaming_method);
}
TEST(MixedService1, CallRawMethod_ClientStreaming) {
PW_RAW_TEST_METHOD_CONTEXT(MixedService1, TestClientStreamRpc) context;
ASSERT_FALSE(context.service().called_client_streaming_method);
context.call();
EXPECT_TRUE(context.service().called_client_streaming_method);
}
TEST(MixedService1, CallNanopbMethod_BidirectionalStreaming) {
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService1, TestBidirectionalStreamRpc)
context;
ASSERT_FALSE(context.service().called_bidirectional_streaming_method);
context.call();
EXPECT_TRUE(context.service().called_bidirectional_streaming_method);
}
TEST(MixedService2, CallNanopbMethod_SyncUnary) {
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService2, TestUnaryRpc) context;
Status status = context.call({});
EXPECT_EQ(Status::Unauthenticated(), status);
}
TEST(MixedService2, CallRawMethod_AsyncUnary) {
PW_RAW_TEST_METHOD_CONTEXT(MixedService2, TestAnotherUnaryRpc) context;
ASSERT_FALSE(context.service().called_async_unary_method);
context.call({});
EXPECT_TRUE(context.service().called_async_unary_method);
}
TEST(MixedService2, CallRawMethod_ServerStreaming) {
PW_RAW_TEST_METHOD_CONTEXT(MixedService2, TestServerStreamRpc) context;
ASSERT_FALSE(context.service().called_server_streaming_method);
context.call({});
EXPECT_TRUE(context.service().called_server_streaming_method);
}
TEST(MixedService2, CallNanopbMethod_ClientStreaming) {
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService2, TestClientStreamRpc) context;
ASSERT_FALSE(context.service().called_client_streaming_method);
context.call();
EXPECT_TRUE(context.service().called_client_streaming_method);
}
TEST(MixedService2, CallRawMethod_BidirectionalStreaming) {
PW_RAW_TEST_METHOD_CONTEXT(MixedService2, TestBidirectionalStreamRpc) context;
ASSERT_FALSE(context.service().called_bidirectional_streaming_method);
context.call();
EXPECT_TRUE(context.service().called_bidirectional_streaming_method);
}
} // namespace
} // namespace pw::rpc