- fix 2699

[oweals/gnunet.git] / src / consensus / test_consensus_api.c

diff --git a/src/consensus/test_consensus_api.c b/src/consensus/test_consensus_api.c
index a6bd60119766ad2d535f8fe0fa090748182c9ca0..c79a6221c27cfc39ad8863c557784bd132ea1d66 100644 (file)
--- a/src/consensus/test_consensus_api.c
+++ b/src/consensus/test_consensus_api.c
@@ -17,6 +17,7 @@
      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
      Boston, MA 02111-1307, USA.
 */
+
 /**
  * @file consensus/test_consensus_api.c
  * @brief testcase for consensus_api.c
@@ -24,7 +25,7 @@
 #include "platform.h"
 #include "gnunet_util_lib.h"
 #include "gnunet_consensus_service.h"
-#include "gnunet_testing_lib-new.h"
+#include "gnunet_testing_lib.h"
 
 
 static struct GNUNET_CONSENSUS_Handle *consensus;
@@ -32,11 +33,54 @@ static struct GNUNET_CONSENSUS_Handle *consensus;
 static struct GNUNET_HashCode session_id;
 
 
-void
+static int
+conclude_done (void *cls, const struct GNUNET_CONSENSUS_Group *group)
+{
+  if (NULL == group)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO, "conclude over\n");
+    GNUNET_SCHEDULER_shutdown ();
+    return GNUNET_NO;
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "concluded\n");
+  return GNUNET_YES;
+}
+
+static int
 on_new_element (void *cls,
                 struct GNUNET_CONSENSUS_Element *element)
 {
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "received new element\n");
+  GNUNET_assert (0);
+  return GNUNET_YES;
+}
+
+static void
+insert_done (void *cls, int success)
+{
+  /* make sure cb is only called once */
+  static int called = GNUNET_NO;
+  GNUNET_assert (GNUNET_NO == called);
+  called = GNUNET_YES;
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "insert done\n");
+  GNUNET_CONSENSUS_conclude (consensus, GNUNET_TIME_UNIT_SECONDS, 0, &conclude_done, NULL);
+}
+
+
+/**
+ * Signature of the main function of a task.
+ *
+ * @param cls closure
+ * @param tc context information (why was this task triggered now)
+ */
+static void
+on_shutdown (void *cls,
+          const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  if (NULL != consensus)
+  {
+    GNUNET_CONSENSUS_destroy (consensus);
+    consensus = NULL;
+  }
 }
 
 
@@ -47,10 +91,23 @@ run (void *cls,
 {
   char *str = "foo";
 
+  struct GNUNET_CONSENSUS_Element el1 = {"foo", 4, 0};
+  struct GNUNET_CONSENSUS_Element el2 = {"bar", 4, 0};
+
+  GNUNET_log_setup ("test_consensus_api",
+                    "INFO",
+                    NULL);
+
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "testing consensus api\n");
+
+  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &on_shutdown, NULL);
+
   GNUNET_CRYPTO_hash (str, strlen (str), &session_id);
-  consensus = GNUNET_CONSENSUS_create (cfg, 0, NULL, &session_id, on_new_element, cls);
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connecting to consensus service.\n");
+  consensus = GNUNET_CONSENSUS_create (cfg, 0, NULL, &session_id, on_new_element, &consensus);
   GNUNET_assert (consensus != NULL);
+
+  GNUNET_CONSENSUS_insert (consensus, &el1, NULL, &consensus);
+  GNUNET_CONSENSUS_insert (consensus, &el2, &insert_done, &consensus);
 }
 
 
@@ -59,12 +116,6 @@ main (int argc, char **argv)
 {
   int ret;
 
-  GNUNET_log_setup ("test_consensus_api",
-                    "DEBUG",
-                    NULL);
-
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "testing consensus api\n");
-
   ret = GNUNET_TESTING_peer_run ("test_consensus_api",
                                  "test_consensus.conf",
                                  &run, NULL);