You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.5 KiB
82 lines
2.5 KiB
// 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::generated::TestService<MixedService1> {
|
|
public:
|
|
StatusWithSize TestRpc(ServerContext&, ConstByteSpan, ByteSpan) {
|
|
return StatusWithSize(123);
|
|
}
|
|
|
|
void TestStreamRpc(ServerContext&,
|
|
const pw_rpc_test_TestRequest&,
|
|
ServerWriter<pw_rpc_test_TestStreamResponse>&) {
|
|
called_streaming_method = true;
|
|
}
|
|
|
|
bool called_streaming_method = false;
|
|
};
|
|
|
|
class MixedService2 : public test::generated::TestService<MixedService2> {
|
|
public:
|
|
Status TestRpc(ServerContext&,
|
|
const pw_rpc_test_TestRequest&,
|
|
pw_rpc_test_TestResponse&) {
|
|
return Status::Unauthenticated();
|
|
}
|
|
|
|
void TestStreamRpc(ServerContext&, ConstByteSpan, RawServerWriter&) {
|
|
called_streaming_method = true;
|
|
}
|
|
|
|
bool called_streaming_method = false;
|
|
};
|
|
|
|
TEST(MixedService1, CallRawMethod) {
|
|
PW_RAW_TEST_METHOD_CONTEXT(MixedService1, TestRpc) context;
|
|
StatusWithSize sws = context.call({});
|
|
EXPECT_TRUE(sws.ok());
|
|
EXPECT_EQ(123u, sws.size());
|
|
}
|
|
|
|
TEST(MixedService1, CallNanopbMethod) {
|
|
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService1, TestStreamRpc) context;
|
|
ASSERT_FALSE(context.service().called_streaming_method);
|
|
context.call({});
|
|
EXPECT_TRUE(context.service().called_streaming_method);
|
|
}
|
|
|
|
TEST(MixedService2, CallNanopbMethod) {
|
|
PW_NANOPB_TEST_METHOD_CONTEXT(MixedService2, TestRpc) context;
|
|
Status status = context.call({});
|
|
EXPECT_EQ(Status::Unauthenticated(), status);
|
|
}
|
|
|
|
TEST(MixedService2, CallRawMethod) {
|
|
PW_RAW_TEST_METHOD_CONTEXT(MixedService2, TestStreamRpc) context;
|
|
ASSERT_FALSE(context.service().called_streaming_method);
|
|
context.call({});
|
|
EXPECT_TRUE(context.service().called_streaming_method);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace pw::rpc
|