platform/framework/test/encoders_buffer_test.dart
Tobe O 609d06f66b Add 'framework/' from commit '64d6729def5ffcf225a2c8f74fdf115218df1c56'
git-subtree-dir: framework
git-subtree-mainline: 6f6510ab27
git-subtree-split: 64d6729def
2020-02-15 18:12:48 -05:00

112 lines
3 KiB
Dart

import 'dart:async';
import 'dart:convert';
import 'dart:io' show BytesBuilder;
import 'dart:io';
import 'package:angel_container/mirrors.dart';
import 'package:angel_framework/angel_framework.dart';
import 'package:angel_framework/http.dart';
import 'package:mock_request/mock_request.dart';
import 'package:test/test.dart';
Future<List<int>> getBody(MockHttpResponse rs) async {
var list = await rs.toList();
var bb = BytesBuilder();
list.forEach(bb.add);
return bb.takeBytes();
}
main() {
Angel app;
setUp(() {
app = Angel(reflector: MirrorsReflector());
app.encoders.addAll(
{
'deflate': zlib.encoder,
'gzip': gzip.encoder,
},
);
app.get('/hello', (req, res) {
res
..useBuffer()
..write('Hello, world!');
});
});
tearDown(() => app.close());
encodingTests(() => app);
}
void encodingTests(Angel getApp()) {
group('encoding', () {
Angel app;
AngelHttp http;
setUp(() {
app = getApp();
http = AngelHttp(app);
});
test('sends plaintext if no accept-encoding', () async {
var rq = MockHttpRequest('GET', Uri.parse('/hello'));
await rq.close();
var rs = rq.response;
await http.handleRequest(rq);
var body = await rs.transform(utf8.decoder).join();
expect(body, 'Hello, world!');
});
test('encodes if wildcard', () async {
var rq = MockHttpRequest('GET', Uri.parse('/hello'))
..headers.set('accept-encoding', '*');
await rq.close();
var rs = rq.response;
await http.handleRequest(rq);
var body = await getBody(rs);
//print(rs.headers);
expect(rs.headers.value('content-encoding'), 'deflate');
expect(body, zlib.encode(utf8.encode('Hello, world!')));
});
test('encodes if wildcard + multiple', () async {
var rq = MockHttpRequest('GET', Uri.parse('/hello'))
..headers.set('accept-encoding', ['foo', 'bar', '*']);
await rq.close();
var rs = rq.response;
await http.handleRequest(rq);
var body = await getBody(rs);
expect(rs.headers.value('content-encoding'), 'deflate');
expect(body, zlib.encode(utf8.encode('Hello, world!')));
});
test('encodes if explicit', () async {
var rq = MockHttpRequest('GET', Uri.parse('/hello'))
..headers.set('accept-encoding', 'gzip');
await rq.close();
var rs = rq.response;
await http.handleRequest(rq);
var body = await getBody(rs);
expect(rs.headers.value('content-encoding'), 'gzip');
expect(body, gzip.encode(utf8.encode('Hello, world!')));
});
test('only uses one encoder', () async {
var rq = MockHttpRequest('GET', Uri.parse('/hello'))
..headers.set('accept-encoding', ['gzip', 'deflate']);
await rq.close();
var rs = rq.response;
await http.handleRequest(rq);
var body = await getBody(rs);
expect(rs.headers.value('content-encoding'), 'gzip');
expect(body, gzip.encode(utf8.encode('Hello, world!')));
});
});
}