src/integration-tests/test_integration_disconnect.py.in

   1 #!@PYTHON@
   2 #    This file is part of GNUnet.
   3 #    (C) 2010, 2017 Christian Grothoff (and other contributing authors)
   4 #
   5 #    GNUnet is free software: you can redistribute it and/or modify it
   6 #    under the terms of the GNU 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 #
  16 import sys
  17 import signal
  18 import os
  19 import subprocess
  20 import re
  21 import shutil
  22 import time
  23 from gnunet_testing import Peer
  24 from gnunet_testing import Test
  25 from gnunet_testing import Check
  26 from gnunet_testing import Condition
  27 from gnunet_testing import *
  28
  29
  30 #
  31 # This test tests if a fresh peer bootstraps from a hostlist server and then
  32 # successfully connects to the server. When both peers are connected
  33 # in transport, core, topology, fs, the server is shutdown
  34 #
  35 # Conditions for successful exit:
  36 # Client peer has 0 connected peer in transport, core, topology, dht, fs
  37
  38 #definitions
  39
  40 testname = "test_integration_disconnect"
  41 verbose = True
  42 check_timeout = 180
  43
  44 if os.name == "nt":
  45   tmp = os.getenv ("TEMP")
  46   signals = [signal.SIGTERM, signal.SIGINT]
  47 else:
  48   tmp = "/tmp"
  49   signals = [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]
  50
  51 def cleanup_onerror (function, path, excinfo):
  52   import stat
  53   if not os.path.exists (path):
  54     pass
  55   elif not os.access(path, os.W_OK):
  56     # Is the error an access error ?
  57     os.chmod (path, stat.S_IWUSR)
  58     function (path)
  59   else:
  60     raise
  61
  62 def cleanup ():
  63   shutil.rmtree (os.path.join (tmp, "c_bootstrap_server"), False, cleanup_onerror)
  64   shutil.rmtree (os.path.join (tmp, "c_no_nat_client"), False, cleanup_onerror)
  65
  66
  67 def success_disconnect_cont (check):
  68         print('Peers disconnected successfully')
  69         global success
  70         success = True;
  71
  72
  73 def fail_disconnect_cont (check):
  74         global success
  75         success = False;
  76         print('Peers failed to disconnect')
  77         check.evaluate(True)
  78
  79 def check_disconnect ():
  80   test.p ('Shutting down bootstrap server')
  81   server.stop ()
  82   check = Check (test)
  83   check.add (StatisticsCondition (client, 'transport', '# peers connected',0))
  84   check.add (StatisticsCondition (client, 'core', '# peers connected',0))
  85   check.add (StatisticsCondition (client, 'topology', '# peers connected',0))
  86   check.add (StatisticsCondition (client, 'dht', '# peers connected',0))
  87   check.add (StatisticsCondition (client, 'fs', '# peers connected',0))
  88   check.run_blocking (check_timeout, success_disconnect_cont, fail_disconnect_cont)
  89
  90
  91 def success_connect_cont (check):
  92         print('Peers connected successfully')
  93         check_disconnect ()
  94
  95
  96 def fail_connect_cont (check):
  97   global success
  98   success= False
  99   print('Peers failed to connected!')
 100   check.evaluate(True)
 101
 102
 103 def check_connect ():
 104   check = Check (test)
 105   check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
 106   check.add (StatisticsCondition (client, 'core', '# peers connected',1))
 107   check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
 108   check.add (StatisticsCondition (client, 'dht', '# peers connected',1))
 109   check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
 110
 111   check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
 112   check.add (StatisticsCondition (server, 'core', '# peers connected',1))
 113   check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
 114   check.add (StatisticsCondition (server, 'dht', '# peers connected',1))
 115   check.add (StatisticsCondition (server, 'fs', '# peers connected',1))
 116
 117   check.run_blocking (check_timeout, success_connect_cont, fail_connect_cont)
 118
 119 #
 120 # Test execution
 121 #
 122
 123 def SigHandler(signum = None, frame = None):
 124         global success
 125         global server
 126         global client
 127
 128         print('Test was aborted!')
 129         if (None != server):
 130                 server.stop ()
 131         if (None != client):
 132                 client.stop ()
 133         cleanup ()
 134         sys.exit(success)
 135
 136 def run ():
 137         global success
 138         global test
 139         global server
 140         global client
 141
 142         server = None
 143         client = None
 144         success = False
 145
 146         for sig in signals:
 147                 signal.signal(sig, SigHandler)
 148
 149         test = Test ('test_integration_bootstrap_and_connect.py', verbose)
 150         cleanup ()
 151
 152         server = Peer(test, './confs/c_bootstrap_server.conf');
 153         client = Peer(test, './confs/c_no_nat_client.conf');
 154
 155         if (True != server.start()):
 156                 print('Failed to start server')
 157                 if (None != server):
 158                         server.stop ()
 159                 cleanup ()
 160                 sys.exit(success)
 161
 162         # Give the server time to start
 163         time.sleep(5)
 164
 165         if (True != client.start()):
 166                 print('Failed to start client')
 167                 if (None != server):
 168                         server.stop ()
 169                 if (None != client):
 170                         client.stop ()
 171                 cleanup ()
 172                 sys.exit(success)
 173
 174         if ((client.started == True) and (server.started == True)):
 175                 test.p ('Peers started, running check')
 176                 time.sleep(5)
 177                 check_connect ()
 178         server.stop ()
 179         client.stop ()
 180
 181         cleanup ()
 182
 183         if (success == False):
 184                 print ('Test failed')
 185                 return False
 186         else:
 187                 return True
 188
 189 try:
 190         run ()
 191 except (KeyboardInterrupt, SystemExit):
 192         print('Test interrupted')
 193         server.stop ()
 194         client.stop ()
 195         cleanup ()
 196 if (success == False):
 197         sys.exit(1)
 198 else:
 199         sys.exit(0)
 200
 201
 202
 203