import 'dart:async'; import 'dart:convert'; import 'dart:io'; import 'package:platform_container/mirrors.dart'; import 'package:platform_framework/platform_framework.dart'; import 'package:platform_framework/http.dart'; import 'package:logging/logging.dart'; import 'package:platform_mocking/platform_mocking.dart'; import 'package:test/test.dart'; import 'encoders_buffer_test.dart' show encodingTests; void main() { late Protevus app; late ProtevusHttp http; setUp(() { app = Protevus(reflector: MirrorsReflector()); http = ProtevusHttp(app, useZone: true); app.logger = Logger('streaming_test') ..onRecord.listen((rec) { print(rec); if (rec.stackTrace != null) print(rec.stackTrace); }); app.encoders.addAll( { 'deflate': zlib.encoder, 'gzip': gzip.encoder, }, ); app.get('/hello', (req, res) { return Stream>.fromIterable(['Hello, world!'.codeUnits]) .pipe(res); }); app.get('/write', (req, res) async { await res.addStream( Stream>.fromIterable(['Hello, world!'.codeUnits])); res.write('bye'); await res.close(); }); app.get('/multiple', (req, res) async { await res.addStream( Stream>.fromIterable(['Hello, world!'.codeUnits])); await res.addStream(Stream>.fromIterable(['bye'.codeUnits])); await res.close(); }); app.get('/overwrite', (req, res) async { res.statusCode = 32; await Stream>.fromIterable(['Hello, world!'.codeUnits]) .pipe(res); var f = Stream>.fromIterable(['Hello, world!'.codeUnits]) .pipe(res) .then((_) => false) .catchError((_) => true); expect(f, completion(true)); }); app.get('/error', (req, res) => res.addError(StateError('wtf'))); app.errorHandler = (e, req, res) async { stderr ..writeln(e.error) ..writeln(e.stackTrace); }; }); tearDown(() => http.close()); void expectHelloBye(String path) async { var rq = MockHttpRequest('GET', Uri.parse(path)); await (rq.close()); await http.handleRequest(rq); var body = await rq.response.transform(utf8.decoder).join(); expect(body, 'Hello, world!bye'); } test('write after addStream', () => expectHelloBye('/write')); test('multiple addStream', () => expectHelloBye('/multiple')); test('cannot write after close', () async { try { var rq = MockHttpRequest('GET', Uri.parse('/overwrite')); await rq.close(); await http.handleRequest(rq); var body = await rq.response.transform(utf8.decoder).join(); if (rq.response.statusCode != 32) { throw 'overwrite should throw error; response: $body'; } } on StateError { // Success } }); test('res => addError', () async { try { var rq = MockHttpRequest('GET', Uri.parse('/error')); await (rq.close()); await http.handleRequest(rq); var body = await rq.response.transform(utf8.decoder).join(); throw 'addError should throw error; response: $body'; } on StateError { // Should throw error... } }); encodingTests(() => app); }