[#2823] Config backend sets allocators

1 files changed, 12 insertions, 0 deletions

diff --git a/src/lib/dhcpsrv/tests/flq_allocator_unittest.cc b/src/lib/dhcpsrv/tests/flq_allocator_unittest.cc
index 5e60cf81b2..33ae382901 100644
--- a/src/lib/dhcpsrv/tests/flq_allocator_unittest.cc
+++ b/src/lib/dhcpsrv/tests/flq_allocator_unittest.cc
@@ -41,6 +41,12 @@ public:
     }
 };
 
+// Test that the allocator returns the correct type.
+TEST_F(FreeLeaseQueueAllocatorTest4, getType) {
+    FreeLeaseQueueAllocator alloc(Lease::TYPE_V4, subnet_);
+    EXPECT_EQ("flq", alloc.getType());
+}
+
 // Test populating free DHCPv4 leases to the queue.
 TEST_F(FreeLeaseQueueAllocatorTest4, populateFreeAddressLeases) {
     FreeLeaseQueueAllocator alloc(Lease::TYPE_V4, subnet_);
@@ -353,6 +359,12 @@ public:
 
 };
 
+// Test that the allocator returns the correct type.
+TEST_F(FreeLeaseQueueAllocatorTest6, getType) {
+    FreeLeaseQueueAllocator alloc(Lease::TYPE_NA, subnet_);
+    EXPECT_EQ("flq", alloc.getType());
+}
+
 // Test populating free DHCPv6 address leases to the queue.
 TEST_F(FreeLeaseQueueAllocatorTest6, populateFreeAddressLeases) {
     FreeLeaseQueueAllocator alloc(Lease::TYPE_NA, subnet_);