diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 21:11:59 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 21:11:59 +0000 |
commit | 3cd01b932e1c85394272ae64fae67ebeda92fb00 (patch) | |
tree | c5a3115d710afc1879ddea5349362a2bc651733c /test-dnsdistasync.cc | |
parent | Initial commit. (diff) | |
download | dnsdist-3cd01b932e1c85394272ae64fae67ebeda92fb00.tar.xz dnsdist-3cd01b932e1c85394272ae64fae67ebeda92fb00.zip |
Adding upstream version 1.8.3.upstream/1.8.3
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | test-dnsdistasync.cc | 167 |
1 files changed, 167 insertions, 0 deletions
diff --git a/test-dnsdistasync.cc b/test-dnsdistasync.cc new file mode 100644 index 0000000..7e8e137 --- /dev/null +++ b/test-dnsdistasync.cc @@ -0,0 +1,167 @@ +/* + * This file is part of PowerDNS or dnsdist. + * Copyright -- PowerDNS.COM B.V. and its contributors + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of version 2 of the GNU General Public License as + * published by the Free Software Foundation. + * + * In addition, for the avoidance of any doubt, permission is granted to + * link this program with OpenSSL and to (re)distribute the binaries + * produced as the result of such linking. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ +#define BOOST_TEST_DYN_LINK +#define BOOST_TEST_NO_MAIN + +#include <boost/test/unit_test.hpp> + +#include "dnsdist-async.hh" + +BOOST_AUTO_TEST_SUITE(test_dnsdistasync) + +class DummyQuerySender : public TCPQuerySender +{ +public: + bool active() const override + { + return true; + } + + void handleResponse(const struct timeval&, TCPResponse&&) override + { + } + + void handleXFRResponse(const struct timeval&, TCPResponse&&) override + { + } + + void notifyIOError(InternalQueryState&&, const struct timeval&) override + { + errorRaised = true; + } + + std::atomic<bool> errorRaised{false}; +}; + +struct DummyCrossProtocolQuery : public CrossProtocolQuery +{ + DummyCrossProtocolQuery() : + CrossProtocolQuery() + { + d_sender = std::make_shared<DummyQuerySender>(); + } + + std::shared_ptr<TCPQuerySender> getTCPQuerySender() override + { + return d_sender; + } + + std::shared_ptr<DummyQuerySender> d_sender; +}; + +BOOST_AUTO_TEST_CASE(test_Basic) +{ + auto holder = std::make_unique<dnsdist::AsynchronousHolder>(); + BOOST_CHECK(holder->empty()); + + { + auto query = holder->get(0, 0); + BOOST_CHECK(query == nullptr); + } + + { + uint16_t asyncID = 1; + uint16_t queryID = 42; + struct timeval ttd; + gettimeofday(&ttd, nullptr); + // timeout in 100 ms + const timeval add{0, 100000}; + ttd = ttd + add; + + holder->push(asyncID, queryID, ttd, std::make_unique<DummyCrossProtocolQuery>()); + BOOST_CHECK(!holder->empty()); + + auto query = holder->get(0, 0); + BOOST_CHECK(query == nullptr); + + query = holder->get(asyncID, queryID); + BOOST_CHECK(holder->empty()); + + query = holder->get(asyncID, queryID); + BOOST_CHECK(query == nullptr); + + // sleep for 200 ms, to be sure the main thread has + // been awakened + usleep(200000); + } + + holder->stop(); +} + +BOOST_AUTO_TEST_CASE(test_TimeoutFailClose) +{ + auto holder = std::make_unique<dnsdist::AsynchronousHolder>(false); + uint16_t asyncID = 1; + uint16_t queryID = 42; + struct timeval ttd; + gettimeofday(&ttd, nullptr); + // timeout in 10 ms + const timeval add{0, 10000}; + ttd = ttd + add; + + std::shared_ptr<DummyQuerySender> sender{nullptr}; + { + auto query = std::make_unique<DummyCrossProtocolQuery>(); + sender = query->d_sender; + BOOST_REQUIRE(sender != nullptr); + holder->push(asyncID, queryID, ttd, std::move(query)); + BOOST_CHECK(!holder->empty()); + } + + // sleep for 20 ms, to be sure + usleep(20000); + + BOOST_CHECK(holder->empty()); + BOOST_CHECK(sender->errorRaised.load()); + + holder->stop(); +} + +BOOST_AUTO_TEST_CASE(test_AddingExpiredEvent) +{ + auto holder = std::make_unique<dnsdist::AsynchronousHolder>(false); + uint16_t asyncID = 1; + uint16_t queryID = 42; + struct timeval ttd; + gettimeofday(&ttd, nullptr); + // timeout was 10 ms ago, for some reason (long processing time, CPU starvation...) + const timeval sub{0, 10000}; + ttd = ttd - sub; + + std::shared_ptr<DummyQuerySender> sender{nullptr}; + { + auto query = std::make_unique<DummyCrossProtocolQuery>(); + sender = query->d_sender; + BOOST_REQUIRE(sender != nullptr); + holder->push(asyncID, queryID, ttd, std::move(query)); + } + + // sleep for 20 ms + usleep(20000); + + BOOST_CHECK(holder->empty()); + BOOST_CHECK(sender->errorRaised.load()); + + holder->stop(); +} + +BOOST_AUTO_TEST_SUITE_END(); |