diff options
Diffstat (limited to '')
-rw-r--r-- | src/seastar/tests/unit/connect_test.cc | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/src/seastar/tests/unit/connect_test.cc b/src/seastar/tests/unit/connect_test.cc new file mode 100644 index 000000000..e4b3b4728 --- /dev/null +++ b/src/seastar/tests/unit/connect_test.cc @@ -0,0 +1,74 @@ +#include <seastar/core/reactor.hh> +#include <seastar/testing/test_case.hh> +#include <seastar/testing/test_runner.hh> +#include <seastar/net/ip.hh> + +using namespace seastar; +using namespace net; + +SEASTAR_TEST_CASE(test_connection_attempt_is_shutdown) { + ipv4_addr server_addr("172.16.0.1"); + auto unconn = make_socket(); + auto f = unconn + .connect(make_ipv4_address(server_addr)) + .then_wrapped([] (auto&& f) { + try { + f.get(); + BOOST_REQUIRE(false); + } catch (...) {} + }); + unconn.shutdown(); + return f.finally([unconn = std::move(unconn)] {}); +} + +SEASTAR_TEST_CASE(test_unconnected_socket_shutsdown_established_connection) { + // Use a random port to reduce chance of conflict. + // TODO: retry a few times on failure. + std::default_random_engine& rnd = testing::local_random_engine; + auto distr = std::uniform_int_distribution<uint16_t>(12000, 65000); + auto sa = make_ipv4_address({"127.0.0.1", distr(rnd)}); + return do_with(engine().net().listen(sa, listen_options()), [sa] (auto& listener) { + auto f = listener.accept(); + auto unconn = make_socket(); + auto connf = unconn.connect(sa); + return connf.then([unconn = std::move(unconn)] (auto&& conn) mutable { + unconn.shutdown(); + return do_with(std::move(conn), [] (auto& conn) { + return do_with(conn.output(1), [] (auto& out) { + return out.write("ping").then_wrapped([] (auto&& f) { + try { + f.get(); + BOOST_REQUIRE(false); + } catch (...) {} + }); + }); + }); + }).finally([f = std::move(f)] () mutable { + return std::move(f); + }); + }); +} + +SEASTAR_TEST_CASE(test_accept_after_abort) { + std::default_random_engine& rnd = testing::local_random_engine; + auto distr = std::uniform_int_distribution<uint16_t>(12000, 65000); + auto sa = make_ipv4_address({"127.0.0.1", distr(rnd)}); + return do_with(seastar::server_socket(engine().net().listen(sa, listen_options())), [] (auto& listener) { + using ftype = future<accept_result>; + promise<ftype> p; + future<ftype> done = p.get_future(); + auto f = listener.accept().then_wrapped([&listener, p = std::move(p)] (auto f) mutable { + f.ignore_ready_future(); + p.set_value(listener.accept()); + }); + listener.abort_accept(); + return done.then([] (ftype f) { + return f.then_wrapped([] (ftype f) { + BOOST_REQUIRE(f.failed()); + if (f.available()) { + f.ignore_ready_future(); + } + }); + }); + }); +} |