1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#include <ccan/invbloom/invbloom.h>
/* Include the C files directly. */
#include <ccan/invbloom/invbloom.c>
#include <ccan/tap/tap.h>
static void singleton_cb(struct invbloom *ib, size_t n, bool before,
unsigned *count)
{
ok1(ib->count[n] == 1 || ib->count[n] == -1);
count[before]++;
}
int main(void)
{
struct invbloom *ib;
const tal_t *ctx = tal(NULL, char);
int val;
unsigned singleton_count[2];
/* This is how many tests you plan to run */
plan_tests(16 + 6 + NUM_HASHES * 3);
/* Single entry ib table keeps it simple. */
ib = invbloom_new(ctx, int, 1, 100);
invbloom_singleton_cb(ib, singleton_cb, singleton_count);
val = 0;
singleton_count[false] = singleton_count[true] = 0;
invbloom_insert(ib, &val);
ok1(ib->count[0] == NUM_HASHES);
ok1(singleton_count[true] == 1);
ok1(singleton_count[false] == 1);
ok1(!invbloom_empty(ib));
/* First delete takes it via singleton. */
invbloom_delete(ib, &val);
ok1(singleton_count[true] == 2);
ok1(singleton_count[false] == 2);
ok1(invbloom_empty(ib));
/* Second delete creates negative singleton. */
invbloom_delete(ib, &val);
ok1(singleton_count[true] == 3);
ok1(singleton_count[false] == 3);
/* Now a larger table: this seed set so entries don't clash */
ib = invbloom_new(ctx, int, 1024, 0);
singleton_count[false] = singleton_count[true] = 0;
invbloom_singleton_cb(ib, singleton_cb, singleton_count);
val = 0;
invbloom_insert(ib, &val);
ok1(singleton_count[true] == 0);
ok1(singleton_count[false] == NUM_HASHES);
/* First delete removes singletons. */
invbloom_delete(ib, &val);
ok1(singleton_count[true] == NUM_HASHES);
ok1(singleton_count[false] == NUM_HASHES);
ok1(invbloom_empty(ib));
/* Second delete creates negative singletons. */
invbloom_delete(ib, &val);
ok1(singleton_count[true] == NUM_HASHES);
ok1(singleton_count[false] == NUM_HASHES * 2);
tal_free(ctx);
/* This exits depending on whether all tests passed */
return exit_status();
}
|
