src/integration-tests/test_integration_restart.py.in

   1 #!@PYTHON@
   2 #    This file is part of GNUnet.
   3 #    (C) 2010 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 2, 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 import sys
  22 import os
  23 import subprocess
  24 import re
  25 import shutil
  26 import time
  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 *
  32
  33
  34 #
  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
  38 #
  39 # Conditions for successful exit:
  40 # Both peers have 1 connected peer in transport, core, topology, fs after restart
  41
  42 #definitions
  43
  44
  45 testname = "test_integration_restart"
  46 verbose = False
  47 check_timeout = 180
  48
  49 if os.name == "nt":
  50   tmp = os.getenv ("TEMP")
  51 else:
  52   tmp = "/tmp"
  53
  54 def cleanup ():
  55     retries = 10
  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)
  60         time.sleep (1)
  61         retries -= 1
  62     if (os.path.exists(path)):
  63         test.p ("Failed to remove " + path)
  64
  65
  66     retries = 10
  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)
  71         time.sleep (1)
  72         retries -= 1
  73     if (os.path.exists(path)):
  74         test.p ("Failed to remove " + path)
  75
  76 def success_restart_cont (check):
  77         global success
  78         test.p ('Shutting down client & server')
  79         server.stop ()
  80         client.stop ()
  81         success = True;
  82
  83
  84 def fail_restart_cont (check):
  85         global success
  86         success = False;
  87         check.evaluate(True)
  88
  89
  90 def success_connect_cont (check):
  91         test.p ('Shutting down client & server for restart')
  92         server.stop ()
  93         client.stop ()
  94
  95         time.sleep(5)
  96
  97         test.p ('Restarting client & server')
  98         server.start ()
  99         client.start ()
 100
 101         check = Check (test)
 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))
 107
 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))
 113
 114         check.run_blocking (check_timeout, success_restart_cont, fail_restart_cont)
 115
 116
 117 def fail_connect_cont (check):
 118     global success
 119     success= False;
 120     check.evaluate(True)
 121
 122
 123 def check_connect ():
 124         check = Check (test)
 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))
 130
 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))
 136
 137         check.run_blocking (check_timeout, success_connect_cont, fail_connect_cont)
 138
 139 #
 140 # Test execution
 141 #
 142
 143 def run ():
 144         global success
 145         global test
 146         global server
 147         global client
 148
 149         success = False
 150
 151         test = Test ('test_integration_disconnect', verbose)
 152         cleanup ()
 153         server = Peer(test, './confs/c_bootstrap_server.conf');
 154         server.start();
 155
 156         client = Peer(test, './confs/c_no_nat_client.conf');
 157         client.start();
 158
 159
 160         if ((client.started == True) and (server.started == True)):
 161             test.p ('Peers started, running check')
 162             check_connect ()
 163
 164         server.stop ()
 165         client.stop ()
 166
 167         cleanup ()
 168
 169         if (success == False):
 170                 print ('Test failed')
 171                 return True
 172         else:
 173                 return False
 174
 175
 176 try:
 177     run ()
 178 except (KeyboardInterrupt, SystemExit):
 179     print 'Test interrupted'
 180     server.stop ()
 181     client.stop ()
 182     cleanup ()
 183 if (success == False):
 184         sys.exit(1)
 185 else:
 186         sys.exit(0)
 187