diff options
Diffstat (limited to 'tests/libknot/test_rrset.c')
| -rw-r--r-- | tests/libknot/test_rrset.c | 42 |
1 files changed, 41 insertions, 1 deletions
diff --git a/tests/libknot/test_rrset.c b/tests/libknot/test_rrset.c index cc67e0f..f91d143 100644 --- a/tests/libknot/test_rrset.c +++ b/tests/libknot/test_rrset.c @@ -1,4 +1,4 @@ -/* Copyright (C) 2019 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz> +/* Copyright (C) 2024 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz> This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -34,6 +34,43 @@ static bool check_rrset(const knot_rrset_t *rrset, const knot_dname_t *owner, rrset->ttl == ttl && rrset->rrs.count == 0; // We do not test rdataset here } +static void check_size(knot_rrset_t *rrset, uint8_t *rdata, size_t rdata_len) +{ + const size_t per_record_const = 10; + const size_t owner_len = knot_dname_size(rrset->owner); + assert(owner_len == 13); + + static size_t real = 0, estim = 0; + if (rdata_len > 0) { + real += owner_len + per_record_const + rdata_len; + estim += owner_len + per_record_const + rdata_len + (rdata_len & 1); + assert(real <= estim); + } + + int ret = knot_rrset_add_rdata(rrset, rdata, rdata_len, NULL); + is_int(ret, KNOT_EOK, "knot_rrset_add_rdata()"); + + is_int(real, knot_rrset_size(rrset), "knot_rrset_size()"); + is_int(estim, knot_rrset_size_estimate(rrset), "knot_rrset_size_estimate()"); +} + +static void test_rrset_size(void) +{ + knot_dname_t *owner = knot_dname_from_str_alloc("rrset.owner."); + assert(owner); + + knot_rrset_t rrset; + knot_rrset_init(&rrset, owner, KNOT_RRTYPE_TXT, KNOT_CLASS_IN, 7200); + + check_size(&rrset, (uint8_t *)"\x01""a", 2); + check_size(&rrset, (uint8_t *)"\x02""ab", 3); + check_size(&rrset, (uint8_t *)"\x03""abc", 4); + check_size(&rrset, (uint8_t *)"\x03""de", 3); + check_size(&rrset, (uint8_t *)"\x01""f", 2); + + knot_rrset_clear(&rrset, NULL); +} + int main(int argc, char *argv[]) { plan_lazy(); @@ -117,5 +154,8 @@ int main(int argc, char *argv[]) knot_rrset_free(rrset, NULL); knot_rrset_free(copy, NULL); + // Test rrset size computation functions. + test_rrset_size(); + return 0; } |