2 This file is part of GNUnet
3 Copyright (C) 2008--2013 GNUnet e.V.
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.
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.
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/>.
20 * @file testbed/test_testbed_api_testbed_run.c
21 * @brief Test cases for testing high-level testbed management
22 * @author Sree Harsha Totakura <sreeharsha@totakura.in>
26 #include "gnunet_util_lib.h"
27 #include "gnunet_testbed_service.h"
30 * Number of peers we want to start
35 * The array of peers; we fill this as the peers are given to us by the testbed
37 static struct GNUNET_TESTBED_Peer *peers[NUM_PEERS];
42 static struct GNUNET_TESTBED_Operation *op;
45 * Abort task identifier
47 static struct GNUNET_SCHEDULER_Task * abort_task;
52 static unsigned int peer_id;
60 * Should we wait forever after testbed is initialized?
62 static int wait_forever;
71 do_shutdown (void *cls)
73 if (NULL != abort_task)
74 GNUNET_SCHEDULER_cancel (abort_task);
75 GNUNET_SCHEDULER_shutdown (); /* Stop scheduler to shutdown testbed run */
80 * abort task to run on test timed out
88 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
89 "Test timed out -- Aborting\n");
90 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
95 * Signature of a main function for a testcase.
98 * @param h the run handle
99 * @param num_peers number of peers in 'peers'
100 * @param peers_ handle to peers run in the testbed
101 * @param links_succeeded the number of overlay link connection attempts that
103 * @param links_failed the number of overlay link connection attempts that
107 test_master (void *cls,
108 struct GNUNET_TESTBED_RunHandle *h,
109 unsigned int num_peers,
110 struct GNUNET_TESTBED_Peer **peers_,
111 unsigned int links_succeeded,
112 unsigned int links_failed)
115 if (GNUNET_YES == wait_forever)
117 if (NULL == abort_task)
118 return; /* abort already scheduled */
119 GNUNET_SCHEDULER_cancel (abort_task);
121 GNUNET_SCHEDULER_add_shutdown (&do_shutdown, NULL);
124 GNUNET_assert (NULL != peers[0]);
125 op = GNUNET_TESTBED_peer_stop (NULL, peers[0], NULL, NULL);
126 GNUNET_assert (NULL != op);
131 * Controller event callback
134 * @param event the controller event
137 controller_event_cb (void *cls,
138 const struct GNUNET_TESTBED_EventInformation *event)
143 case GNUNET_TESTBED_ET_PEER_START:
144 GNUNET_assert (NULL == peers[peer_id]);
145 GNUNET_assert (NULL != event->details.peer_start.peer);
146 peers[peer_id++] = event->details.peer_start.peer;
148 case GNUNET_TESTBED_ET_PEER_STOP:
149 GNUNET_assert (NULL != op);
150 GNUNET_TESTBED_operation_done (op);
151 GNUNET_assert (peers[0] == event->details.peer_stop.peer);
152 GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
164 * @param args arguments passed to GNUNET_PROGRAM_run
165 * @param cfgfile the path to configuration file
166 * @param cfg the configuration file handle
172 const struct GNUNET_CONFIGURATION_Handle *config)
177 event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_START);
178 event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_STOP);
179 GNUNET_TESTBED_run (NULL, config, NUM_PEERS, event_mask,
180 &controller_event_cb, NULL,
183 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
184 (GNUNET_TIME_UNIT_SECONDS, 300),
194 main (int argc, char **argv)
197 "test_testbed_api_testbed_run",
201 struct GNUNET_GETOPT_CommandLineOption options[] = {
202 GNUNET_GETOPT_OPTION_END
205 char *config_filename;
208 if (NULL == (testname = strrchr (argv[0], (int) '_')))
214 testname = GNUNET_strdup (testname);
219 /* check and remove .exe extension */
220 period = strrchr (testname, (int) '.');
224 GNUNET_break (0); /* Windows with no .exe? */
227 if (0 == strcmp ("waitforever", testname))
228 wait_forever = GNUNET_YES;
229 if ( (GNUNET_YES != wait_forever) && (0 != strcmp ("run", testname)) )
231 GNUNET_asprintf (&config_filename, "test_testbed_api_testbed_run_%s.conf",
235 config_filename = GNUNET_strdup ("test_testbed_api.conf");
236 GNUNET_free (testname);
237 argv2[2] = config_filename;
238 result = GNUNET_SYSERR;
240 GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1, argv2,
241 "test_testbed_api_testbed_run", "nohelp", options,
243 GNUNET_free (config_filename);
244 if ((GNUNET_OK != ret) || (GNUNET_OK != result))
249 /* end of test_testbed_api_testbed_run.c */