- test for external iterator
[oweals/gnunet.git] / src / util / test_container_multihashmap.c
1 /*
2      This file is part of GNUnet.
3      (C) 2008 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file util/test_container_multihashmap.c
23  * @brief Test for container_multihashmap.c
24  * @author Christian Grothoff
25  */
26
27 #include "platform.h"
28 #include "gnunet_common.h"
29 #include "gnunet_container_lib.h"
30
31 #define ABORT() { fprintf(stderr, "Error at %s:%d\n", __FILE__, __LINE__); if (m != NULL) GNUNET_CONTAINER_multihashmap_destroy(m); return 1; }
32 #define CHECK(c) { if (! (c)) ABORT(); }
33
34 static int
35 testMap (int i)
36 {
37   struct GNUNET_CONTAINER_MultiHashMap *m;
38   struct GNUNET_HashCode k1;
39   struct GNUNET_HashCode k2;
40   struct GNUNET_CONTAINER_MultiHashMapIterator *iter;
41   struct GNUNET_HashCode key_ret;
42   const char *ret;
43   int j;
44
45   CHECK (NULL != (m = GNUNET_CONTAINER_multihashmap_create (i, GNUNET_NO)));
46   memset (&k1, 0, sizeof (k1));
47   memset (&k2, 1, sizeof (k2));
48   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k1));
49   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k2));
50   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_remove (m, &k1, NULL));
51   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_remove (m, &k2, NULL));
52   CHECK (NULL == GNUNET_CONTAINER_multihashmap_get (m, &k1));
53   CHECK (NULL == GNUNET_CONTAINER_multihashmap_get (m, &k2));
54   CHECK (0 == GNUNET_CONTAINER_multihashmap_remove_all (m, &k1));
55   CHECK (0 == GNUNET_CONTAINER_multihashmap_size (m));
56   CHECK (0 == GNUNET_CONTAINER_multihashmap_iterate (m, NULL, NULL));
57   CHECK (0 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k1, NULL, NULL));
58
59   CHECK (GNUNET_OK ==
60          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v1",
61                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE));
62   CHECK (1 == GNUNET_CONTAINER_multihashmap_size (m));
63   ret = GNUNET_CONTAINER_multihashmap_get (m, &k1);
64   GNUNET_assert (ret != NULL);
65   CHECK (0 == strcmp ("v1", ret));
66   CHECK (GNUNET_NO ==
67          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v1",
68                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE));
69   CHECK (1 == GNUNET_CONTAINER_multihashmap_size (m));
70   CHECK (GNUNET_OK ==
71          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v2",
72                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
73   CHECK (GNUNET_OK ==
74          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v3",
75                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
76   CHECK (3 == GNUNET_CONTAINER_multihashmap_size (m));
77   CHECK (GNUNET_OK == GNUNET_CONTAINER_multihashmap_remove (m, &k1, "v3"));
78   CHECK (2 == GNUNET_CONTAINER_multihashmap_size (m));
79   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_contains (m, &k1));
80   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k2));
81   CHECK (2 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k1, NULL, NULL));
82   CHECK (0 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k2, NULL, NULL));
83   CHECK (2 == GNUNET_CONTAINER_multihashmap_iterate (m, NULL, NULL));
84   iter = GNUNET_CONTAINER_multihashmap_iterator_create (m);
85   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, &key_ret, (const void **)&ret));
86   CHECK (0 == memcmp (&key_ret, &k1, sizeof (key_ret)));
87   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, &key_ret, (const void **)&ret));
88   CHECK (0 == memcmp (&key_ret, &k1, sizeof (key_ret)));
89   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
90
91   CHECK (2 == GNUNET_CONTAINER_multihashmap_remove_all (m, &k1));
92   for (j = 0; j < 1024; j++)
93     CHECK (GNUNET_OK ==
94            GNUNET_CONTAINER_multihashmap_put (m, &k1, "v2",
95                                               GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
96   iter = GNUNET_CONTAINER_multihashmap_iterator_create (m);
97   for (j = 0; j < GNUNET_CONTAINER_multihashmap_size (m); j++)
98     CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
99   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
100
101   GNUNET_CONTAINER_multihashmap_destroy (m);
102   return 0;
103 }
104
105 int
106 main (int argc, char *argv[])
107 {
108   int failureCount = 0;
109   int i;
110
111   GNUNET_log_setup ("test-container-multihashmap", "WARNING", NULL);
112   for (i = 1; i < 255; i++)
113     failureCount += testMap (i);
114   if (failureCount != 0)
115     return 1;
116   return 0;
117 }
118
119 /* end of test_container_multihashmap.c */