test: add tests for AddrList

Signed-off-by: Leonard Ossa <leonard.ossa@openvpn.com>

Author: leoossa

test/unittests/CMakeLists.txt

@@ -33,6 +33,7 @@ add_executable(coreUnitTests
         test_acc.cpp
         test_alignment.cpp
         test_acc_certcheck.cpp
+        test_addrlist.cpp
         test_route_emulation.cpp
         test_log.cpp
         test_comp.cpp

test/unittests/test_addrlist.cpp

@@ -0,0 +1,62 @@
+//    OpenVPN -- An application to securely tunnel IP networks
+//               over a single port, with support for SSL/TLS-based
+//               session authentication and key exchange,
+//               packet encryption, packet authentication, and
+//               packet compression.
+//
+//    Copyright (C) 2024- OpenVPN Inc.
+//
+//    SPDX-License-Identifier: MPL-2.0 OR AGPL-3.0-only WITH openvpn3-openssl-exception
+//
+
+#include "test_common.hpp"
+#include "test_generators.hpp"
+
+#include <set>
+#include <vector>
+
+#include <openvpn/addr/addrlist.hpp>
+
+RC_GTEST_PROP(AddrList, AddMarksAddressAsExisting, ())
+{
+    openvpn::IP::AddrList address_list;
+    const auto address = *rc::genIPAddr();
+    address_list.add(address);
+    RC_ASSERT(address_list.exists(address));
+}
+
+RC_GTEST_PROP(AddrList, AddIsIdempotent, ())
+{
+    openvpn::IP::AddrList list;
+    const auto addr = *rc::genIPAddr();
+    list.add(addr);
+    list.add(addr);
+    RC_ASSERT(list.size() == 1U);
+    RC_ASSERT(list[0] == addr);
+}
+
+RC_GTEST_PROP(AddrList, AddDistinctAddressesKeepsUniqueCount, ())
+{
+    const auto addrs = *rc::gen::container<std::vector<openvpn::IP::Addr>>(rc::genIPAddr());
+    const std::set unique_addrs(addrs.begin(), addrs.end());
+    openvpn::IP::AddrList list;
+    for (const auto &addr : addrs)
+    {
+        list.add(addr);
+    }
+    RC_ASSERT(list.size() == unique_addrs.size());
+}
+
+RC_GTEST_PROP(AddrList, ExistsMatchesMembership, ())
+{
+    const auto addrs = *rc::gen::container<std::vector<openvpn::IP::Addr>>(rc::genIPAddr());
+    const std::set unique_addrs(addrs.begin(), addrs.end());
+    openvpn::IP::AddrList list;
+    for (const auto &addr : addrs)
+    {
+        list.add(addr);
+    }
+    const auto probe = *rc::gen::oneOf(rc::gen::elementOf(unique_addrs), rc::genIPAddr());
+    const bool expected = unique_addrs.contains(probe);
+    RC_ASSERT(list.exists(probe) == expected);
+}

test/unittests/test_generators.hpp

@@ -917,5 +917,13 @@ inline auto genDNSOptions() -> Gen<openvpn::DnsOptions>
         gen::set(&openvpn::DnsOptions::servers, generateDnsServerMap()));
 }
 
+inline auto genIPAddr() -> Gen<openvpn::IP::Addr>
+{
+    return gen::map(gen::oneOf(IPv4Address(), IPv6Address()),
+                    [](const std::string &ip)
+                    {
+                        return openvpn::IP::Addr::from_string(ip);
+                    });
+}
 } // namespace rc
 #endif // TEST_GENERATORS_HPP