2 # This file is part of GNUnet.
3 # (C) 2010 Christian Grothoff (and other contributing authors)
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 2, or (at your
8 # 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 # General Public License for more details.
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.
27 from gnunet_testing import Peer
28 from gnunet_testing import Test
29 from gnunet_testing import Check
30 from gnunet_testing import Condition
31 from gnunet_testing import *
35 # This test tests if a fresh peer bootstraps from a hostlist server and then
36 # successfully connects to the server. When both peers are connected
37 # in transport, core, topology, fs, botth peers are shutdown and restarted
39 # Conditions for successful exit:
40 # Both peers have 1 connected peer in transport, core, topology, fs after restart
45 testname = "test_integration_restart"
50 tmp = os.getenv ("TEMP")
56 path = os.path.join (tmp, "c_bootstrap_server")
57 test.p ("Removing " + path)
58 while ((os.path.exists(path)) and (retries > 0)):
59 shutil.rmtree ((path), False)
62 if (os.path.exists(path)):
63 test.p ("Failed to remove " + path)
67 path = os.path.join (tmp, "c_no_nat_client")
68 test.p ("Removing " + path)
69 while ((os.path.exists(path)) and (retries > 0)):
70 shutil.rmtree ((path), False)
73 if (os.path.exists(path)):
74 test.p ("Failed to remove " + path)
76 def success_restart_cont (check):
78 test.p ('Shutting down client & server')
84 def fail_restart_cont (check):
90 def success_connect_cont (check):
91 test.p ('Shutting down client & server for restart')
97 test.p ('Restarting client & server')
102 check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
103 check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',1))
104 check.add (StatisticsCondition (client, 'core', '# peers connected',1))
105 check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
106 check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
108 check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
109 check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',1))
110 check.add (StatisticsCondition (server, 'core', '# peers connected',1))
111 check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
112 check.add (StatisticsCondition (server, 'fs', '# peers connected',1))
114 check.run_blocking (check_timeout, success_restart_cont, fail_restart_cont)
117 def fail_connect_cont (check):
123 def check_connect ():
125 check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
126 check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',1))
127 check.add (StatisticsCondition (client, 'core', '# peers connected',1))
128 check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
129 check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
131 check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
132 check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',1))
133 check.add (StatisticsCondition (server, 'core', '# peers connected',1))
134 check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
135 check.add (StatisticsCondition (server, 'fs', '# peers connected',1))
137 check.run_blocking (check_timeout, success_connect_cont, fail_connect_cont)
151 test = Test ('test_integration_disconnect', verbose)
153 server = Peer(test, './confs/c_bootstrap_server.conf');
156 client = Peer(test, './confs/c_no_nat_client.conf');
160 if ((client.started == True) and (server.started == True)):
161 test.p ('Peers started, running check')
169 if (success == False):
170 print ('Test failed')
178 except (KeyboardInterrupt, SystemExit):
179 print 'Test interrupted'
183 if (success == False):