Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
| 3 | * Copyright 2015, Google Inc. |
| 4 | * All rights reserved. |
| 5 | * |
| 6 | * Redistribution and use in source and binary forms, with or without |
| 7 | * modification, are permitted provided that the following conditions are |
| 8 | * met: |
| 9 | * |
| 10 | * * Redistributions of source code must retain the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer. |
| 12 | * * Redistributions in binary form must reproduce the above |
| 13 | * copyright notice, this list of conditions and the following disclaimer |
| 14 | * in the documentation and/or other materials provided with the |
| 15 | * distribution. |
| 16 | * * Neither the name of Google Inc. nor the names of its |
| 17 | * contributors may be used to endorse or promote products derived from |
| 18 | * this software without specific prior written permission. |
| 19 | * |
| 20 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 21 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 22 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 23 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 24 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 25 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 26 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 27 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 28 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 29 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 30 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 31 | * |
| 32 | */ |
| 33 | |
Jorge Canizales | 9195233 | 2015-07-31 17:50:49 -0700 | [diff] [blame] | 34 | #import "InteropTests.h" |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 35 | |
Jorge Canizales | 9195233 | 2015-07-31 17:50:49 -0700 | [diff] [blame] | 36 | #include <grpc/status.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 37 | |
Makarand Dharmapurikar | de733bd | 2016-05-25 10:38:56 -0700 | [diff] [blame] | 38 | #import <Cronet/Cronet.h> |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 39 | #import <GRPCClient/GRPCCall+ChannelArg.h> |
Jorge Canizales | b2bd067 | 2015-08-01 23:19:11 -0700 | [diff] [blame] | 40 | #import <GRPCClient/GRPCCall+Tests.h> |
Makarand Dharmapurikar | 9b5f640 | 2016-05-20 15:57:34 -0700 | [diff] [blame] | 41 | #import <GRPCClient/GRPCCall+Cronet.h> |
Jorge Canizales | c42a38e | 2015-06-21 14:44:25 -0700 | [diff] [blame] | 42 | #import <ProtoRPC/ProtoRPC.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 43 | #import <RemoteTest/Messages.pbobjc.h> |
| 44 | #import <RemoteTest/Test.pbobjc.h> |
| 45 | #import <RemoteTest/Test.pbrpc.h> |
Jorge Canizales | c42a38e | 2015-06-21 14:44:25 -0700 | [diff] [blame] | 46 | #import <RxLibrary/GRXBufferedPipe.h> |
| 47 | #import <RxLibrary/GRXWriter+Immediate.h> |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 48 | |
| 49 | // Convenience constructors for the generated proto messages: |
| 50 | |
| 51 | @interface RMTStreamingOutputCallRequest (Constructors) |
| 52 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize |
| 53 | requestedResponseSize:(NSNumber *)responseSize; |
| 54 | @end |
| 55 | |
| 56 | @implementation RMTStreamingOutputCallRequest (Constructors) |
| 57 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize |
| 58 | requestedResponseSize:(NSNumber *)responseSize { |
| 59 | RMTStreamingOutputCallRequest *request = [self message]; |
| 60 | RMTResponseParameters *parameters = [RMTResponseParameters message]; |
Yuchen Zeng | afb62f8 | 2016-06-24 19:11:08 -0700 | [diff] [blame] | 61 | parameters.size = responseSize.intValue; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 62 | [request.responseParametersArray addObject:parameters]; |
| 63 | request.payload.body = [NSMutableData dataWithLength:payloadSize.unsignedIntegerValue]; |
| 64 | return request; |
| 65 | } |
| 66 | @end |
| 67 | |
| 68 | @interface RMTStreamingOutputCallResponse (Constructors) |
| 69 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize; |
| 70 | @end |
| 71 | |
| 72 | @implementation RMTStreamingOutputCallResponse (Constructors) |
| 73 | + (instancetype)messageWithPayloadSize:(NSNumber *)payloadSize { |
| 74 | RMTStreamingOutputCallResponse * response = [self message]; |
| 75 | response.payload.type = RMTPayloadType_Compressable; |
| 76 | response.payload.body = [NSMutableData dataWithLength:payloadSize.unsignedIntegerValue]; |
| 77 | return response; |
| 78 | } |
| 79 | @end |
| 80 | |
Jorge Canizales | b2bd067 | 2015-08-01 23:19:11 -0700 | [diff] [blame] | 81 | #pragma mark Tests |
| 82 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 83 | @implementation InteropTests { |
| 84 | RMTTestService *_service; |
| 85 | } |
| 86 | |
Jorge Canizales | 945f72c | 2015-07-31 17:40:59 -0700 | [diff] [blame] | 87 | + (NSString *)host { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 88 | return nil; |
Jorge Canizales | 945f72c | 2015-07-31 17:40:59 -0700 | [diff] [blame] | 89 | } |
Jorge Canizales | ce8ee61 | 2015-07-24 20:52:49 -0700 | [diff] [blame] | 90 | |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 91 | - (int32_t)encodingOverhead { |
| 92 | return 0; |
| 93 | } |
| 94 | |
Muxi Yan | be4fcc4 | 2016-10-13 14:33:08 -0700 | [diff] [blame] | 95 | + (void)setUp { |
| 96 | #ifdef GRPC_COMPILE_WITH_CRONET |
| 97 | // Cronet setup |
| 98 | [Cronet setHttp2Enabled:YES]; |
| 99 | [Cronet start]; |
| 100 | [GRPCCall useCronetWithEngine:[Cronet getGlobalEngine]]; |
| 101 | #endif |
| 102 | } |
| 103 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 104 | - (void)setUp { |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 105 | self.continueAfterFailure = NO; |
| 106 | |
Jorge Canizales | 32fde7a | 2016-07-29 17:31:07 -0700 | [diff] [blame] | 107 | [GRPCCall resetHostSettings]; |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 108 | |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 109 | _service = self.class.host ? [RMTTestService serviceWithHost:self.class.host] : nil; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 110 | } |
| 111 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 112 | - (void)testEmptyUnaryRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 113 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 114 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"EmptyUnary"]; |
| 115 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 116 | GPBEmpty *request = [GPBEmpty message]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 117 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 118 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 119 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 120 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 121 | id expectedResponse = [GPBEmpty message]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 122 | XCTAssertEqualObjects(response, expectedResponse); |
| 123 | |
| 124 | [expectation fulfill]; |
| 125 | }]; |
| 126 | |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 127 | [self waitForExpectationsWithTimeout:4 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 128 | } |
| 129 | |
| 130 | - (void)testLargeUnaryRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 131 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 132 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"LargeUnary"]; |
| 133 | |
| 134 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
| 135 | request.responseType = RMTPayloadType_Compressable; |
| 136 | request.responseSize = 314159; |
| 137 | request.payload.body = [NSMutableData dataWithLength:271828]; |
| 138 | |
| 139 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 140 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 141 | |
| 142 | RMTSimpleResponse *expectedResponse = [RMTSimpleResponse message]; |
| 143 | expectedResponse.payload.type = RMTPayloadType_Compressable; |
| 144 | expectedResponse.payload.body = [NSMutableData dataWithLength:314159]; |
| 145 | XCTAssertEqualObjects(response, expectedResponse); |
| 146 | |
| 147 | [expectation fulfill]; |
| 148 | }]; |
| 149 | |
Jorge Canizales | 83c57cb | 2015-08-09 16:36:49 -0700 | [diff] [blame] | 150 | [self waitForExpectationsWithTimeout:16 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 151 | } |
| 152 | |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 153 | - (void)test4MBResponsesAreAccepted { |
| 154 | XCTAssertNotNil(self.class.host); |
| 155 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"MaxResponseSize"]; |
| 156 | |
| 157 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 158 | const int32_t kPayloadSize = 4 * 1024 * 1024 - self.encodingOverhead; // 4MB - encoding overhead |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 159 | request.responseSize = kPayloadSize; |
| 160 | |
| 161 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 162 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 163 | XCTAssertEqual(response.payload.body.length, kPayloadSize); |
| 164 | [expectation fulfill]; |
| 165 | }]; |
| 166 | |
| 167 | [self waitForExpectationsWithTimeout:16 handler:nil]; |
| 168 | } |
| 169 | |
| 170 | - (void)testResponsesOverMaxSizeFailWithActionableMessage { |
| 171 | XCTAssertNotNil(self.class.host); |
| 172 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ResponseOverMaxSize"]; |
| 173 | |
| 174 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
Jorge Canizales | 9e83d7e | 2016-08-01 12:51:31 -0700 | [diff] [blame] | 175 | const int32_t kPayloadSize = 4 * 1024 * 1024 - self.encodingOverhead + 1; // 1B over max size |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 176 | request.responseSize = kPayloadSize; |
| 177 | |
| 178 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
Jorge Canizales | ec0743e | 2016-07-29 01:50:06 -0700 | [diff] [blame] | 179 | // TODO(jcanizales): Catch the error and rethrow it with an actionable message: |
| 180 | // - Use +[GRPCCall setResponseSizeLimit:forHost:] to set a higher limit. |
| 181 | // - If you're developing the server, consider using response streaming, or let clients filter |
| 182 | // responses by setting a google.protobuf.FieldMask in the request: |
| 183 | // https://github.com/google/protobuf/blob/master/src/google/protobuf/field_mask.proto |
Mark D. Roth | a4d9ee2 | 2016-08-31 13:17:26 -0700 | [diff] [blame] | 184 | XCTAssertEqualObjects(error.localizedDescription, @"Received message larger than max (4194305 vs. 4194304)"); |
Jorge Canizales | 6af4add | 2016-07-28 23:05:18 -0700 | [diff] [blame] | 185 | [expectation fulfill]; |
| 186 | }]; |
| 187 | |
| 188 | [self waitForExpectationsWithTimeout:16 handler:nil]; |
| 189 | } |
| 190 | |
| 191 | - (void)testResponsesOver4MBAreAcceptedIfOptedIn { |
| 192 | XCTAssertNotNil(self.class.host); |
| 193 | __weak XCTestExpectation *expectation = |
| 194 | [self expectationWithDescription:@"HigherResponseSizeLimit"]; |
| 195 | |
| 196 | RMTSimpleRequest *request = [RMTSimpleRequest message]; |
| 197 | const size_t kPayloadSize = 5 * 1024 * 1024; // 5MB |
| 198 | request.responseSize = kPayloadSize; |
| 199 | |
| 200 | [GRPCCall setResponseSizeLimit:6 * 1024 * 1024 forHost:self.class.host]; |
| 201 | |
| 202 | [_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
| 203 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 204 | XCTAssertEqual(response.payload.body.length, kPayloadSize); |
| 205 | [expectation fulfill]; |
| 206 | }]; |
| 207 | |
| 208 | [self waitForExpectationsWithTimeout:16 handler:nil]; |
| 209 | } |
| 210 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 211 | - (void)testClientStreamingRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 212 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 213 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ClientStreaming"]; |
| 214 | |
| 215 | RMTStreamingInputCallRequest *request1 = [RMTStreamingInputCallRequest message]; |
| 216 | request1.payload.body = [NSMutableData dataWithLength:27182]; |
| 217 | |
| 218 | RMTStreamingInputCallRequest *request2 = [RMTStreamingInputCallRequest message]; |
| 219 | request2.payload.body = [NSMutableData dataWithLength:8]; |
| 220 | |
| 221 | RMTStreamingInputCallRequest *request3 = [RMTStreamingInputCallRequest message]; |
| 222 | request3.payload.body = [NSMutableData dataWithLength:1828]; |
| 223 | |
| 224 | RMTStreamingInputCallRequest *request4 = [RMTStreamingInputCallRequest message]; |
| 225 | request4.payload.body = [NSMutableData dataWithLength:45904]; |
| 226 | |
Jorge Canizales | a8c5d96 | 2015-07-16 18:55:31 -0700 | [diff] [blame] | 227 | GRXWriter *writer = [GRXWriter writerWithContainer:@[request1, request2, request3, request4]]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 228 | |
| 229 | [_service streamingInputCallWithRequestsWriter:writer |
| 230 | handler:^(RMTStreamingInputCallResponse *response, |
| 231 | NSError *error) { |
| 232 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 233 | |
| 234 | RMTStreamingInputCallResponse *expectedResponse = [RMTStreamingInputCallResponse message]; |
| 235 | expectedResponse.aggregatedPayloadSize = 74922; |
| 236 | XCTAssertEqualObjects(response, expectedResponse); |
| 237 | |
| 238 | [expectation fulfill]; |
| 239 | }]; |
| 240 | |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 241 | [self waitForExpectationsWithTimeout:8 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 242 | } |
| 243 | |
| 244 | - (void)testServerStreamingRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 245 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 246 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"ServerStreaming"]; |
| 247 | |
| 248 | NSArray *expectedSizes = @[@31415, @9, @2653, @58979]; |
| 249 | |
| 250 | RMTStreamingOutputCallRequest *request = [RMTStreamingOutputCallRequest message]; |
| 251 | for (NSNumber *size in expectedSizes) { |
| 252 | RMTResponseParameters *parameters = [RMTResponseParameters message]; |
Yuchen Zeng | afb62f8 | 2016-06-24 19:11:08 -0700 | [diff] [blame] | 253 | parameters.size = [size intValue]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 254 | [request.responseParametersArray addObject:parameters]; |
| 255 | } |
| 256 | |
| 257 | __block int index = 0; |
| 258 | [_service streamingOutputCallWithRequest:request |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 259 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 260 | RMTStreamingOutputCallResponse *response, |
| 261 | NSError *error){ |
| 262 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 263 | XCTAssertTrue(done || response, @"Event handler called without an event."); |
| 264 | |
| 265 | if (response) { |
| 266 | XCTAssertLessThan(index, 4, @"More than 4 responses received."); |
| 267 | id expected = [RMTStreamingOutputCallResponse messageWithPayloadSize:expectedSizes[index]]; |
| 268 | XCTAssertEqualObjects(response, expected); |
| 269 | index += 1; |
| 270 | } |
| 271 | |
| 272 | if (done) { |
| 273 | XCTAssertEqual(index, 4, @"Received %i responses instead of 4.", index); |
| 274 | [expectation fulfill]; |
| 275 | } |
| 276 | }]; |
| 277 | |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 278 | [self waitForExpectationsWithTimeout:8 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 279 | } |
| 280 | |
| 281 | - (void)testPingPongRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 282 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 283 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"PingPong"]; |
| 284 | |
| 285 | NSArray *requests = @[@27182, @8, @1828, @45904]; |
| 286 | NSArray *responses = @[@31415, @9, @2653, @58979]; |
| 287 | |
| 288 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
| 289 | |
| 290 | __block int index = 0; |
| 291 | |
| 292 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:requests[index] |
| 293 | requestedResponseSize:responses[index]]; |
| 294 | [requestsBuffer writeValue:request]; |
| 295 | |
| 296 | [_service fullDuplexCallWithRequestsWriter:requestsBuffer |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 297 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 298 | RMTStreamingOutputCallResponse *response, |
| 299 | NSError *error) { |
| 300 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 301 | XCTAssertTrue(done || response, @"Event handler called without an event."); |
| 302 | |
| 303 | if (response) { |
| 304 | XCTAssertLessThan(index, 4, @"More than 4 responses received."); |
| 305 | id expected = [RMTStreamingOutputCallResponse messageWithPayloadSize:responses[index]]; |
| 306 | XCTAssertEqualObjects(response, expected); |
| 307 | index += 1; |
| 308 | if (index < 4) { |
| 309 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:requests[index] |
| 310 | requestedResponseSize:responses[index]]; |
| 311 | [requestsBuffer writeValue:request]; |
| 312 | } else { |
| 313 | [requestsBuffer writesFinishedWithError:nil]; |
| 314 | } |
| 315 | } |
| 316 | |
| 317 | if (done) { |
| 318 | XCTAssertEqual(index, 4, @"Received %i responses instead of 4.", index); |
| 319 | [expectation fulfill]; |
| 320 | } |
| 321 | }]; |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 322 | [self waitForExpectationsWithTimeout:4 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 323 | } |
| 324 | |
Makarand Dharmapurikar | 9b5f640 | 2016-05-20 15:57:34 -0700 | [diff] [blame] | 325 | #ifndef GRPC_COMPILE_WITH_CRONET |
| 326 | // TODO(makdharma@): Fix this test |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 327 | - (void)testEmptyStreamRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 328 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 329 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"EmptyStream"]; |
| 330 | [_service fullDuplexCallWithRequestsWriter:[GRXWriter emptyWriter] |
murgatroid99 | 25a2661 | 2015-06-25 11:22:23 -0700 | [diff] [blame] | 331 | eventHandler:^(BOOL done, |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 332 | RMTStreamingOutputCallResponse *response, |
| 333 | NSError *error) { |
| 334 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 335 | XCTAssert(done, @"Unexpected response: %@", response); |
| 336 | [expectation fulfill]; |
| 337 | }]; |
| 338 | [self waitForExpectationsWithTimeout:2 handler:nil]; |
| 339 | } |
Makarand Dharmapurikar | 9b5f640 | 2016-05-20 15:57:34 -0700 | [diff] [blame] | 340 | #endif |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 341 | |
| 342 | - (void)testCancelAfterBeginRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 343 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 344 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"CancelAfterBegin"]; |
| 345 | |
| 346 | // A buffered pipe to which we never write any value acts as a writer that just hangs. |
| 347 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
| 348 | |
Yuchen Zeng | 3d399cb | 2016-06-21 11:57:46 -0700 | [diff] [blame] | 349 | GRPCProtoCall *call = |
| 350 | [_service RPCToStreamingInputCallWithRequestsWriter:requestsBuffer |
| 351 | handler:^(RMTStreamingInputCallResponse *response, |
| 352 | NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 353 | XCTAssertEqual(error.code, GRPC_STATUS_CANCELLED); |
| 354 | [expectation fulfill]; |
| 355 | }]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 356 | XCTAssertEqual(call.state, GRXWriterStateNotStarted); |
| 357 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 358 | [call start]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 359 | XCTAssertEqual(call.state, GRXWriterStateStarted); |
| 360 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 361 | [call cancel]; |
Jorge Canizales | 47b80bc | 2016-04-30 10:32:49 -0700 | [diff] [blame] | 362 | XCTAssertEqual(call.state, GRXWriterStateFinished); |
| 363 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 364 | [self waitForExpectationsWithTimeout:1 handler:nil]; |
| 365 | } |
| 366 | |
| 367 | - (void)testCancelAfterFirstResponseRPC { |
Jorge Canizales | 91db5ac | 2015-10-28 14:53:38 -0700 | [diff] [blame] | 368 | XCTAssertNotNil(self.class.host); |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 369 | __weak XCTestExpectation *expectation = [self expectationWithDescription:@"CancelAfterFirstResponse"]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 370 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 371 | // A buffered pipe to which we write a single value but never close |
| 372 | GRXBufferedPipe *requestsBuffer = [[GRXBufferedPipe alloc] init]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 373 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 374 | __block BOOL receivedResponse = NO; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 375 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 376 | id request = [RMTStreamingOutputCallRequest messageWithPayloadSize:@21782 |
| 377 | requestedResponseSize:@31415]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 378 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 379 | [requestsBuffer writeValue:request]; |
Nate Kibler | 2d32771 | 2015-09-25 11:56:05 -0700 | [diff] [blame] | 380 | |
Yuchen Zeng | dbe2b9e | 2016-06-15 20:23:04 -0700 | [diff] [blame] | 381 | __block GRPCProtoCall *call = |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 382 | [_service RPCToFullDuplexCallWithRequestsWriter:requestsBuffer |
| 383 | eventHandler:^(BOOL done, |
| 384 | RMTStreamingOutputCallResponse *response, |
| 385 | NSError *error) { |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 386 | if (receivedResponse) { |
| 387 | XCTAssert(done, @"Unexpected extra response %@", response); |
| 388 | XCTAssertEqual(error.code, GRPC_STATUS_CANCELLED); |
| 389 | [expectation fulfill]; |
| 390 | } else { |
| 391 | XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
| 392 | XCTAssertFalse(done, @"Finished without response"); |
| 393 | XCTAssertNotNil(response); |
| 394 | receivedResponse = YES; |
| 395 | [call cancel]; |
| 396 | } |
| 397 | }]; |
| 398 | [call start]; |
Jorge Canizales | 5580e5d | 2015-07-15 23:35:48 -0700 | [diff] [blame] | 399 | [self waitForExpectationsWithTimeout:8 handler:nil]; |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 400 | } |
| 401 | |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 402 | - (void)testRPCAfterClosingOpenConnections { |
| 403 | XCTAssertNotNil(self.class.host); |
| 404 | __weak XCTestExpectation *expectation = |
| 405 | [self expectationWithDescription:@"RPC after closing connection"]; |
| 406 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 407 | GPBEmpty *request = [GPBEmpty message]; |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 408 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 409 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 410 | XCTAssertNil(error, @"First RPC finished with unexpected error: %@", error); |
| 411 | |
Jorge Canizales | 00ca8ea | 2016-06-24 16:02:56 -0700 | [diff] [blame] | 412 | #pragma clang diagnostic push |
| 413 | #pragma clang diagnostic ignored "-Wdeprecated-declarations" |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 414 | [GRPCCall closeOpenConnections]; |
Jorge Canizales | 00ca8ea | 2016-06-24 16:02:56 -0700 | [diff] [blame] | 415 | #pragma clang diagnostic pop |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 416 | |
Makarand Dharmapurikar | 2b220f8 | 2016-07-08 10:47:57 -0700 | [diff] [blame] | 417 | [_service emptyCallWithRequest:request handler:^(GPBEmpty *response, NSError *error) { |
Jorge Canizales | fa70dac | 2016-04-29 19:58:52 -0700 | [diff] [blame] | 418 | XCTAssertNil(error, @"Second RPC finished with unexpected error: %@", error); |
| 419 | [expectation fulfill]; |
| 420 | }]; |
| 421 | }]; |
| 422 | |
| 423 | [self waitForExpectationsWithTimeout:4 handler:nil]; |
| 424 | } |
| 425 | |
Jorge Canizales | 7307684 | 2015-05-30 12:04:43 -0700 | [diff] [blame] | 426 | @end |