]>
Commit | Line | Data |
---|---|---|
647f6824 ZJS |
1 | /*** |
2 | This file is part of systemd | |
3 | ||
4 | Copyright 2014 Zbigniew Jędrzejewski-Szmek | |
5 | ||
6 | systemd is free software; you can redistribute it and/or modify it | |
7 | under the terms of the GNU Lesser General Public License as published by | |
8 | the Free Software Foundation; either version 2.1 of the License, or | |
9 | (at your option) any later version. | |
10 | ||
11 | systemd is distributed in the hope that it will be useful, but | |
12 | WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | Lesser General Public License for more details. | |
15 | ||
16 | You should have received a copy of the GNU Lesser General Public License | |
17 | along with systemd; If not, see <http://www.gnu.org/licenses/>. | |
18 | ***/ | |
19 | ||
647f6824 ZJS |
20 | #include "set.h" |
21 | ||
22 | static void test_set_steal_first(void) { | |
23 | _cleanup_set_free_ Set *m = NULL; | |
24 | int seen[3] = {}; | |
25 | char *val; | |
26 | ||
27 | m = set_new(&string_hash_ops); | |
28 | assert_se(m); | |
29 | ||
30 | assert_se(set_put(m, (void*) "1") == 1); | |
31 | assert_se(set_put(m, (void*) "22") == 1); | |
32 | assert_se(set_put(m, (void*) "333") == 1); | |
33 | ||
34 | while ((val = set_steal_first(m))) | |
35 | seen[strlen(val) - 1]++; | |
36 | ||
bdf7026e | 37 | assert_se(seen[0] == 1 && seen[1] == 1 && seen[2] == 1); |
647f6824 ZJS |
38 | |
39 | assert_se(set_isempty(m)); | |
40 | } | |
41 | ||
756c09e6 RC |
42 | static void test_set_put(void) { |
43 | _cleanup_set_free_ Set *m = NULL; | |
44 | ||
45 | m = set_new(&string_hash_ops); | |
46 | assert_se(m); | |
47 | ||
48 | assert_se(set_put(m, (void*) "1") == 1); | |
49 | assert_se(set_put(m, (void*) "22") == 1); | |
50 | assert_se(set_put(m, (void*) "333") == 1); | |
51 | assert_se(set_put(m, (void*) "333") == 0); | |
52 | assert_se(set_remove(m, (void*) "333")); | |
53 | assert_se(set_put(m, (void*) "333") == 1); | |
54 | assert_se(set_put(m, (void*) "333") == 0); | |
55 | assert_se(set_put(m, (void*) "22") == 0); | |
56 | } | |
57 | ||
647f6824 ZJS |
58 | int main(int argc, const char *argv[]) { |
59 | test_set_steal_first(); | |
756c09e6 | 60 | test_set_put(); |
647f6824 ZJS |
61 | |
62 | return 0; | |
63 | } |