52 lines
1.5 KiB
Dart
52 lines
1.5 KiB
Dart
|
import 'package:angel_framework/angel_framework.dart';
|
||
|
|
import 'package:angel_security/angel_security.dart';
|
||
|
|
import 'package:angel_test/angel_test.dart';
|
||
|
|
import 'package:test/test.dart';
|
||
|
|
|
||
|
|
main() {
|
||
|
|
Angel app;
|
||
|
|
TestClient client;
|
||
|
|
|
||
|
|
setUp(() async {
|
||
|
|
app = new Angel()
|
||
|
|
..chain(banIp('*.*.*.*')).get('/ban', 'WTF')
|
||
|
|
..chain(banOrigin('*')).get('/ban-origin', 'WTF')
|
||
|
|
..chain(banOrigin('*.foo.bar')).get('/allow-origin', 'YAY');
|
||
|
|
|
||
|
|
client = await connectTo(app);
|
||
|
|
});
|
||
|
|
|
||
|
|
tearDown(() => client.close());
|
||
|
|
|
||
|
|
test('ban everyone', () async {
|
||
|
|
var response = await client.get('/ban');
|
||
|
|
print(response.body);
|
||
|
|
expect(response, hasStatus(403));
|
||
|
|
expect(response.body.contains('WTF'), isFalse);
|
||
|
|
});
|
||
|
|
|
||
|
|
group('origin', () {
|
||
|
|
test('ban everyone', () async {
|
||
|
|
var response = await client
|
||
|
|
.get('/ban-origin', headers: {'Origin': 'www.example.com'});
|
||
|
|
print(response.body);
|
||
|
|
expect(response, hasStatus(403));
|
||
|
|
expect(response.body.contains('WTF'), isFalse);
|
||
|
|
});
|
||
|
|
|
||
|
|
test('ban specific', () async {
|
||
|
|
var response =
|
||
|
|
await client.get('/allow-origin', headers: {'Origin': 'www.foo.bar'});
|
||
|
|
print(response.body);
|
||
|
|
expect(response, hasStatus(403));
|
||
|
|
expect(response.body.contains('YAY'), isFalse);
|
||
|
|
|
||
|
|
response = await client
|
||
|
|
.get('/allow-origin', headers: {'Origin': 'www.example.com'});
|
||
|
|
print(response.body);
|
||
|
|
expect(response, hasStatus(200));
|
||
|
|
expect(response.body, contains('YAY'));
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|