1 files changed, 112 insertions, 0 deletions

diff --git a/src/util/test_container_heap.c b/src/util/test_container_heap.c
new file mode 100644
index 000000000..511589af5
--- /dev/null
+++ b/src/util/test_container_heap.c
@@ -0,0 +1,112 @@
+/*
+ This file is part of GNUnet.
+ (C) 2008 Christian Grothoff (and other contributing authors)
+
+ GNUnet is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published
+ by the Free Software Foundation; either version 2, or (at your
+ option) any later version.
+
+ GNUnet is distributed in the hope that it will be useful, but
+ WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with GNUnet; see the file COPYING.  If not, write to the
+ Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ Boston, MA 02111-1307, USA.
+ */
+
+/**
+ * @author Nathan Evans
+ * @file util/containers/heaptest.c
+ * @brief Test of heap operations
+ */
+
+#include "gnunet_util.h"
+#include "gnunet_util_containers.h"
+#include "dv.h"
+
+static int
+iterator_callback (void *element, GNUNET_CONTAINER_HeapCost cost,
+                   struct GNUNET_CONTAINER_Heap *root, void *cls)
+{
+  struct GNUNET_dv_neighbor *node;
+  node = (struct GNUNET_dv_neighbor *) element;
+  fprintf (stdout, "%d\n", node->cost);
+  //fprintf (stdout, "%d\n", ((struct GNUNET_dv_neighbor *)element)->cost);
+
+  return GNUNET_OK;
+}
+
+
+int
+main (int argc, char **argv)
+{
+  struct GNUNET_CONTAINER_Heap *myHeap;
+  struct GNUNET_dv_neighbor *neighbor1;
+  struct GNUNET_dv_neighbor *neighbor2;
+  struct GNUNET_dv_neighbor *neighbor3;
+  struct GNUNET_dv_neighbor *neighbor4;
+  struct GNUNET_dv_neighbor *neighbor5;
+  struct GNUNET_dv_neighbor *neighbor6;
+
+  GNUNET_log_setup ("test-container-heap", "WARNING", NULL);
+
+  myHeap = GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MAX);
+
+  neighbor1 = malloc (sizeof (struct GNUNET_dv_neighbor));
+  neighbor2 = malloc (sizeof (struct GNUNET_dv_neighbor));
+  neighbor3 = malloc (sizeof (struct GNUNET_dv_neighbor));
+  neighbor4 = malloc (sizeof (struct GNUNET_dv_neighbor));
+  neighbor5 = malloc (sizeof (struct GNUNET_dv_neighbor));
+  neighbor6 = malloc (sizeof (struct GNUNET_dv_neighbor));
+
+  neighbor1->cost = 60;
+  neighbor2->cost = 50;
+  neighbor3->cost = 70;
+  neighbor4->cost = 120;
+  neighbor5->cost = 100;
+  neighbor6->cost = 30;
+
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor1, neighbor1->cost);
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor2, neighbor2->cost);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor3, neighbor3->cost);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor4, neighbor4->cost);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor5, neighbor5->cost);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_insert (myHeap, neighbor6, neighbor6->cost);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_remove_node (myHeap, neighbor5);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_remove_root (myHeap);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  GNUNET_CONTAINER_heap_update_cost (myHeap, neighbor6, 200);
+
+  GNUNET_CONTAINER_heap_iterate (myHeap, iterator_callback, NULL);
+  fprintf (stdout, "\n");
+  return 0;
+}
+
+/* end of heaptest.c */