From: James Coglan <james@neighbourhood.ie>
Date: Fri, 31 May 2024 15:50:17 +0000 (+0100)
Subject: resolved: tests for dns_zone_remove_rrs_by_key()
X-Git-Tag: v257-rc1~838^2~1
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=330d5c8a36d0a6001cc504e8653dc6f8c371c7a2;p=thirdparty%2Fsystemd.git

resolved: tests for dns_zone_remove_rrs_by_key()
---

diff --git a/src/resolve/test-dns-zone.c b/src/resolve/test-dns-zone.c
index d7990a7d39d..9984466db06 100644
--- a/src/resolve/test-dns-zone.c
+++ b/src/resolve/test-dns-zone.c
@@ -167,4 +167,52 @@ TEST(dns_zone_remove_rr_different_payload) {
         ASSERT_NOT_NULL(dns_zone_get(zone, rr_in));
 }
 
+/* ================================================================
+ * dns_zone_remove_rrs_by_key()
+ * ================================================================ */
+
+TEST(dns_zone_remove_rrs_by_key) {
+        Manager manager = {};
+        _cleanup_(dns_scope_freep) DnsScope *scope = NULL;
+        DnsZone *zone = NULL;
+        _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr1 = NULL, *rr2 = NULL, *rr3 = NULL;
+        DnsResourceKey *key = NULL;
+
+        dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
+        ASSERT_NOT_NULL(scope);
+        zone = &scope->zone;
+
+        rr1 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
+        ASSERT_NOT_NULL(rr1);
+        dns_zone_put(zone, scope, rr1, 0);
+
+        rr2 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_AAAA, "www.example.com");
+        ASSERT_NOT_NULL(rr2);
+        dns_zone_put(zone, scope, rr2, 0);
+
+        rr3 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
+        ASSERT_NOT_NULL(rr3);
+        rr3->cname.name = strdup("www.example.com");
+        dns_zone_put(zone, scope, rr3, 0);
+
+        key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "www.example.com");
+        ASSERT_NOT_NULL(key);
+        ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
+        ASSERT_NOT_NULL(dns_zone_get(zone, rr3));
+        dns_resource_key_unref(key);
+
+        key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
+        ASSERT_NOT_NULL(key);
+        ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
+        ASSERT_NULL(dns_zone_get(zone, rr3));
+        dns_resource_key_unref(key);
+
+        key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_ANY, "www.example.com");
+        ASSERT_NOT_NULL(key);
+        ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
+        ASSERT_NULL(dns_zone_get(zone, rr1));
+        ASSERT_NULL(dns_zone_get(zone, rr2));
+        dns_resource_key_unref(key);
+}
+
 DEFINE_TEST_MAIN(LOG_DEBUG);