- fix error check

[oweals/gnunet.git] / src / mesh / test_mesh_api.c

diff --git a/src/mesh/test_mesh_api.c b/src/mesh/test_mesh_api.c
index 9828ee8d9a8f405b564056c2b5cf41642295930f..dcac19085ef4348d716ea38252ecc72f512c5386 100644 (file)
--- a/src/mesh/test_mesh_api.c
+++ b/src/mesh/test_mesh_api.c
@@ -23,20 +23,20 @@
  * @brief test mesh api: dummy test of callbacks
  * @author Bartlomiej Polot
  */
-
 #include "platform.h"
 #include "gnunet_util_lib.h"
+#include "gnunet_testing_lib-new.h"
 #include "gnunet_dht_service.h"
-#include "gnunet_mesh_service_new.h"
-
-#define VERBOSE 1
-#define VERBOSE_ARM 0
+#include "gnunet_mesh_service.h"
 
-static struct GNUNET_OS_Process *arm_pid;
 static struct GNUNET_MESH_Handle *mesh;
+
+static struct GNUNET_MESH_Tunnel *t;
+
 static int result;
-GNUNET_SCHEDULER_TaskIdentifier abort_task;
-GNUNET_SCHEDULER_TaskIdentifier test_task;
+
+static GNUNET_SCHEDULER_TaskIdentifier abort_task;
+
 
 /**
  * Function is called whenever a message is received.
@@ -51,24 +51,28 @@ GNUNET_SCHEDULER_TaskIdentifier test_task;
  *         GNUNET_SYSERR to close it (signal serious error)
  */
 static int
-callback (void *cls,
-          struct GNUNET_MESH_Tunnel *tunnel,
-          void **tunnel_ctx,
+callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
           const struct GNUNET_PeerIdentity *sender,
           const struct GNUNET_MessageHeader *message,
-          const struct GNUNET_TRANSPORT_ATS_Information *atsi)
+          const struct GNUNET_ATS_Information *atsi)
 {
-  return 0;
+  return GNUNET_OK;
 }
 
-static struct GNUNET_MESH_MessageHandler handlers[] = { {&callback, 1, 0},
-{NULL, 0, 0}
+
+static struct GNUNET_MESH_MessageHandler handlers[] = { 
+  { &callback, 1, 0 },
+  { NULL, 0, 0 }
 };
 
 
 static void
 do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
+  if (NULL != t)
+  {
+    GNUNET_MESH_tunnel_destroy (t);
+  }
   if (0 != abort_task)
   {
     GNUNET_SCHEDULER_cancel (abort_task);
@@ -77,113 +81,58 @@ do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   {
     GNUNET_MESH_disconnect (mesh);
   }
-  if (0 != GNUNET_OS_process_kill (arm_pid, SIGTERM))
-  {
-    GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
-  }
-  GNUNET_assert (GNUNET_OK == GNUNET_OS_process_wait (arm_pid));
-  GNUNET_OS_process_close (arm_pid);
 }
 
+
 static void
 do_abort (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-  if (0 != test_task)
-  {
-    GNUNET_SCHEDULER_cancel (test_task);
-  }
   result = GNUNET_SYSERR;
   abort_task = 0;
   do_shutdown (cls, tc);
 }
 
+
 static void
-test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+run (void *cls, 
+     const struct GNUNET_CONFIGURATION_Handle *cfg,
+     struct GNUNET_TESTING_Peer *peer)
 {
-  struct GNUNET_CONFIGURATION_Handle *cfg = cls;
-  GNUNET_MESH_ApplicationType app[3];
-
-  test_task = (GNUNET_SCHEDULER_TaskIdentifier) 0;
-  app[0] = (GNUNET_MESH_ApplicationType) 1;
-  app[1] = (GNUNET_MESH_ApplicationType) 2;
-  app[2] = (GNUNET_MESH_ApplicationType) 0;
-  mesh = GNUNET_MESH_connect (cfg, NULL, NULL, handlers, &app);
+  static const GNUNET_MESH_ApplicationType app[] =
+    { 1, 2, 3, 4, 5, 6, 7, 8, 0 };
+
+  mesh = GNUNET_MESH_connect (cfg, NULL, NULL, NULL, handlers, app);
   if (NULL == mesh)
   {
-    GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Couldn't connect to mesh :(\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "test: Couldn't connect to mesh :(\n");
+    result = GNUNET_SYSERR;
     return;
   }
   else
   {
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "YAY! CONNECTED TO MESH :D\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: YAY! CONNECTED TO MESH :D\n");
   }
-
+  t = GNUNET_MESH_tunnel_create (mesh, NULL, NULL, NULL, NULL);
   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
-                                (GNUNET_TIME_UNIT_SECONDS, 1), &do_shutdown,
+                                (GNUNET_TIME_UNIT_SECONDS, 5), &do_shutdown,
                                 NULL);
-}
-
-
-static void
-run (void *cls,
-     char *const *args,
-     const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
-{
-  arm_pid = GNUNET_OS_start_process (NULL, NULL,
-                                     "gnunet-service-arm", "gnunet-service-arm",
-#if VERBOSE_ARM
-                                     "-L", "DEBUG",
-#endif
-                                     "-c", "test_mesh.conf", NULL);
-
   abort_task =
       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
                                     (GNUNET_TIME_UNIT_SECONDS, 20), &do_abort,
                                     NULL);
-  test_task =
-      GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
-                                    (GNUNET_TIME_UNIT_SECONDS, 1), &test,
-                                    (void *) cfg);
-
 }
 
 
 int
 main (int argc, char *argv[])
 {
-  int ret;
-
-  char *const argv2[] = { "test-mesh-api",
-    "-c", "test_mesh.conf",
-#if VERBOSE
-    "-L", "DEBUG",
-#endif
-    NULL
-  };
-  struct GNUNET_GETOPT_CommandLineOption options[] = {
-    GNUNET_GETOPT_OPTION_END
-  };
-  GNUNET_log_setup ("test-mesh-api",
-#if VERBOSE
-                    "DEBUG",
-#else
-                    "WARNING",
-#endif
-                    NULL);
-  ret = GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1,
-                            argv2, "test-mesh-api", "nohelp",
-                            options, &run, NULL);
-
-  if (GNUNET_OK != ret)
-  {
-    GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
-                "run failed with error code %d\n", ret);
-    return 1;
-  }
-  if (GNUNET_SYSERR == result)
-  {
-    GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "test failed\n");
+  result = GNUNET_OK;
+  if (0 != GNUNET_TESTING_peer_run ("test-mesh-api",
+                                   "test_mesh.conf",
+                                   &run, NULL))
     return 1;
-  }
-  return 0;
+  return (result == GNUNET_OK) ? 0 : 1;
 }
+
+/* end of test_mesh_api.c */
+