1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// Copyright 2022 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/pwpb/client_server_testing_threaded.h"
#include "pw_rpc_test_protos/test.rpc.pwpb.h"
#include "pw_sync/binary_semaphore.h"
#include "pw_thread/test_threads.h"
namespace pw::rpc {
namespace {
namespace TestRequest = ::pw::rpc::test::pwpb::TestRequest;
namespace TestResponse = ::pw::rpc::test::pwpb::TestResponse;
namespace TestStreamResponse = ::pw::rpc::test::pwpb::TestStreamResponse;
} // namespace
namespace test {
using GeneratedService = ::pw::rpc::test::pw_rpc::pwpb::TestService;
class TestService final : public GeneratedService::Service<TestService> {
public:
Status TestUnaryRpc(const TestRequest::Message& request,
TestResponse::Message& response) {
response.value = request.integer + 1;
return static_cast<Status::Code>(request.status_code);
}
void TestAnotherUnaryRpc(const TestRequest::Message&,
PwpbUnaryResponder<TestResponse::Message>&) {}
static void TestServerStreamRpc(const TestRequest::Message&,
ServerWriter<TestStreamResponse::Message>&) {}
void TestClientStreamRpc(
ServerReader<TestRequest::Message, TestStreamResponse::Message>&) {}
void TestBidirectionalStreamRpc(
ServerReaderWriter<TestRequest::Message, TestStreamResponse::Message>&) {}
};
} // namespace test
namespace {
class RpcCaller {
public:
void BlockOnResponse(uint32_t i, Client& client, uint32_t channel_id) {
TestRequest::Message request{.integer = i,
.status_code = OkStatus().code()};
auto call = test::GeneratedService::TestUnaryRpc(
client,
channel_id,
request,
[this](const TestResponse::Message&, Status) { semaphore_.release(); },
[](Status) {});
semaphore_.acquire();
}
private:
pw::sync::BinarySemaphore semaphore_;
};
TEST(PwpbClientServerTestContextThreaded, ReceivesUnaryRpcReponseThreaded) {
PwpbClientServerTestContextThreaded<> ctx(thread::test::TestOptionsThread0());
test::TestService service;
ctx.server().RegisterService(service);
RpcCaller caller;
constexpr auto value = 1;
caller.BlockOnResponse(value, ctx.client(), ctx.channel().id());
const auto request =
ctx.request<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(0);
const auto response =
ctx.response<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(0);
EXPECT_EQ(value, request.integer);
EXPECT_EQ(value + 1, response.value);
}
TEST(PwpbClientServerTestContextThreaded, ReceivesMultipleReponsesThreaded) {
PwpbClientServerTestContextThreaded<> ctx(thread::test::TestOptionsThread0());
test::TestService service;
ctx.server().RegisterService(service);
RpcCaller caller;
constexpr auto value1 = 1;
constexpr auto value2 = 2;
caller.BlockOnResponse(value1, ctx.client(), ctx.channel().id());
caller.BlockOnResponse(value2, ctx.client(), ctx.channel().id());
const auto request1 =
ctx.request<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(0);
const auto request2 =
ctx.request<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(1);
const auto response1 =
ctx.response<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(0);
const auto response2 =
ctx.response<test::pw_rpc::pwpb::TestService::TestUnaryRpc>(1);
EXPECT_EQ(value1, request1.integer);
EXPECT_EQ(value2, request2.integer);
EXPECT_EQ(value1 + 1, response1.value);
EXPECT_EQ(value2 + 1, response2.value);
}
} // namespace
} // namespace pw::rpc
|
