78 lines
1.8 KiB
D
78 lines
1.8 KiB
D
/++ dub.sdl:
|
|
name "tests"
|
|
description "TCP disconnect task issue"
|
|
dependency "vibe-core" path="../"
|
|
+/
|
|
module tests;
|
|
|
|
import vibe.core.core;
|
|
import vibe.core.net;
|
|
import core.time : msecs;
|
|
import vibe.core.log;
|
|
|
|
ubyte[] buf;
|
|
|
|
void performTest(bool reverse)
|
|
{
|
|
auto l = listenTCP(11375, (conn) @safe nothrow {
|
|
bool read_ex = false;
|
|
bool write_ex = false;
|
|
auto rt = runTask!TCPConnection((conn) {
|
|
try {
|
|
conn.read(buf);
|
|
assert(false, "Expected read() to throw an exception.");
|
|
} catch (Exception) {
|
|
read_ex = true;
|
|
conn.close();
|
|
logInfo("read out");
|
|
} // expected
|
|
}, conn);
|
|
auto wt = runTask!TCPConnection((conn) {
|
|
sleep(reverse ? 100.msecs : 20.msecs); // give the connection time to establish
|
|
try {
|
|
// write enough to let the connection block long enough to let
|
|
// the remote end close the connection
|
|
// NOTE: on Windows, the first write() can actually complete
|
|
// immediately, but the second one blocks
|
|
foreach (i; 0 .. 2) conn.write(buf);
|
|
assert(false, "Expected write() to throw an exception.");
|
|
} catch (Exception) {
|
|
write_ex = true;
|
|
conn.close();
|
|
logInfo("write out");
|
|
} // expected
|
|
}, conn);
|
|
|
|
try {
|
|
rt.join();
|
|
wt.join();
|
|
} catch (Exception e)
|
|
assert(0, e.msg);
|
|
assert(read_ex, "No read exception thrown");
|
|
assert(write_ex, "No write exception thrown");
|
|
logInfo("Test has finished successfully.");
|
|
exitEventLoop();
|
|
}, "127.0.0.1");
|
|
|
|
runTask({
|
|
try {
|
|
auto conn = connectTCP("127.0.0.1", 11375);
|
|
sleep(reverse ? 20.msecs : 100.msecs);
|
|
conn.close();
|
|
} catch (Exception e) assert(false, e.msg);
|
|
});
|
|
|
|
runEventLoop();
|
|
|
|
l.stopListening();
|
|
}
|
|
|
|
void main()
|
|
{
|
|
setTimer(10000.msecs, { assert(false, "Test has hung."); });
|
|
buf = new ubyte[512*1024*1024];
|
|
|
|
performTest(false);
|
|
performTest(true);
|
|
}
|