|
7 | 7 | using Microsoft.VisualStudio.TestTools.UnitTesting; |
8 | 8 | using RT.Util.ExtensionMethods; |
9 | 9 |
|
10 | | -namespace RT.Servers.Tests |
| 10 | +namespace RT.Servers.Tests; |
| 11 | + |
| 12 | +[TestClass] |
| 13 | +public sealed class DisconnectionTests |
11 | 14 | { |
12 | | - [TestClass] |
13 | | - public sealed class DisconnectionTests |
| 15 | + private static IEnumerable<string> enumInfinite() |
14 | 16 | { |
15 | | - private IEnumerable<string> enumInfinite() |
| 17 | + while (true) |
16 | 18 | { |
17 | | - while (true) |
18 | | - { |
19 | | - yield return "blah!"; |
20 | | - Thread.Sleep(0); |
21 | | - } |
| 19 | + yield return "blah!"; |
| 20 | + Thread.Sleep(0); |
22 | 21 | } |
| 22 | + } |
23 | 23 |
|
24 | | - [TestMethod, Timeout(60 * 1000, CooperativeCancellation = true)] |
25 | | - public void TestMidResponseSocketClosure() |
| 24 | + [TestMethod] |
| 25 | + public void TestMidResponseSocketClosure() |
| 26 | + { |
| 27 | + var instance = new HttpServer(TestHelpers.Port + 1) |
26 | 28 | { |
27 | | - var instance = new HttpServer(TestHelpers.Port + 1) |
28 | | - { |
29 | | - Handler = new UrlResolver( |
30 | | - new UrlMapping(req => { return HttpResponse.Create(enumInfinite(), "text/plain"); }, path: "/infinite-and-slow") |
31 | | - ).Handle |
32 | | - }; |
33 | | - try |
34 | | - { |
35 | | - instance.StartListening(); |
36 | | - |
37 | | - ThreadStart thread = () => |
38 | | - { |
39 | | - for (int i = 0; i < 20; i++) |
40 | | - { |
41 | | - TcpClient cl = new(); |
42 | | - cl.Connect("localhost", TestHelpers.Port + 1); |
43 | | - cl.ReceiveTimeout = 1000; // 1 sec |
44 | | - Socket sck = cl.Client; |
45 | | - sck.Send("GET /infinite-and-slow HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n".ToUtf8()); |
46 | | - Thread.Sleep(100); |
47 | | - sck.Close(); |
48 | | - GC.Collect(); |
49 | | - } |
50 | | - }; |
51 | | - var threads = Enumerable.Range(0, 10).Select(_ => new Thread(thread)).ToList(); |
52 | | - foreach (var t in threads) |
53 | | - t.Start(); |
54 | | - foreach (var t in threads) |
55 | | - t.Join(); |
| 29 | + Handler = new UrlResolver( |
| 30 | + new UrlMapping(req => { return HttpResponse.Create(enumInfinite(), "text/plain"); }, path: "/infinite-and-slow") |
| 31 | + ).Handle |
| 32 | + }; |
| 33 | + try |
| 34 | + { |
| 35 | + instance.StartListening(); |
56 | 36 |
|
57 | | - instance.StopListening(true); // server must not throw after this; that’s the point of the test |
58 | | - } |
59 | | - finally |
| 37 | + static void thread() |
60 | 38 | { |
61 | | - instance.StopListening(brutal: true); |
| 39 | + for (var i = 0; i < 20; i++) |
| 40 | + { |
| 41 | + var cl = new TcpClient(); |
| 42 | + cl.Connect("localhost", TestHelpers.Port + 1); |
| 43 | + cl.ReceiveTimeout = 1000; // 1 sec |
| 44 | + Socket sck = cl.Client; |
| 45 | + sck.Send("GET /infinite-and-slow HTTP/1.1\r\nHost: localhost\r\nConnection: keep-alive\r\n\r\n".ToUtf8()); |
| 46 | + Thread.Sleep(100); |
| 47 | + sck.Close(); |
| 48 | + GC.Collect(); |
| 49 | + } |
62 | 50 | } |
63 | | - } |
| 51 | + var threads = Enumerable.Range(0, 10).Select(_ => new Thread(thread)).ToList(); |
| 52 | + foreach (var t in threads) |
| 53 | + t.Start(); |
| 54 | + foreach (var t in threads) |
| 55 | + t.Join(); |
64 | 56 |
|
65 | | - [TestMethod, Timeout(60 * 1000, CooperativeCancellation = true)] |
66 | | - public void TestHalfOpenConnection() |
| 57 | + instance.StopListening(true); // server must not throw after this; that’s the point of the test |
| 58 | + } |
| 59 | + finally |
67 | 60 | { |
68 | | - var instance = new HttpServer(TestHelpers.Port + 2, new HttpServerOptions { OutputExceptionInformation = true }); |
69 | | - instance.Handler = req => HttpResponse.PlainText(" thingy stuff "); |
70 | | - try |
71 | | - { |
72 | | - instance.StartListening(); |
| 61 | + instance.StopListening(brutal: true); |
| 62 | + } |
| 63 | + } |
73 | 64 |
|
74 | | - // A proper request ending in a half closed connection |
75 | | - using (var cl = new TcpClient()) |
76 | | - { |
77 | | - cl.ReceiveTimeout = 1000; // 1 sec |
78 | | - cl.Connect("localhost", TestHelpers.Port + 2); |
79 | | - cl.Client.Send("GET /static HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n".ToUtf8()); |
80 | | - cl.Client.Shutdown(SocketShutdown.Send); |
81 | | - var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
82 | | - var code = (HttpStatusCode) int.Parse(response.Substring("HTTP/1.1 ".Length, 3)); |
83 | | - var parts = response.Split("\r\n\r\n"); |
84 | | - Assert.AreEqual(HttpStatusCode._200_OK, code); |
85 | | - Assert.AreEqual(" thingy stuff ", parts[1]); |
86 | | - } |
| 65 | + [TestMethod] |
| 66 | + public void TestHalfOpenConnection() |
| 67 | + { |
| 68 | + var instance = new HttpServer(TestHelpers.Port + 2, new HttpServerOptions { OutputExceptionInformation = true }) { Handler = req => HttpResponse.PlainText(" thingy stuff ") }; |
| 69 | + try |
| 70 | + { |
| 71 | + instance.StartListening(); |
87 | 72 |
|
88 | | - // An incomplete request ending in a half closed connection |
89 | | - using (var cl = new TcpClient()) |
90 | | - { |
91 | | - cl.Connect("localhost", TestHelpers.Port + 2); |
92 | | - cl.Client.Send("GET /static HTTP/1.1\r\nHost: localhost\r\nConnection: close".ToUtf8()); |
93 | | - cl.Client.Shutdown(SocketShutdown.Send); |
94 | | - var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
95 | | - // the test is that it doesn't wait forever |
96 | | - } |
| 73 | + // A proper request ending in a half closed connection |
| 74 | + using (var cl = new TcpClient()) |
| 75 | + { |
| 76 | + cl.ReceiveTimeout = 1000; // 1 sec |
| 77 | + cl.Connect("localhost", TestHelpers.Port + 2); |
| 78 | + cl.Client.Send("GET /static HTTP/1.1\r\nHost: localhost\r\nConnection: close\r\n\r\n".ToUtf8()); |
| 79 | + cl.Client.Shutdown(SocketShutdown.Send); |
| 80 | + var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
| 81 | + var code = (HttpStatusCode) int.Parse(response.Substring("HTTP/1.1 ".Length, 3)); |
| 82 | + var parts = response.Split("\r\n\r\n"); |
| 83 | + Assert.AreEqual(HttpStatusCode._200_OK, code); |
| 84 | + Assert.AreEqual(" thingy stuff ", parts[1]); |
| 85 | + } |
97 | 86 |
|
98 | | - // A malformed request ending in a half closed connection |
99 | | - using (var cl = new TcpClient()) |
100 | | - { |
101 | | - cl.Connect("localhost", TestHelpers.Port + 2); |
102 | | - cl.Client.Send("xz".ToUtf8()); |
103 | | - cl.Client.Shutdown(SocketShutdown.Send); |
104 | | - var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
105 | | - // the test is that it doesn't wait forever |
106 | | - } |
| 87 | + // An incomplete request ending in a half closed connection |
| 88 | + using (var cl = new TcpClient()) |
| 89 | + { |
| 90 | + cl.Connect("localhost", TestHelpers.Port + 2); |
| 91 | + cl.Client.Send("GET /static HTTP/1.1\r\nHost: localhost\r\nConnection: close".ToUtf8()); |
| 92 | + cl.Client.Shutdown(SocketShutdown.Send); |
| 93 | + var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
| 94 | + // the test is that it doesn't wait forever |
107 | 95 | } |
108 | | - finally |
| 96 | + |
| 97 | + // A malformed request ending in a half closed connection |
| 98 | + using (var cl = new TcpClient()) |
109 | 99 | { |
110 | | - instance.StopListening(brutal: true); |
| 100 | + cl.Connect("localhost", TestHelpers.Port + 2); |
| 101 | + cl.Client.Send("xz".ToUtf8()); |
| 102 | + cl.Client.Shutdown(SocketShutdown.Send); |
| 103 | + var response = Encoding.UTF8.GetString(cl.Client.ReceiveAllBytes()); |
| 104 | + // the test is that it doesn't wait forever |
111 | 105 | } |
112 | 106 | } |
| 107 | + finally |
| 108 | + { |
| 109 | + instance.StopListening(brutal: true); |
| 110 | + } |
113 | 111 | } |
114 | 112 | } |
0 commit comments