diff options
Diffstat (limited to 'src/bin/d2/tests/d2_queue_mgr_unittests.cc')
-rw-r--r-- | src/bin/d2/tests/d2_queue_mgr_unittests.cc | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/src/bin/d2/tests/d2_queue_mgr_unittests.cc b/src/bin/d2/tests/d2_queue_mgr_unittests.cc index a00de29f4b..d2d41f5e56 100644 --- a/src/bin/d2/tests/d2_queue_mgr_unittests.cc +++ b/src/bin/d2/tests/d2_queue_mgr_unittests.cc @@ -9,6 +9,7 @@ #include <asiolink/io_service.h> #include <asiolink/interval_timer.h> #include <d2/d2_queue_mgr.h> +#include <d2srv/testutils/stats_test_utils.h> #include <dhcp_ddns/ncr_udp.h> #include <util/time_utilities.h> @@ -21,6 +22,7 @@ using namespace std; using namespace isc; using namespace isc::dhcp_ddns; using namespace isc::d2; +using namespace isc::d2::test; namespace { @@ -203,7 +205,7 @@ bool checkSendVsReceived(NameChangeRequestPtr sent_ncr, /// @brief Text fixture that allows testing a listener and sender together /// It derives from both the receive and send handler classes and contains /// and instance of UDP listener and UDP sender. -class QueueMgrUDPTest : public virtual ::testing::Test, +class QueueMgrUDPTest : public virtual ::testing::Test, public D2StatTest, NameChangeSender::RequestSendHandler { public: asiolink::IOServicePtr io_service_; @@ -384,6 +386,13 @@ TEST_F (QueueMgrUDPTest, liveFeed) { EXPECT_EQ(0, queue_mgr_->getQueueSize()); } + StatMap stats_ncr = { + { "ncr-received", 3}, + { "ncr-invalid", 0}, + { "ncr-error", 0} + }; + checkStats(stats_ncr); + // Iterate over the list of requests, sending and receiving // each one. Allow them to accumulate in the queue. for (int i = 0; i < VALID_MSG_CNT; i++) { @@ -397,6 +406,13 @@ TEST_F (QueueMgrUDPTest, liveFeed) { EXPECT_EQ(i+1, queue_mgr_->getQueueSize()); } + StatMap stats_ncr_new = { + { "ncr-received", 6}, + { "ncr-invalid", 0}, + { "ncr-error", 0} + }; + checkStats(stats_ncr_new); + // Verify that the queue is at max capacity. EXPECT_EQ(queue_mgr_->getMaxQueueSize(), queue_mgr_->getQueueSize()); @@ -428,7 +444,6 @@ TEST_F (QueueMgrUDPTest, liveFeed) { EXPECT_NO_THROW(queue_mgr_->clearQueue()); EXPECT_EQ(0, queue_mgr_->getQueueSize()); - // Verify that we can again receive requests. // Send should be fine. ASSERT_NO_THROW(sender_->sendRequest(send_ncr)); |