[#2815] Fix unit test

diff --git a/src/lib/dhcpsrv/testutils/generic_host_data_source_unittest.cc b/src/lib/dhcpsrv/testutils/generic_host_data_source_unittest.cc
index 4d1c4f0ebc011a69e13c9f4b34ba9589953bbb80..ccad42630d4565c58a055e7ea23643244639870f 100644 (file)
--- a/src/lib/dhcpsrv/testutils/generic_host_data_source_unittest.cc
+++ b/src/lib/dhcpsrv/testutils/generic_host_data_source_unittest.cc
@@ -3074,7 +3074,7 @@ HostMgrTest::testGetAll(BaseHostDataSource& data_source1,
     EXPECT_EQ(reservations_in_primary_source, hosts.size());
     if (is_first_source_primary) {
         found = false;
-        for (unsigned i = 0; i < 2; ++i) {
+        for (unsigned i = 0; i < hosts.size(); ++i) {
             if (hosts[i]->getIPv4Reservation() == IOAddress("192.0.2.5")) {
                 ASSERT_EQ(1, hosts[i]->getIPv4SubnetID());
                 found = true;
@@ -3088,7 +3088,7 @@ HostMgrTest::testGetAll(BaseHostDataSource& data_source1,
     }
     if (is_second_source_primary) {
         found = false;
-        for (unsigned i = 0; i < 2; ++i) {
+        for (unsigned i = 0; i < hosts.size(); ++i) {
             if (hosts[i]->getIPv4Reservation() == IOAddress("192.0.3.10")) {
                 ASSERT_EQ(10, hosts[i]->getIPv4SubnetID());
                 found = true;