src/nse/test_nse_api.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 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  * @file nse/test_nse_api.c
  22  * @brief testcase for nse_api.c
  23  */
  24 #include "platform.h"
  25 #include "gnunet_util_lib.h"
  26 #include "gnunet_nse_service.h"
  27 #include "gnunet_testing_lib.h"
  28
  29
  30 static struct GNUNET_NSE_Handle *h;
  31
  32 static GNUNET_SCHEDULER_TaskIdentifier die_task;
  33
  34
  35 /**
  36  * Signature of the main function of a task.
  37  *
  38  * @param cls closure
  39  * @param tc context information (why was this task triggered now)
  40  */
  41 static void
  42 end_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  43 {
  44   if (h != NULL)
  45   {
  46     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from NSE service.\n");
  47     GNUNET_NSE_disconnect (h);
  48   }
  49 }
  50
  51 /**
  52  * Callback to call when network size estimate is updated.
  53  *
  54  * @param cls unused
  55  * @param timestamp time when the estimate was received from the server (or created by the server)
  56  * @param estimate the value of the current network size estimate
  57  * @param std_dev standard deviation (rounded down to nearest integer)
  58  *                of the size estimation values seen
  59  *
  60  */
  61 static void
  62 check_nse_message (void *cls, struct GNUNET_TIME_Absolute timestamp,
  63                    double estimate, double std_dev)
  64 {
  65   int *ok = cls;
  66
  67   FPRINTF (stderr,
  68            "Received NSE message, estimate %f, standard deviation %f.\n",
  69            estimate, std_dev);
  70   /* Fantastic check below. Expect NaN, the only thing not equal to itself. */
  71   (*ok) = 0;
  72   if (die_task != GNUNET_SCHEDULER_NO_TASK)
  73     GNUNET_SCHEDULER_cancel (die_task);
  74   die_task = GNUNET_SCHEDULER_add_now (&end_test, NULL);
  75 }
  76
  77
  78 static void
  79 run (void *cls,
  80      const struct GNUNET_CONFIGURATION_Handle *cfg,
  81      struct GNUNET_TESTING_Peer *peer)
  82 {
  83   die_task =
  84       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
  85                                     (GNUNET_TIME_UNIT_MINUTES, 1), &end_test,
  86                                     NULL);
  87
  88   h = GNUNET_NSE_connect (cfg, &check_nse_message, cls);
  89   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connecting to NSE service.\n");
  90   GNUNET_assert (h != NULL);
  91 }
  92
  93
  94 int
  95 main (int argc, char *argv[])
  96 {
  97   int ok = 1;
  98
  99   if (0 != GNUNET_TESTING_peer_run ("test_nse_api",
 100                                     "test_nse.conf",
 101                                     &run, &ok))
 102     return 1;
 103   return ok;
 104 }
 105
 106 /* end of test_nse_api.c */