]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/test/test-set.c
Merge pull request #12223 from yuwata/network-wireguard-preshared-key-file
[thirdparty/systemd.git] / src / test / test-set.c
CommitLineData
53e1b683 1/* SPDX-License-Identifier: LGPL-2.1+ */
647f6824 2
647f6824 3#include "set.h"
ef79eae0 4#include "strv.h"
647f6824
ZJS
5
6static void test_set_steal_first(void) {
7 _cleanup_set_free_ Set *m = NULL;
8 int seen[3] = {};
9 char *val;
10
11 m = set_new(&string_hash_ops);
12 assert_se(m);
13
14 assert_se(set_put(m, (void*) "1") == 1);
15 assert_se(set_put(m, (void*) "22") == 1);
16 assert_se(set_put(m, (void*) "333") == 1);
17
18 while ((val = set_steal_first(m)))
19 seen[strlen(val) - 1]++;
20
bdf7026e 21 assert_se(seen[0] == 1 && seen[1] == 1 && seen[2] == 1);
647f6824
ZJS
22
23 assert_se(set_isempty(m));
24}
25
224b0e7a
ZJS
26typedef struct Item {
27 int seen;
28} Item;
29static void item_seen(Item *item) {
30 item->seen++;
31}
32
33static void test_set_free_with_destructor(void) {
34 Set *m;
35 struct Item items[4] = {};
36 unsigned i;
37
38 assert_se(m = set_new(NULL));
39 for (i = 0; i < ELEMENTSOF(items) - 1; i++)
40 assert_se(set_put(m, items + i) == 1);
41
42 m = set_free_with_destructor(m, item_seen);
43 assert_se(items[0].seen == 1);
44 assert_se(items[1].seen == 1);
45 assert_se(items[2].seen == 1);
46 assert_se(items[3].seen == 0);
47}
48
98233ee5
YW
49DEFINE_PRIVATE_HASH_OPS_WITH_VALUE_DESTRUCTOR(item_hash_ops, void, trivial_hash_func, trivial_compare_func, Item, item_seen);
50
51static void test_set_free_with_hash_ops(void) {
52 Set *m;
53 struct Item items[4] = {};
54 unsigned i;
55
56 assert_se(m = set_new(&item_hash_ops));
57 for (i = 0; i < ELEMENTSOF(items) - 1; i++)
58 assert_se(set_put(m, items + i) == 1);
59
60 m = set_free(m);
61 assert_se(items[0].seen == 1);
62 assert_se(items[1].seen == 1);
63 assert_se(items[2].seen == 1);
64 assert_se(items[3].seen == 0);
65}
66
756c09e6
RC
67static void test_set_put(void) {
68 _cleanup_set_free_ Set *m = NULL;
69
70 m = set_new(&string_hash_ops);
71 assert_se(m);
72
73 assert_se(set_put(m, (void*) "1") == 1);
74 assert_se(set_put(m, (void*) "22") == 1);
75 assert_se(set_put(m, (void*) "333") == 1);
76 assert_se(set_put(m, (void*) "333") == 0);
77 assert_se(set_remove(m, (void*) "333"));
78 assert_se(set_put(m, (void*) "333") == 1);
79 assert_se(set_put(m, (void*) "333") == 0);
80 assert_se(set_put(m, (void*) "22") == 0);
ef79eae0
ZJS
81
82 _cleanup_free_ char **t = set_get_strv(m);
83 assert_se(strv_contains(t, "1"));
84 assert_se(strv_contains(t, "22"));
85 assert_se(strv_contains(t, "333"));
86 assert_se(strv_length(t) == 3);
756c09e6
RC
87}
88
647f6824
ZJS
89int main(int argc, const char *argv[]) {
90 test_set_steal_first();
224b0e7a 91 test_set_free_with_destructor();
98233ee5 92 test_set_free_with_hash_ops();
756c09e6 93 test_set_put();
647f6824
ZJS
94
95 return 0;
96}