]>
git.ipfire.org Git - thirdparty/systemd.git/blob - src/test/test-set.c
1 /* SPDX-License-Identifier: LGPL-2.1+ */
6 static void test_set_steal_first(void) {
7 _cleanup_set_free_ Set
*m
= NULL
;
11 m
= set_new(&string_hash_ops
);
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);
18 while ((val
= set_steal_first(m
)))
19 seen
[strlen(val
) - 1]++;
21 assert_se(seen
[0] == 1 && seen
[1] == 1 && seen
[2] == 1);
23 assert_se(set_isempty(m
));
29 static void item_seen(Item
*item
) {
33 static void test_set_free_with_destructor(void) {
35 struct Item items
[4] = {};
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);
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);
49 DEFINE_PRIVATE_HASH_OPS_WITH_VALUE_DESTRUCTOR(item_hash_ops
, void, trivial_hash_func
, trivial_compare_func
, Item
, item_seen
);
51 static void test_set_free_with_hash_ops(void) {
53 struct Item items
[4] = {};
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);
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);
67 static void test_set_put(void) {
68 _cleanup_set_free_ Set
*m
= NULL
;
70 m
= set_new(&string_hash_ops
);
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);
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);
89 int main(int argc
, const char *argv
[]) {
90 test_set_steal_first();
91 test_set_free_with_destructor();
92 test_set_free_with_hash_ops();