1 files changed, 63 insertions, 0 deletions
diff --git a/src/lib/dhcp/tests/packet_queue_testutils.h b/src/lib/dhcp/tests/packet_queue_testutils.h
new file mode 100644
index 0000000..211b8cb
--- /dev/null
+++ b/src/lib/dhcp/tests/packet_queue_testutils.h
@@ -0,0 +1,63 @@
+// Copyright (C) 2018 Internet Systems Consortium, Inc. ("ISC")
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+#ifndef PACKET_QUEUE_TESTUTILS_H
+#define PACKET_QUEUE_TESTUTILS_H
+
+#include <config.h>
+
+#include <dhcp/packet_queue.h>
+
+#include <boost/shared_ptr.hpp>
+#include <gtest/gtest.h>
+
+namespace isc {
+namespace dhcp {
+namespace test {
+
+template<typename PacketQueuePtrType> void checkInfo(PacketQueuePtrType queue, const std::string& exp_json) {
+    ASSERT_TRUE(queue) << "packet queue ptr is null";
+    // Fetch the queue info and verify it has all the expected values.
+    data::ElementPtr info;
+    ASSERT_NO_THROW(info  = queue->getInfo());
+    ASSERT_TRUE(info);
+    data::ElementPtr exp_elems;
+    ASSERT_NO_THROW(exp_elems = data::Element::fromJSON(exp_json)) <<
+        " exp_elems is invalid JSON : " << exp_json << " test is broken";
+    EXPECT_TRUE(exp_elems->equals(*info));
+}
+
+#define CHECK_QUEUE_INFO(queue, stream) \
+    { \
+        std::ostringstream oss__; \
+        oss__ << stream; \
+        checkInfo(queue, oss__.str().c_str());\
+    }
+
+
+template<typename PacketQueuePtrType> void checkIntStat(PacketQueuePtrType queue, 
+                                                     const std::string& name, size_t exp_value) {
+    ASSERT_TRUE(queue) << "packet queue ptr is null";
+    data::ElementPtr info;
+    ASSERT_NO_THROW(info  = queue->getInfo());
+    ASSERT_TRUE(info);
+
+    data::ConstElementPtr elem;
+    ASSERT_NO_THROW(elem = info->get(name)) << "stat: " << name << " not in info" << std::endl;
+    ASSERT_TRUE(elem);
+
+    int64_t value;
+    ASSERT_NO_THROW(value = elem->intValue());
+    EXPECT_EQ(exp_value, value) << "stat: " << name << " is wrong" << std::endl;;
+}
+
+extern data::ElementPtr makeQueueConfig(const std::string& queue_type, size_t capacity, bool enable_queue=true);
+
+}; // end of namespace isc::dhcp::test
+}; // end of namespace isc::dhcp
+}; // end of namespace isc
+
+#endif // PACKET_QUEUE_TESTUTILS_H