mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-09-20 04:15:19 +00:00
99 lines
2.1 KiB
C
99 lines
2.1 KiB
C
/*
|
|
* BIRD Library -- Trie index Tests
|
|
*
|
|
* (c) 2019 Maria Matejka <mq@jmq.cz>
|
|
* (c) 2019 CZ.NIC z.s.p.o.
|
|
*
|
|
* Can be freely distributed and used under the terms of the GNU GPL.
|
|
*/
|
|
|
|
#include "test/birdtest.h"
|
|
#include "test/bt-utils.h"
|
|
#include "lib/tindex.h"
|
|
|
|
struct test_trie {
|
|
struct tindex *ti;
|
|
u64 *data;
|
|
u64 len;
|
|
};
|
|
|
|
static inline void test_trie_add(struct test_trie *tt, u64 data) {
|
|
u32 dtb[2] = { data >> 32, data };
|
|
u64 idx = tindex_find(tt->ti, dtb, 64, TINDEX_CREATE);
|
|
bt_assert(idx > 0);
|
|
|
|
u64 nlen = tt->len;
|
|
while (idx >= nlen)
|
|
nlen *= 2;
|
|
|
|
if (nlen > tt->len) {
|
|
tt->data = mb_realloc(tt->data, nlen * sizeof(u64));
|
|
memset(&(tt->data[tt->len]), 0, (nlen - tt->len) * sizeof(u64));
|
|
tt->len = nlen;
|
|
}
|
|
|
|
tt->data[idx]++;
|
|
}
|
|
|
|
static inline void test_trie_get(struct test_trie *tt, u64 data, u64 cnt) {
|
|
u64 out = 0;
|
|
u32 dtb[2] = { data >> 32, data };
|
|
u64 idx = tindex_find(tt->ti, dtb, 64, TINDEX_FIND);
|
|
if (idx) out = tt->data[idx];
|
|
bt_assert_msg(out == cnt, "Index %lu shall be in trie %lu times, is %lu times.", data, cnt, out);
|
|
}
|
|
|
|
/*
|
|
static inline void test_trie_remove(struct test_trie *tt, u64 data) {
|
|
u64 idx = tindex_find(tt->ti, &data, 64, TINDEX_FIND);
|
|
ASSERT(idx);
|
|
ASSERT(tt->data[idx]);
|
|
if (!--tt->data[idx])
|
|
tindex_delete(tt->ti, idx);
|
|
}
|
|
*/
|
|
|
|
static int
|
|
t_simple(void)
|
|
{
|
|
const u64 mul = 0xf906f046b1fd4863ULL;
|
|
const u64 add = 0xb3a35ec46d09489bULL;
|
|
pool *p = rp_new(&root_pool, "tindex test");
|
|
struct test_trie tt = {
|
|
.ti = tindex_new(p),
|
|
.data = mb_allocz(p, sizeof(u64) * 256),
|
|
.len = 256,
|
|
};
|
|
|
|
const u64 max = bt_benchmark ? (1<<19) : (1<<16);
|
|
|
|
bt_assert(tt.ti);
|
|
for (u64 i = 0; i < max; i++) {
|
|
bt_debug("Trie add: %lu\n", i);
|
|
test_trie_add(&tt, i);
|
|
test_trie_add(&tt, i * mul + add);
|
|
}
|
|
|
|
for (u64 i = 0; i < max; i++) {
|
|
test_trie_get(&tt, i, 1);
|
|
test_trie_get(&tt, i * mul + add, 1);
|
|
}
|
|
|
|
/*
|
|
for (u64 i = 0; i < 20; i++)
|
|
test_trie_remove(&tt, i);
|
|
*/
|
|
|
|
tindex_dump(tt.ti);
|
|
|
|
return 1;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
bt_init(argc, argv);
|
|
bt_bird_init();
|
|
bt_test_suite(t_simple, "tindex");
|
|
return bt_exit_value();
|
|
}
|