mirror of https://github.com/grpc/grpc-dart.git
85 lines
2.7 KiB
Dart
85 lines
2.7 KiB
Dart
@TestOn('vm')
|
|
import 'dart:async';
|
|
import 'dart:io';
|
|
import 'package:grpc/grpc.dart';
|
|
import 'package:grpc/service_api.dart' as api;
|
|
import 'package:grpc/src/client/channel.dart' hide ClientChannel;
|
|
import 'package:grpc/src/client/connection.dart';
|
|
import 'package:grpc/src/client/http2_connection.dart';
|
|
import 'package:http2/http2.dart';
|
|
import 'package:test/test.dart';
|
|
|
|
class TestClient extends Client {
|
|
static final _$stream = ClientMethod<int, int>('/test.TestService/stream',
|
|
(int value) => [value], (List<int> value) => value[0]);
|
|
|
|
TestClient(api.ClientChannel channel) : super(channel);
|
|
ResponseStream<int> stream(int request, {CallOptions options}) {
|
|
final call =
|
|
$createCall(_$stream, Stream.fromIterable([request]), options: options);
|
|
return ResponseStream(call);
|
|
}
|
|
}
|
|
|
|
class TestService extends Service {
|
|
String get $name => 'test.TestService';
|
|
|
|
TestService() {
|
|
$addMethod(ServiceMethod<int, int>('stream', stream, false, true,
|
|
(List<int> value) => value[0], (int value) => [value]));
|
|
}
|
|
|
|
Stream<int> stream(ServiceCall call, Future request) async* {
|
|
yield 1;
|
|
yield 2;
|
|
yield 3;
|
|
}
|
|
}
|
|
|
|
class FixedConnectionClientChannel extends ClientChannelBase {
|
|
final Http2ClientConnection clientConnection;
|
|
List<ConnectionState> states = <ConnectionState>[];
|
|
FixedConnectionClientChannel(this.clientConnection) {
|
|
clientConnection.onStateChanged = (c) => states.add(c.state);
|
|
}
|
|
@override
|
|
ClientConnection createConnection() => clientConnection;
|
|
}
|
|
|
|
main() async {
|
|
test('round trip insecure connection', () async {
|
|
final Server server = Server([TestService()]);
|
|
await server.serve(port: 0);
|
|
|
|
final channel = FixedConnectionClientChannel(Http2ClientConnection(
|
|
'localhost',
|
|
server.port,
|
|
ChannelOptions(credentials: ChannelCredentials.insecure()),
|
|
));
|
|
final testClient = TestClient(channel);
|
|
expect(await testClient.stream(1).toList(), [1, 2, 3]);
|
|
server.shutdown();
|
|
});
|
|
|
|
test('round trip secure connection', () async {
|
|
final Server server = Server([TestService()]);
|
|
await server.serve(
|
|
port: 0,
|
|
security: ServerTlsCredentials(
|
|
certificate: File('test/data/localhost.crt').readAsBytesSync(),
|
|
privateKey: File('test/data/localhost.key').readAsBytesSync()));
|
|
|
|
final channel = FixedConnectionClientChannel(Http2ClientConnection(
|
|
'localhost',
|
|
server.port,
|
|
ChannelOptions(
|
|
credentials: ChannelCredentials.secure(
|
|
certificates: File('test/data/localhost.crt').readAsBytesSync(),
|
|
authority: 'localhost')),
|
|
));
|
|
final testClient = TestClient(channel);
|
|
expect(await testClient.stream(1).toList(), [1, 2, 3]);
|
|
server.shutdown();
|
|
});
|
|
}
|