src/testing/test_testing_peerstartup2.c

   1 /*
   2       This file is part of GNUnet
   3       Copyright (C) 2008, 2009, 2012 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      SPDX-License-Identifier: AGPL3.0-or-later
  19  */
  20
  21 /**
  22  * @file testing/test_testing_new_peerstartup.c
  23  * @brief test case for testing peer startup and shutdown using new testing
  24  *          library
  25  * @author Sree Harsha Totakura
  26  */
  27
  28 #include "platform.h"
  29 #include "gnunet_util_lib.h"
  30 #include "gnunet_testing_lib.h"
  31
  32 #define LOG(kind,...)                           \
  33   GNUNET_log (kind, __VA_ARGS__)
  34
  35
  36 #define FAIL_TEST(cond)                         \
  37   do {                                          \
  38     if (!(cond)) {                              \
  39       GNUNET_break (0);                         \
  40       if (GNUNET_OK == status) {                \
  41         status = GNUNET_SYSERR;                 \
  42       }                                         \
  43     }                                           \
  44   } while (0)                                   \
  45
  46
  47 /**
  48  * The status of the test
  49  */
  50 int status;
  51
  52 /**
  53  * The testing context
  54  */
  55 struct TestingContext
  56 {
  57   /**
  58    * The testing system
  59    */
  60   struct GNUNET_TESTING_System *system;
  61
  62   /**
  63    * The peer which has been started by the testing system
  64    */
  65   struct GNUNET_TESTING_Peer *peer;
  66
  67   /**
  68    * The running configuration of the peer
  69    */
  70   struct GNUNET_CONFIGURATION_Handle *cfg;
  71
  72   /**
  73    * State
  74    */
  75   enum {
  76     PEER_INIT,
  77
  78     PEER_STARTED,
  79
  80     PEER_STOPPED
  81   } state;
  82 };
  83
  84
  85 static void
  86 do_shutdown2 (void *cls)
  87 {
  88   struct TestingContext *test_ctx = cls;
  89
  90   if (NULL != test_ctx->peer)
  91     GNUNET_TESTING_peer_destroy (test_ctx->peer);
  92   if (NULL != test_ctx->cfg)
  93     GNUNET_CONFIGURATION_destroy (test_ctx->cfg);
  94   if (NULL != test_ctx->system)
  95     GNUNET_TESTING_system_destroy (test_ctx->system, GNUNET_YES);
  96   GNUNET_free (test_ctx);
  97
  98 }
  99
 100
 101 /**
 102  * Task for shutdown
 103  *
 104  * @param cls the testing context
 105  */
 106 static void
 107 do_shutdown (void *cls);
 108
 109
 110 static void
 111 peer_status_cb (void *cls, struct GNUNET_TESTING_Peer *peer, int success)
 112 {
 113   struct TestingContext *test_ctx = cls;
 114
 115   switch (test_ctx->state)
 116   {
 117   case PEER_INIT:
 118     FAIL_TEST (0);
 119     break;
 120   case PEER_STARTED:
 121     FAIL_TEST (GNUNET_YES == success);
 122     test_ctx->state = PEER_STOPPED;
 123     GNUNET_SCHEDULER_add_now (&do_shutdown2, cls);
 124     break;
 125   case PEER_STOPPED:
 126     FAIL_TEST (0);
 127   }
 128 }
 129
 130
 131 /**
 132  * Task for shutdown
 133  *
 134  * @param cls the testing context
 135  */
 136 static void
 137 do_shutdown (void *cls)
 138 {
 139   struct TestingContext *test_ctx = cls;
 140
 141   GNUNET_assert (NULL != test_ctx);
 142   if (NULL != test_ctx->peer)
 143   {
 144     FAIL_TEST (GNUNET_OK ==
 145                GNUNET_TESTING_peer_stop_async (test_ctx->peer,
 146                                                &peer_status_cb,
 147                                                test_ctx));
 148   }
 149   else
 150     do_shutdown2 (test_ctx);
 151 }
 152
 153
 154 /**
 155  * Main point of test execution
 156  */
 157 static void
 158 run (void *cls, char *const *args, const char *cfgfile,
 159      const struct GNUNET_CONFIGURATION_Handle *cfg)
 160 {
 161   struct TestingContext *test_ctx;
 162   char *emsg;
 163   struct GNUNET_PeerIdentity id;
 164
 165   test_ctx = GNUNET_new (struct TestingContext);
 166   test_ctx->system =
 167       GNUNET_TESTING_system_create ("test-gnunet-testing",
 168                                     "127.0.0.1", NULL, NULL);
 169   emsg = NULL;
 170   if (NULL == test_ctx->system)
 171     goto end;
 172   test_ctx->cfg = GNUNET_CONFIGURATION_dup (cfg);
 173   test_ctx->peer =
 174       GNUNET_TESTING_peer_configure (test_ctx->system,
 175                                      test_ctx->cfg,
 176                                      0, &id, &emsg);
 177   if (NULL == test_ctx->peer)
 178   {
 179     if (NULL != emsg)
 180       printf ("Test failed upon error: %s", emsg);
 181     goto end;
 182   }
 183   if (GNUNET_OK != GNUNET_TESTING_peer_start (test_ctx->peer))
 184     goto end;
 185   test_ctx->state = PEER_STARTED;
 186   FAIL_TEST (GNUNET_OK ==
 187              GNUNET_TESTING_peer_stop_async (test_ctx->peer,
 188                                              &peer_status_cb,
 189                                              test_ctx));
 190   GNUNET_TESTING_peer_stop_async_cancel (test_ctx->peer);
 191   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
 192                                 &do_shutdown, test_ctx);
 193   return;
 194
 195  end:
 196   FAIL_TEST (0);
 197   GNUNET_SCHEDULER_add_now (&do_shutdown, test_ctx);
 198   GNUNET_free_non_null (emsg);
 199 }
 200
 201
 202 int main (int argc, char *argv[])
 203 {
 204   struct GNUNET_GETOPT_CommandLineOption options[] = {
 205     GNUNET_GETOPT_OPTION_END
 206   };
 207
 208   status = GNUNET_OK;
 209   if (GNUNET_OK !=
 210       GNUNET_PROGRAM_run (argc, argv,
 211                           "test_testing_new_peerstartup",
 212                           "test case for peerstartup using new testing library",
 213                           options, &run, NULL))
 214     return 1;
 215   return (GNUNET_OK == status) ? 0 : 1;
 216 }
 217
 218 /* end of test_testing_peerstartup.c */