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