Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 1 | // Copyright 2020 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 | |
Wyatt Hepler | 8ec2cf8 | 2021-07-09 17:18:41 -0700 | [diff] [blame] | 15 | #include "pw_rpc/nanopb/client_call.h" |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 16 | |
| 17 | #include "gtest/gtest.h" |
Wyatt Hepler | fa6edcc | 2021-08-20 08:30:08 -0700 | [diff] [blame] | 18 | #include "pw_rpc/internal/test_utils.h" |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 19 | #include "pw_rpc_nanopb_private/internal_test_utils.h" |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 20 | #include "pw_rpc_test_protos/test.pb.h" |
| 21 | |
| 22 | namespace pw::rpc { |
| 23 | namespace { |
| 24 | |
Wyatt Hepler | fa6edcc | 2021-08-20 08:30:08 -0700 | [diff] [blame] | 25 | using internal::ClientContextForTest; |
| 26 | |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 27 | constexpr uint32_t kServiceId = 16; |
| 28 | constexpr uint32_t kUnaryMethodId = 111; |
| 29 | constexpr uint32_t kServerStreamingMethodId = 112; |
| 30 | |
| 31 | class FakeGeneratedServiceClient { |
| 32 | public: |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 33 | static NanopbClientCall<internal::UnaryCallbacks<pw_rpc_test_TestResponse>> |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 34 | TestUnaryRpc( |
| 35 | Channel& channel, |
| 36 | const pw_rpc_test_TestRequest& request, |
| 37 | Function<void(const pw_rpc_test_TestResponse&, Status)> on_response, |
| 38 | Function<void(Status)> on_error = nullptr) { |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 39 | auto call = NanopbClientCall( |
| 40 | &channel, |
| 41 | kServiceId, |
| 42 | kUnaryMethodId, |
| 43 | internal::UnaryCallbacks(std::move(on_response), std::move(on_error)), |
| 44 | pw_rpc_test_TestRequest_fields, |
| 45 | pw_rpc_test_TestResponse_fields); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 46 | call.SendRequest(&request); |
| 47 | return call; |
| 48 | } |
| 49 | |
| 50 | static NanopbClientCall< |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 51 | internal::ServerStreamingCallbacks<pw_rpc_test_TestStreamResponse>> |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 52 | TestServerStreamRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 53 | Channel& channel, |
| 54 | const pw_rpc_test_TestRequest& request, |
| 55 | Function<void(const pw_rpc_test_TestStreamResponse&)> on_response, |
| 56 | Function<void(Status)> on_stream_end, |
| 57 | Function<void(Status)> on_error = nullptr) { |
| 58 | auto call = NanopbClientCall( |
| 59 | &channel, |
| 60 | kServiceId, |
| 61 | kServerStreamingMethodId, |
| 62 | internal::ServerStreamingCallbacks(std::move(on_response), |
| 63 | std::move(on_stream_end), |
| 64 | std::move(on_error)), |
| 65 | pw_rpc_test_TestRequest_fields, |
| 66 | pw_rpc_test_TestStreamResponse_fields); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 67 | call.SendRequest(&request); |
| 68 | return call; |
| 69 | } |
| 70 | }; |
| 71 | |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 72 | TEST(NanopbClientCall, Unary_SendsRequestPacket) { |
| 73 | ClientContextForTest context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 74 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 75 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 76 | context.channel(), {.integer = 123, .status_code = 0}, nullptr); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 77 | |
| 78 | EXPECT_EQ(context.output().packet_count(), 1u); |
| 79 | auto packet = context.output().sent_packet(); |
| 80 | EXPECT_EQ(packet.channel_id(), context.channel().id()); |
| 81 | EXPECT_EQ(packet.service_id(), kServiceId); |
| 82 | EXPECT_EQ(packet.method_id(), kUnaryMethodId); |
| 83 | |
| 84 | PW_DECODE_PB(pw_rpc_test_TestRequest, sent_proto, packet.payload()); |
| 85 | EXPECT_EQ(sent_proto.integer, 123); |
| 86 | } |
| 87 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 88 | class UnaryClientCall : public ::testing::Test { |
| 89 | protected: |
| 90 | Status last_status_ = Status::Unknown(); |
| 91 | Status last_error_ = Status::Unknown(); |
| 92 | int responses_received_ = 0; |
| 93 | int last_response_value_ = 0; |
| 94 | }; |
| 95 | |
| 96 | TEST_F(UnaryClientCall, InvokesCallbackOnValidResponse) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 97 | ClientContextForTest context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 98 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 99 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 100 | context.channel(), |
| 101 | {.integer = 123, .status_code = 0}, |
| 102 | [this](const pw_rpc_test_TestResponse& response, Status status) { |
| 103 | ++responses_received_; |
| 104 | last_status_ = status; |
| 105 | last_response_value_ = response.value; |
| 106 | }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 107 | |
| 108 | PW_ENCODE_PB(pw_rpc_test_TestResponse, response, .value = 42); |
Wyatt Hepler | 1b3da3a | 2021-01-07 13:26:57 -0800 | [diff] [blame] | 109 | context.SendResponse(OkStatus(), response); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 110 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 111 | ASSERT_EQ(responses_received_, 1); |
| 112 | EXPECT_EQ(last_status_, OkStatus()); |
| 113 | EXPECT_EQ(last_response_value_, 42); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 114 | } |
| 115 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 116 | TEST_F(UnaryClientCall, DoesNothingOnNullCallback) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 117 | ClientContextForTest context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 118 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 119 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 120 | context.channel(), {.integer = 123, .status_code = 0}, nullptr); |
| 121 | |
| 122 | PW_ENCODE_PB(pw_rpc_test_TestResponse, response, .value = 42); |
| 123 | context.SendResponse(OkStatus(), response); |
| 124 | |
| 125 | ASSERT_EQ(responses_received_, 0); |
| 126 | } |
| 127 | |
| 128 | TEST_F(UnaryClientCall, InvokesErrorCallbackOnInvalidResponse) { |
| 129 | ClientContextForTest context; |
| 130 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 131 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 132 | context.channel(), |
| 133 | {.integer = 123, .status_code = 0}, |
| 134 | [this](const pw_rpc_test_TestResponse& response, Status status) { |
| 135 | ++responses_received_; |
| 136 | last_status_ = status; |
| 137 | last_response_value_ = response.value; |
| 138 | }, |
| 139 | [this](Status status) { last_error_ = status; }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 140 | |
| 141 | constexpr std::byte bad_payload[]{ |
| 142 | std::byte{0xab}, std::byte{0xcd}, std::byte{0xef}}; |
Wyatt Hepler | 1b3da3a | 2021-01-07 13:26:57 -0800 | [diff] [blame] | 143 | context.SendResponse(OkStatus(), bad_payload); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 144 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 145 | EXPECT_EQ(responses_received_, 0); |
| 146 | EXPECT_EQ(last_error_, Status::DataLoss()); |
Alexei Frolov | d727622 | 2020-10-01 12:41:59 -0700 | [diff] [blame] | 147 | } |
| 148 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 149 | TEST_F(UnaryClientCall, InvokesErrorCallbackOnServerError) { |
Alexei Frolov | d727622 | 2020-10-01 12:41:59 -0700 | [diff] [blame] | 150 | ClientContextForTest context; |
Alexei Frolov | d727622 | 2020-10-01 12:41:59 -0700 | [diff] [blame] | 151 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 152 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 153 | context.channel(), |
| 154 | {.integer = 123, .status_code = 0}, |
| 155 | [this](const pw_rpc_test_TestResponse& response, Status status) { |
| 156 | ++responses_received_; |
| 157 | last_status_ = status; |
| 158 | last_response_value_ = response.value; |
| 159 | }, |
| 160 | [this](Status status) { last_error_ = status; }); |
Alexei Frolov | d727622 | 2020-10-01 12:41:59 -0700 | [diff] [blame] | 161 | |
| 162 | context.SendPacket(internal::PacketType::SERVER_ERROR, Status::NotFound()); |
| 163 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 164 | EXPECT_EQ(responses_received_, 0); |
| 165 | EXPECT_EQ(last_error_, Status::NotFound()); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 166 | } |
| 167 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 168 | TEST_F(UnaryClientCall, DoesNothingOnErrorWithoutCallback) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 169 | ClientContextForTest context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 170 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 171 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 172 | context.channel(), |
| 173 | {.integer = 123, .status_code = 0}, |
| 174 | [this](const pw_rpc_test_TestResponse& response, Status status) { |
| 175 | ++responses_received_; |
| 176 | last_status_ = status; |
| 177 | last_response_value_ = response.value; |
| 178 | }); |
| 179 | |
| 180 | constexpr std::byte bad_payload[]{ |
| 181 | std::byte{0xab}, std::byte{0xcd}, std::byte{0xef}}; |
| 182 | context.SendResponse(OkStatus(), bad_payload); |
| 183 | |
| 184 | EXPECT_EQ(responses_received_, 0); |
| 185 | } |
| 186 | |
| 187 | TEST_F(UnaryClientCall, OnlyReceivesOneResponse) { |
| 188 | ClientContextForTest context; |
| 189 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 190 | auto call = FakeGeneratedServiceClient::TestUnaryRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 191 | context.channel(), |
| 192 | {.integer = 123, .status_code = 0}, |
| 193 | [this](const pw_rpc_test_TestResponse& response, Status status) { |
| 194 | ++responses_received_; |
| 195 | last_status_ = status; |
| 196 | last_response_value_ = response.value; |
| 197 | }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 198 | |
| 199 | PW_ENCODE_PB(pw_rpc_test_TestResponse, r1, .value = 42); |
| 200 | context.SendResponse(Status::Unimplemented(), r1); |
| 201 | PW_ENCODE_PB(pw_rpc_test_TestResponse, r2, .value = 44); |
| 202 | context.SendResponse(Status::OutOfRange(), r2); |
| 203 | PW_ENCODE_PB(pw_rpc_test_TestResponse, r3, .value = 46); |
| 204 | context.SendResponse(Status::Internal(), r3); |
| 205 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 206 | EXPECT_EQ(responses_received_, 1); |
| 207 | EXPECT_EQ(last_status_, Status::Unimplemented()); |
| 208 | EXPECT_EQ(last_response_value_, 42); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 209 | } |
| 210 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 211 | class ServerStreamingClientCall : public ::testing::Test { |
| 212 | protected: |
| 213 | bool active_ = true; |
| 214 | Status stream_status_ = Status::Unknown(); |
| 215 | Status rpc_error_ = Status::Unknown(); |
| 216 | int responses_received_ = 0; |
| 217 | int last_response_number_ = 0; |
| 218 | }; |
| 219 | |
| 220 | TEST_F(ServerStreamingClientCall, SendsRequestPacket) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 221 | ClientContextForTest<128, 128, 99, kServiceId, kServerStreamingMethodId> |
| 222 | context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 223 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 224 | auto call = FakeGeneratedServiceClient::TestServerStreamRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 225 | context.channel(), {.integer = 71, .status_code = 0}, nullptr, nullptr); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 226 | |
| 227 | EXPECT_EQ(context.output().packet_count(), 1u); |
| 228 | auto packet = context.output().sent_packet(); |
| 229 | EXPECT_EQ(packet.channel_id(), context.channel().id()); |
| 230 | EXPECT_EQ(packet.service_id(), kServiceId); |
| 231 | EXPECT_EQ(packet.method_id(), kServerStreamingMethodId); |
| 232 | |
| 233 | PW_DECODE_PB(pw_rpc_test_TestRequest, sent_proto, packet.payload()); |
| 234 | EXPECT_EQ(sent_proto.integer, 71); |
| 235 | } |
| 236 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 237 | TEST_F(ServerStreamingClientCall, InvokesCallbackOnValidResponse) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 238 | ClientContextForTest<128, 128, 99, kServiceId, kServerStreamingMethodId> |
| 239 | context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 240 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 241 | auto call = FakeGeneratedServiceClient::TestServerStreamRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 242 | context.channel(), |
| 243 | {.integer = 71, .status_code = 0}, |
| 244 | [this](const pw_rpc_test_TestStreamResponse& response) { |
| 245 | ++responses_received_; |
| 246 | last_response_number_ = response.number; |
| 247 | }, |
| 248 | [this](Status status) { |
| 249 | active_ = false; |
| 250 | stream_status_ = status; |
| 251 | }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 252 | |
| 253 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r1, .chunk = {}, .number = 11u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 254 | context.SendServerStream(r1); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 255 | EXPECT_TRUE(active_); |
| 256 | EXPECT_EQ(responses_received_, 1); |
| 257 | EXPECT_EQ(last_response_number_, 11); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 258 | |
| 259 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r2, .chunk = {}, .number = 22u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 260 | context.SendServerStream(r2); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 261 | EXPECT_TRUE(active_); |
| 262 | EXPECT_EQ(responses_received_, 2); |
| 263 | EXPECT_EQ(last_response_number_, 22); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 264 | |
| 265 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r3, .chunk = {}, .number = 33u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 266 | context.SendServerStream(r3); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 267 | EXPECT_TRUE(active_); |
| 268 | EXPECT_EQ(responses_received_, 3); |
| 269 | EXPECT_EQ(last_response_number_, 33); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 270 | } |
| 271 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 272 | TEST_F(ServerStreamingClientCall, InvokesStreamEndOnFinish) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 273 | ClientContextForTest<128, 128, 99, kServiceId, kServerStreamingMethodId> |
| 274 | context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 275 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 276 | auto call = FakeGeneratedServiceClient::TestServerStreamRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 277 | context.channel(), |
| 278 | {.integer = 71, .status_code = 0}, |
| 279 | [this](const pw_rpc_test_TestStreamResponse& response) { |
| 280 | ++responses_received_; |
| 281 | last_response_number_ = response.number; |
| 282 | }, |
| 283 | [this](Status status) { |
| 284 | active_ = false; |
| 285 | stream_status_ = status; |
| 286 | }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 287 | |
| 288 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r1, .chunk = {}, .number = 11u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 289 | context.SendServerStream(r1); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 290 | EXPECT_TRUE(active_); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 291 | |
| 292 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r2, .chunk = {}, .number = 22u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 293 | context.SendServerStream(r2); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 294 | EXPECT_TRUE(active_); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 295 | |
| 296 | // Close the stream. |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 297 | context.SendResponse(Status::NotFound()); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 298 | |
| 299 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r3, .chunk = {}, .number = 33u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 300 | context.SendServerStream(r3); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 301 | EXPECT_FALSE(active_); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 302 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 303 | EXPECT_EQ(responses_received_, 2); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 304 | } |
| 305 | |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 306 | TEST_F(ServerStreamingClientCall, InvokesErrorCallbackOnInvalidResponses) { |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 307 | ClientContextForTest<128, 128, 99, kServiceId, kServerStreamingMethodId> |
| 308 | context; |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 309 | |
Wyatt Hepler | 8211018 | 2021-08-13 11:43:19 -0700 | [diff] [blame] | 310 | auto call = FakeGeneratedServiceClient::TestServerStreamRpc( |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 311 | context.channel(), |
| 312 | {.integer = 71, .status_code = 0}, |
| 313 | [this](const pw_rpc_test_TestStreamResponse& response) { |
| 314 | ++responses_received_; |
| 315 | last_response_number_ = response.number; |
| 316 | }, |
| 317 | nullptr, |
| 318 | [this](Status error) { rpc_error_ = error; }); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 319 | |
| 320 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r1, .chunk = {}, .number = 11u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 321 | context.SendServerStream(r1); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 322 | EXPECT_TRUE(active_); |
| 323 | EXPECT_EQ(responses_received_, 1); |
| 324 | EXPECT_EQ(last_response_number_, 11); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 325 | |
| 326 | constexpr std::byte bad_payload[]{ |
| 327 | std::byte{0xab}, std::byte{0xcd}, std::byte{0xef}}; |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 328 | context.SendServerStream(bad_payload); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 329 | EXPECT_EQ(responses_received_, 1); |
| 330 | EXPECT_EQ(rpc_error_, Status::DataLoss()); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 331 | |
| 332 | PW_ENCODE_PB(pw_rpc_test_TestStreamResponse, r2, .chunk = {}, .number = 22u); |
Wyatt Hepler | 5ba8064 | 2021-06-18 12:56:17 -0700 | [diff] [blame] | 333 | context.SendServerStream(r2); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 334 | EXPECT_TRUE(active_); |
| 335 | EXPECT_EQ(responses_received_, 2); |
| 336 | EXPECT_EQ(last_response_number_, 22); |
Alexei Frolov | d727622 | 2020-10-01 12:41:59 -0700 | [diff] [blame] | 337 | |
| 338 | context.SendPacket(internal::PacketType::SERVER_ERROR, Status::NotFound()); |
Alexei Frolov | bebba90 | 2021-06-09 17:03:52 -0700 | [diff] [blame] | 339 | EXPECT_EQ(responses_received_, 2); |
| 340 | EXPECT_EQ(rpc_error_, Status::NotFound()); |
Alexei Frolov | 4d2adde | 2020-08-04 10:19:24 -0700 | [diff] [blame] | 341 | } |
| 342 | |
| 343 | } // namespace |
| 344 | } // namespace pw::rpc |