-
Notifications
You must be signed in to change notification settings - Fork 420
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add generation of client interceptors (#1272)
- Loading branch information
Showing
7 changed files
with
209 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
173 changes: 173 additions & 0 deletions
173
Tests/GRPCTests/AsyncAwaitSupport/InterceptorsAsyncTests.swift
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,173 @@ | ||
| /* | ||
| * Copyright 2021, gRPC Authors All rights reserved. | ||
| * | ||
| * 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 | ||
| * | ||
| * http://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. | ||
| */ | ||
| #if compiler(>=5.5) | ||
| import EchoImplementation | ||
| import EchoModel | ||
| import GRPC | ||
| import HelloWorldModel | ||
| import NIOCore | ||
| import NIOHPACK | ||
| import NIOPosix | ||
| import SwiftProtobuf | ||
| import XCTest | ||
|
|
||
| @available(macOS 12, iOS 15, tvOS 15, watchOS 8, *) | ||
| class InterceptorsAsyncTests: GRPCTestCase { | ||
| private var group: EventLoopGroup! | ||
| private var server: Server! | ||
| private var connection: ClientConnection! | ||
| private var echo: Echo_EchoAsyncClient! | ||
|
|
||
| override func setUp() { | ||
| super.setUp() | ||
| let group = MultiThreadedEventLoopGroup(numberOfThreads: 1) | ||
| self.group = group | ||
|
|
||
| let server = try! Server.insecure(group: group) | ||
| .withServiceProviders([EchoProvider()]) | ||
| .withLogger(self.serverLogger) | ||
| .bind(host: "127.0.0.1", port: 0) | ||
| .wait() | ||
|
|
||
| self.server = server | ||
|
|
||
| let connection = ClientConnection.insecure(group: group) | ||
| .withBackgroundActivityLogger(self.clientLogger) | ||
| .connect(host: "127.0.0.1", port: server.channel.localAddress!.port!) | ||
|
|
||
| self.connection = connection | ||
|
|
||
| self.echo = Echo_EchoAsyncClient( | ||
| channel: connection, | ||
| defaultCallOptions: CallOptions(logger: self.clientLogger), | ||
| interceptors: ReversingInterceptors() | ||
| ) | ||
| } | ||
|
|
||
| override func tearDown() { | ||
| if let connection = self.connection { | ||
| XCTAssertNoThrow(try connection.close().wait()) | ||
| } | ||
| if let server = self.server { | ||
| XCTAssertNoThrow(try server.close().wait()) | ||
| } | ||
| if let group = self.group { | ||
| XCTAssertNoThrow(try group.syncShutdownGracefully()) | ||
| } | ||
|
|
||
| super.tearDown() | ||
| } | ||
|
|
||
| func testUnaryCall() { XCTAsyncTest { | ||
| let get = try await self.echo.get(.with { $0.text = "hello" }) | ||
| await assertThat(get, .is(.with { $0.text = "hello :teg ohce tfiwS" })) | ||
| } } | ||
|
|
||
| func testMakingUnaryCall() { XCTAsyncTest { | ||
| let call = self.echo.makeGetCall(.with { $0.text = "hello" }) | ||
| await assertThat(try await call.response, .is(.with { $0.text = "hello :teg ohce tfiwS" })) | ||
| } } | ||
|
|
||
| func testClientStreamingSequence() { XCTAsyncTest { | ||
| let requests = ["1 2", "3 4"].map { item in | ||
| Echo_EchoRequest.with { $0.text = item } | ||
| } | ||
| let response = try await self.echo.collect(requests, callOptions: .init()) | ||
|
|
||
| await assertThat(response, .is(.with { $0.text = "3 4 1 2 :tcelloc ohce tfiwS" })) | ||
| } } | ||
|
|
||
| func testClientStreamingAsyncSequence() { XCTAsyncTest { | ||
| let stream = AsyncStream<Echo_EchoRequest> { continuation in | ||
| continuation.yield(.with { $0.text = "1 2" }) | ||
| continuation.yield(.with { $0.text = "3 4" }) | ||
| continuation.finish() | ||
| } | ||
| let response = try await self.echo.collect(stream, callOptions: .init()) | ||
|
|
||
| await assertThat(response, .is(.with { $0.text = "3 4 1 2 :tcelloc ohce tfiwS" })) | ||
| } } | ||
|
|
||
| func testMakingCallClientStreaming() { XCTAsyncTest { | ||
| let call = self.echo.makeCollectCall(callOptions: .init()) | ||
| try await call.requestStream.send(.with { $0.text = "1 2" }) | ||
| try await call.requestStream.send(.with { $0.text = "3 4" }) | ||
| try await call.requestStream.finish() | ||
|
|
||
| await assertThat( | ||
| try await call.response, | ||
| .is(.with { $0.text = "3 4 1 2 :tcelloc ohce tfiwS" }) | ||
| ) | ||
| } } | ||
|
|
||
| func testServerStreaming() { XCTAsyncTest { | ||
| let responses = self.echo.expand(.with { $0.text = "hello" }, callOptions: .init()) | ||
| for try await response in responses { | ||
| // Expand splits on spaces, so we only expect one response. | ||
| await assertThat(response, .is(.with { $0.text = "hello :)0( dnapxe ohce tfiwS" })) | ||
| } | ||
| } } | ||
|
|
||
| func testMakingCallServerStreaming() { XCTAsyncTest { | ||
| let call = self.echo.makeExpandCall(.with { $0.text = "hello" }, callOptions: .init()) | ||
| for try await response in call.responses { | ||
| // Expand splits on spaces, so we only expect one response. | ||
| await assertThat(response, .is(.with { $0.text = "hello :)0( dnapxe ohce tfiwS" })) | ||
| } | ||
| } } | ||
|
|
||
| func testBidirectionalStreaming() { XCTAsyncTest { | ||
| let requests = ["1 2", "3 4"].map { item in | ||
| Echo_EchoRequest.with { $0.text = item } | ||
| } | ||
| let responses = self.echo.update(requests, callOptions: .init()) | ||
|
|
||
| var count = 0 | ||
| for try await response in responses { | ||
| switch count { | ||
| case 0: | ||
| await assertThat(response, .is(.with { $0.text = "1 2 :)0( etadpu ohce tfiwS" })) | ||
| case 1: | ||
| await assertThat(response, .is(.with { $0.text = "3 4 :)1( etadpu ohce tfiwS" })) | ||
| default: | ||
| XCTFail("Got more than 2 responses") | ||
| } | ||
| count += 1 | ||
| } | ||
| } } | ||
|
|
||
| func testMakingCallBidirectionalStreaming() { XCTAsyncTest { | ||
| let call = self.echo.makeUpdateCall(callOptions: .init()) | ||
| try await call.requestStream.send(.with { $0.text = "1 2" }) | ||
| try await call.requestStream.send(.with { $0.text = "3 4" }) | ||
| try await call.requestStream.finish() | ||
|
|
||
| var count = 0 | ||
| for try await response in call.responses { | ||
| switch count { | ||
| case 0: | ||
| await assertThat(response, .is(.with { $0.text = "1 2 :)0( etadpu ohce tfiwS" })) | ||
| case 1: | ||
| await assertThat(response, .is(.with { $0.text = "3 4 :)1( etadpu ohce tfiwS" })) | ||
| default: | ||
| XCTFail("Got more than 2 responses") | ||
| } | ||
| count += 1 | ||
| } | ||
| } } | ||
| } | ||
|
|
||
| #endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters