src/util/test_container_multihashmap.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2008 GNUnet e.V.
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your 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      Affero General Public License for more details.
  14
  15      You should have received a copy of the GNU Affero General Public License
  16      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 /**
  20  * @file util/test_container_multihashmap.c
  21  * @brief Test for container_multihashmap.c
  22  * @author Christian Grothoff
  23  */
  24
  25 #include "platform.h"
  26 #include "gnunet_util_lib.h"
  27
  28 #define ABORT() { fprintf(stderr, "Error at %s:%d\n", __FILE__, __LINE__); if (m != NULL) GNUNET_CONTAINER_multihashmap_destroy(m); if (NULL != iter) GNUNET_CONTAINER_multihashmap_iterator_destroy (iter); return 1; }
  29 #define CHECK(c) { if (! (c)) ABORT(); }
  30
  31 static int
  32 testMap (int i)
  33 {
  34   struct GNUNET_CONTAINER_MultiHashMap *m;
  35   struct GNUNET_HashCode k1;
  36   struct GNUNET_HashCode k2;
  37   struct GNUNET_CONTAINER_MultiHashMapIterator *iter = NULL;
  38   struct GNUNET_HashCode key_ret;
  39   const char *ret;
  40   int j;
  41
  42   CHECK (NULL != (m = GNUNET_CONTAINER_multihashmap_create (i, GNUNET_NO)));
  43   memset (&k1, 0, sizeof (k1));
  44   memset (&k2, 1, sizeof (k2));
  45   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k1));
  46   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k2));
  47   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_remove (m, &k1, NULL));
  48   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_remove (m, &k2, NULL));
  49   CHECK (NULL == GNUNET_CONTAINER_multihashmap_get (m, &k1));
  50   CHECK (NULL == GNUNET_CONTAINER_multihashmap_get (m, &k2));
  51   CHECK (0 == GNUNET_CONTAINER_multihashmap_remove_all (m, &k1));
  52   CHECK (0 == GNUNET_CONTAINER_multihashmap_size (m));
  53   CHECK (0 == GNUNET_CONTAINER_multihashmap_iterate (m, NULL, NULL));
  54   CHECK (0 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k1, NULL, NULL));
  55
  56   CHECK (GNUNET_OK ==
  57          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v1",
  58                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE));
  59   CHECK (1 == GNUNET_CONTAINER_multihashmap_size (m));
  60   ret = GNUNET_CONTAINER_multihashmap_get (m, &k1);
  61   GNUNET_assert (ret != NULL);
  62   CHECK (0 == strcmp ("v1", ret));
  63   CHECK (GNUNET_NO ==
  64          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v1",
  65                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE));
  66   CHECK (1 == GNUNET_CONTAINER_multihashmap_size (m));
  67   CHECK (GNUNET_OK ==
  68          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v2",
  69                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
  70   CHECK (GNUNET_OK ==
  71          GNUNET_CONTAINER_multihashmap_put (m, &k1, "v3",
  72                                             GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
  73   CHECK (3 == GNUNET_CONTAINER_multihashmap_size (m));
  74   CHECK (GNUNET_OK == GNUNET_CONTAINER_multihashmap_remove (m, &k1, "v3"));
  75   CHECK (2 == GNUNET_CONTAINER_multihashmap_size (m));
  76   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_contains (m, &k1));
  77   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_contains (m, &k2));
  78   CHECK (2 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k1, NULL, NULL));
  79   CHECK (0 == GNUNET_CONTAINER_multihashmap_get_multiple (m, &k2, NULL, NULL));
  80   CHECK (2 == GNUNET_CONTAINER_multihashmap_iterate (m, NULL, NULL));
  81   iter = GNUNET_CONTAINER_multihashmap_iterator_create (m);
  82   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, &key_ret, (const void **)&ret));
  83   CHECK (0 == memcmp (&key_ret, &k1, sizeof (key_ret)));
  84   CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, &key_ret, (const void **)&ret));
  85   CHECK (0 == memcmp (&key_ret, &k1, sizeof (key_ret)));
  86   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
  87   GNUNET_free (iter);
  88
  89   CHECK (2 == GNUNET_CONTAINER_multihashmap_remove_all (m, &k1));
  90   for (j = 0; j < 1024; j++)
  91     CHECK (GNUNET_OK ==
  92            GNUNET_CONTAINER_multihashmap_put (m, &k1, "v2",
  93                                               GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
  94   iter = GNUNET_CONTAINER_multihashmap_iterator_create (m);
  95   for (j = 0; j < GNUNET_CONTAINER_multihashmap_size (m); j++)
  96     CHECK (GNUNET_YES == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
  97   CHECK (GNUNET_NO == GNUNET_CONTAINER_multihashmap_iterator_next (iter, NULL, NULL));
  98   GNUNET_CONTAINER_multihashmap_iterator_destroy (iter);
  99
 100   GNUNET_CONTAINER_multihashmap_destroy (m);
 101   return 0;
 102 }
 103
 104 int
 105 main (int argc, char *argv[])
 106 {
 107   int failureCount = 0;
 108   int i;
 109
 110   GNUNET_log_setup ("test-container-multihashmap", "WARNING", NULL);
 111   for (i = 1; i < 255; i++)
 112     failureCount += testMap (i);
 113   if (failureCount != 0)
 114     return 1;
 115   return 0;
 116 }
 117
 118 /* end of test_container_multihashmap.c */