| OLD | NEW |
| 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2012, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 import 'dart:async'; | 5 import 'dart:async'; |
| 6 import 'dart:io'; | 6 import 'dart:io'; |
| 7 | 7 |
| 8 const SESSION_ID = "DARTSESSID"; | 8 const SESSION_ID = "DARTSESSID"; |
| 9 | 9 |
| 10 String getSessionId(List<Cookie> cookies) { | 10 String getSessionId(List<Cookie> cookies) { |
| 11 var id = cookies.reduce(null, (last, cookie) { | 11 var id = cookies.reduce(null, (last, cookie) { |
| 12 if (last != null) return last; | 12 if (last != null) return last; |
| 13 if (cookie.name.toUpperCase() == SESSION_ID) { | 13 if (cookie.name.toUpperCase() == SESSION_ID) { |
| 14 Expect.isTrue(cookie.httpOnly); | 14 Expect.isTrue(cookie.httpOnly); |
| 15 return cookie.value; | 15 return cookie.value; |
| 16 } | 16 } |
| 17 return null; | 17 return null; |
| 18 }); | 18 }); |
| 19 Expect.isNotNull(id); | 19 Expect.isNotNull(id); |
| 20 return id; | 20 return id; |
| 21 } | 21 } |
| 22 | 22 |
| 23 Future<String> connectGetSession(int port, [String session]) { | 23 Future<String> connectGetSession(int port, [String session]) { |
| 24 var c = new Completer(); | |
| 25 var client = new HttpClient(); | 24 var client = new HttpClient(); |
| 26 var conn = client.get("127.0.0.1", port, "/"); | 25 return client.get("127.0.0.1", port, "/") |
| 27 conn.onRequest = (request) { | 26 .then((request) { |
| 28 if (session != null) { | 27 if (session != null) { |
| 29 request.cookies.add(new Cookie(SESSION_ID, session)); | 28 request.cookies.add(new Cookie(SESSION_ID, session)); |
| 30 } | 29 } |
| 31 request.outputStream.close(); | 30 return request.close(); |
| 32 }; | 31 }) |
| 33 conn.onResponse = (response) { | 32 .then((response) { |
| 34 response.inputStream.onData = response.inputStream.read; | 33 return response.reduce(getSessionId(response.cookies), (v, _) => v); |
| 35 response.inputStream.onClosed = () { | 34 }); |
| 36 client.shutdown(); | |
| 37 c.complete(getSessionId(response.cookies)); | |
| 38 }; | |
| 39 }; | |
| 40 return c.future; | |
| 41 } | 35 } |
| 42 | 36 |
| 43 void testSessions(int sessionCount) { | 37 void testSessions(int sessionCount) { |
| 44 HttpServer server = new HttpServer(); | 38 HttpServer server = new HttpServer("127.0.0.1", 0); |
| 45 server.listen("127.0.0.1", 0); | |
| 46 var sessions = new Set(); | 39 var sessions = new Set(); |
| 47 server.defaultRequestHandler = (request, response) { | 40 server.listen((request) { |
| 48 sessions.add(request.session().id); | 41 sessions.add(request.session().id); |
| 49 response.outputStream.close(); | 42 request.response.close(); |
| 50 }; | 43 }); |
| 51 | 44 |
| 52 var futures = []; | 45 var futures = []; |
| 53 for (int i = 0; i < sessionCount; i++) { | 46 for (int i = 0; i < sessionCount; i++) { |
| 54 futures.add(connectGetSession(server.port).then((session) { | 47 futures.add(connectGetSession(server.port).then((session) { |
| 55 Expect.isNotNull(session); | 48 Expect.isNotNull(session); |
| 56 Expect.isTrue(sessions.contains(session)); | 49 Expect.isTrue(sessions.contains(session)); |
| 57 return connectGetSession(server.port, session).then((session2) { | 50 return connectGetSession(server.port, session).then((session2) { |
| 58 Expect.equals(session2, session); | 51 Expect.equals(session2, session); |
| 59 Expect.isTrue(sessions.contains(session2)); | 52 Expect.isTrue(sessions.contains(session2)); |
| 60 return session2; | 53 return session2; |
| 61 }); | 54 }); |
| 62 })); | 55 })); |
| 63 } | 56 } |
| 64 Futures.wait(futures).then((clientSessions) { | 57 Futures.wait(futures).then((clientSessions) { |
| 65 Expect.equals(sessions.length, sessionCount); | 58 Expect.equals(sessions.length, sessionCount); |
| 66 Expect.setEquals(new Set.from(clientSessions), sessions); | 59 Expect.setEquals(new Set.from(clientSessions), sessions); |
| 67 server.close(); | 60 server.close(); |
| 68 }); | 61 }); |
| 69 } | 62 } |
| 70 | 63 |
| 71 void testTimeout(int sessionCount) { | 64 void testTimeout(int sessionCount) { |
| 72 HttpServer server = new HttpServer(); | 65 HttpServer server = new HttpServer("127.0.0.1", 0); |
| 73 server.sessionTimeout = 0; | 66 server.sessionTimeout = 0; |
| 74 server.listen("127.0.0.1", 0); | |
| 75 var timeouts = []; | 67 var timeouts = []; |
| 76 server.defaultRequestHandler = (request, response) { | 68 server.listen((request) { |
| 77 var c = new Completer(); | 69 var c = new Completer(); |
| 78 timeouts.add(c.future); | 70 timeouts.add(c.future); |
| 79 request.session().onTimeout = () { | 71 request.session().onTimeout = () { |
| 80 c.complete(null); | 72 c.complete(null); |
| 81 }; | 73 }; |
| 82 response.outputStream.close(); | 74 request.response.close(); |
| 83 }; | 75 }); |
| 84 | 76 |
| 85 var futures = []; | 77 var futures = []; |
| 86 for (int i = 0; i < sessionCount; i++) { | 78 for (int i = 0; i < sessionCount; i++) { |
| 87 futures.add(connectGetSession(server.port)); | 79 futures.add(connectGetSession(server.port)); |
| 88 } | 80 } |
| 89 Futures.wait(futures).then((clientSessions) { | 81 Futures.wait(futures).then((clientSessions) { |
| 90 Futures.wait(timeouts).then((_) { | 82 Futures.wait(timeouts).then((_) { |
| 91 futures = []; | 83 futures = []; |
| 92 for (var id in clientSessions) { | 84 for (var id in clientSessions) { |
| 93 futures.add(connectGetSession(server.port, id).then((session) { | 85 futures.add(connectGetSession(server.port, id).then((session) { |
| 94 Expect.isNotNull(session); | 86 Expect.isNotNull(session); |
| 95 Expect.notEquals(id, session); | 87 Expect.notEquals(id, session); |
| 96 })); | 88 })); |
| 97 } | 89 } |
| 98 Futures.wait(futures).then((_) { | 90 Futures.wait(futures).then((_) { |
| 99 server.close(); | 91 server.close(); |
| 100 }); | 92 }); |
| 101 }); | 93 }); |
| 102 }); | 94 }); |
| 103 } | 95 } |
| 104 | 96 |
| 105 void main() { | 97 void main() { |
| 106 testSessions(5); | 98 testSessions(5); |
| 107 testTimeout(5); | 99 testTimeout(5); |
| 108 } | 100 } |
| OLD | NEW |