mirror of https://github.com/grpc/grpc.git
commit
8de0c88b02
13 changed files with 669 additions and 35 deletions
@ -0,0 +1,46 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#import <Foundation/Foundation.h> |
||||||
|
#import <gRPC/GRPCCall.h> |
||||||
|
|
||||||
|
@interface ProtoRPC : GRPCCall |
||||||
|
|
||||||
|
- (instancetype)initWithHost:(NSString *)host |
||||||
|
method:(GRPCMethodName *)method |
||||||
|
requestsWriter:(id<GRXWriter>)requestsWriter |
||||||
|
responseClass:(Class)responseClass |
||||||
|
responsesWriteable:(id<GRXWriteable>)responsesWriteable NS_DESIGNATED_INITIALIZER; |
||||||
|
|
||||||
|
- (void)start; |
||||||
|
@end |
@ -0,0 +1,91 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#import "ProtoRPC.h" |
||||||
|
|
||||||
|
#import <gRPC/GRXWriteable.h> |
||||||
|
#import <gRPC/GRXWriter.h> |
||||||
|
#import <gRPC/GRXWriter+Transformations.h> |
||||||
|
#import <ProtocolBuffers/ProtocolBuffers.h> |
||||||
|
|
||||||
|
@implementation ProtoRPC { |
||||||
|
id<GRXWriteable> _responseWriteable; |
||||||
|
} |
||||||
|
|
||||||
|
- (instancetype)initWithHost:(NSString *)host |
||||||
|
method:(GRPCMethodName *)method |
||||||
|
requestsWriter:(id<GRXWriter>)requestsWriter { |
||||||
|
return [self initWithHost:host |
||||||
|
method:method |
||||||
|
requestsWriter:requestsWriter |
||||||
|
responseClass:nil |
||||||
|
responsesWriteable:nil]; |
||||||
|
} |
||||||
|
|
||||||
|
// Designated initializer |
||||||
|
- (instancetype)initWithHost:(NSString *)host |
||||||
|
method:(GRPCMethodName *)method |
||||||
|
requestsWriter:(id<GRXWriter>)requestsWriter |
||||||
|
responseClass:(Class)responseClass |
||||||
|
responsesWriteable:(id<GRXWriteable>)responsesWriteable { |
||||||
|
// Because we can't tell the type system to constrain the class, we need to check at runtime: |
||||||
|
if (![responseClass respondsToSelector:@selector(parseFromData:)]) { |
||||||
|
[NSException raise:NSInvalidArgumentException |
||||||
|
format:@"A protobuf class to parse the responses must be provided."]; |
||||||
|
} |
||||||
|
// A writer that serializes the proto messages to send. |
||||||
|
id<GRXWriter> bytesWriter = |
||||||
|
[[[GRXWriter alloc] initWithWriter:requestsWriter] map:^id(PBGeneratedMessage *proto) { |
||||||
|
return [proto data]; |
||||||
|
}]; |
||||||
|
if ((self = [super initWithHost:host method:method requestsWriter:bytesWriter])) { |
||||||
|
// A writeable that parses the proto messages received. |
||||||
|
_responseWriteable = [[GRXWriteable alloc] initWithValueHandler:^(NSData *value) { |
||||||
|
[responsesWriteable didReceiveValue:[responseClass parseFromData:value]]; |
||||||
|
} completionHandler:^(NSError *errorOrNil) { |
||||||
|
[responsesWriteable didFinishWithError:errorOrNil]; |
||||||
|
}]; |
||||||
|
} |
||||||
|
return self; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)start { |
||||||
|
[self startWithWriteable:_responseWriteable]; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)startWithWriteable:(id<GRXWriteable>)writeable { |
||||||
|
[super startWithWriteable:writeable]; |
||||||
|
// Break retain cycles. |
||||||
|
_responseWriteable = nil; |
||||||
|
} |
||||||
|
@end |
@ -0,0 +1,49 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#import <Foundation/Foundation.h> |
||||||
|
|
||||||
|
@class ProtoRPC; |
||||||
|
@protocol GRXWriteable; |
||||||
|
@protocol GRXWriter; |
||||||
|
|
||||||
|
@interface ProtoService : NSObject |
||||||
|
- (instancetype)initWithHost:(NSString *)host |
||||||
|
packageName:(NSString *)packageName |
||||||
|
serviceName:(NSString *)serviceName NS_DESIGNATED_INITIALIZER; |
||||||
|
|
||||||
|
- (ProtoRPC *)RPCToMethod:(NSString *)method |
||||||
|
requestsWriter:(id<GRXWriter>)requestsWriter |
||||||
|
responseClass:(Class)responseClass |
||||||
|
responsesWriteable:(id<GRXWriteable>)responsesWriteable; |
||||||
|
@end |
@ -0,0 +1,81 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#import "ProtoService.h" |
||||||
|
|
||||||
|
#import <gRPC/GRPCMethodName.h> |
||||||
|
#import <gRPC/GRXWriteable.h> |
||||||
|
#import <gRPC/GRXWriter.h> |
||||||
|
|
||||||
|
#import "ProtoRPC.h" |
||||||
|
|
||||||
|
@implementation ProtoService { |
||||||
|
NSString *_host; |
||||||
|
NSString *_packageName; |
||||||
|
NSString *_serviceName; |
||||||
|
} |
||||||
|
|
||||||
|
- (instancetype)init { |
||||||
|
return [self initWithHost:nil packageName:nil serviceName:nil]; |
||||||
|
} |
||||||
|
|
||||||
|
// Designated initializer |
||||||
|
- (instancetype)initWithHost:(NSString *)host |
||||||
|
packageName:(NSString *)packageName |
||||||
|
serviceName:(NSString *)serviceName { |
||||||
|
if (!host || !serviceName) { |
||||||
|
[NSException raise:NSInvalidArgumentException |
||||||
|
format:@"Neither host nor serviceName can be nil."]; |
||||||
|
} |
||||||
|
if ((self = [super init])) { |
||||||
|
_host = [host copy]; |
||||||
|
_packageName = [packageName copy]; |
||||||
|
_serviceName = [serviceName copy]; |
||||||
|
} |
||||||
|
return self; |
||||||
|
} |
||||||
|
|
||||||
|
- (ProtoRPC *)RPCToMethod:(NSString *)method |
||||||
|
requestsWriter:(id<GRXWriter>)requestsWriter |
||||||
|
responseClass:(Class)responseClass |
||||||
|
responsesWriteable:(id<GRXWriteable>)responsesWriteable { |
||||||
|
GRPCMethodName *methodName = [[GRPCMethodName alloc] initWithPackage:_packageName |
||||||
|
interface:_serviceName |
||||||
|
method:method]; |
||||||
|
return [[ProtoRPC alloc] initWithHost:_host |
||||||
|
method:methodName |
||||||
|
requestsWriter:requestsWriter |
||||||
|
responseClass:responseClass |
||||||
|
responsesWriteable:responsesWriteable]; |
||||||
|
} |
||||||
|
@end |
@ -0,0 +1,90 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#import <UIKit/UIKit.h> |
||||||
|
#import <XCTest/XCTest.h> |
||||||
|
|
||||||
|
#import <RemoteTest/Messages.pb.h> |
||||||
|
#import <RemoteTest/Test.pb.h> |
||||||
|
|
||||||
|
@interface RemoteProtoTests : XCTestCase |
||||||
|
@end |
||||||
|
|
||||||
|
@implementation RemoteProtoTests { |
||||||
|
RMTTestService *_service; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)setUp { |
||||||
|
_service = [[RMTTestService alloc] initWithHost:@"grpc-test.sandbox.google.com"]; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)testEmptyRPC { |
||||||
|
__weak XCTestExpectation *noRPCError = [self expectationWithDescription:@"RPC succeeded."]; |
||||||
|
__weak XCTestExpectation *responded = [self expectationWithDescription:@"Response received."]; |
||||||
|
|
||||||
|
[_service emptyCallWithRequest:[RMTEmpty defaultInstance] |
||||||
|
handler:^(RMTEmpty *response, NSError *error) { |
||||||
|
XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
||||||
|
[noRPCError fulfill]; |
||||||
|
XCTAssertNotNil(response, @"nil response received."); |
||||||
|
[responded fulfill]; |
||||||
|
}]; |
||||||
|
|
||||||
|
[self waitForExpectationsWithTimeout:2. handler:nil]; |
||||||
|
} |
||||||
|
|
||||||
|
- (void)testSimpleProtoRPC { |
||||||
|
__weak XCTestExpectation *noRPCError = [self expectationWithDescription:@"RPC succeeded."]; |
||||||
|
__weak XCTestExpectation *responded = [self expectationWithDescription:@"Response received."]; |
||||||
|
__weak XCTestExpectation *validResponse = [self expectationWithDescription:@"Valid response."]; |
||||||
|
|
||||||
|
RMTSimpleRequest *request = [[[[[[RMTSimpleRequestBuilder alloc] init] |
||||||
|
setResponseSize:100] |
||||||
|
setFillUsername:YES] |
||||||
|
setFillOauthScope:YES] |
||||||
|
build]; |
||||||
|
[_service unaryCallWithRequest:request handler:^(RMTSimpleResponse *response, NSError *error) { |
||||||
|
XCTAssertNil(error, @"Finished with unexpected error: %@", error); |
||||||
|
[noRPCError fulfill]; |
||||||
|
XCTAssertNotNil(response, @"nil response received."); |
||||||
|
[responded fulfill]; |
||||||
|
// We expect empty strings, not nil: |
||||||
|
XCTAssertNotNil(response.username, @"Response's username is nil."); |
||||||
|
XCTAssertNotNil(response.oauthScope, @"Response's OAuth scope is nil."); |
||||||
|
[validResponse fulfill]; |
||||||
|
}]; |
||||||
|
|
||||||
|
[self waitForExpectationsWithTimeout:2. handler:nil]; |
||||||
|
} |
||||||
|
|
||||||
|
@end |
Loading…
Reference in new issue