From bbeb49f4c9512b3e5869c098d9af5e60cae52104 Mon Sep 17 00:00:00 2001 From: James Coglan Date: Fri, 31 May 2024 11:08:14 +0100 Subject: [PATCH] resolved: tests for dns_question_first_name() --- src/resolve/test-dns-question.c | 37 +++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) diff --git a/src/resolve/test-dns-question.c b/src/resolve/test-dns-question.c index 9affb0cb3d..41d58477fb 100644 --- a/src/resolve/test-dns-question.c +++ b/src/resolve/test-dns-question.c @@ -901,4 +901,41 @@ TEST(dns_question_dump) { check_dump_contents(f, expected, 3); } +/* ================================================================ + * dns_question_first_name() + * ================================================================ */ + +TEST(dns_question_first_name_empty) { + _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL; + const char *name = NULL; + + question = dns_question_new(0); + ASSERT_NOT_NULL(question); + + name = dns_question_first_name(question); + ASSERT_NULL(name); +} + +TEST(dns_question_first_name_multi) { + _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL; + DnsResourceKey *key = NULL; + const char *name = NULL; + + question = dns_question_new(2); + ASSERT_NOT_NULL(question); + + key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com"); + ASSERT_NOT_NULL(key); + dns_question_add(question, key, 0); + dns_resource_key_unref(key); + + key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "mail.example.com"); + ASSERT_NOT_NULL(key); + dns_question_add(question, key, 0); + dns_resource_key_unref(key); + + name = dns_question_first_name(question); + ASSERT_STREQ(name, "www.example.com"); +} + DEFINE_TEST_MAIN(LOG_DEBUG); -- 2.25.1