src/integration-tests/test_integration_bootstrap_and_connect_and_disconnect_nat.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
  37 #
  38 # Conditions for successful exit:
  39 # Both peers have 1 connected peer in transport, core, topology, fs
  40
  41 #
  42 # This test tests if a fresh peer bootstraps from a hostlist server and then
  43 # successfully connects to the server
  44 #
  45 # Conditions for successful exit:
  46 # Both peers have 1 connected peer in transport, core, topology, fs
  47
  48 #definitions
  49
  50 testname = "test_integration_bootstrap_and_connect"
  51 verbose = False
  52 check_timeout = 180
  53
  54 if os.name == "nt":
  55   tmp = os.getenv ("TEMP")
  56 else:
  57   tmp = "/tmp"
  58
  59 def cleanup ():
  60     retries = 10
  61     path = os.path.join (tmp, "c_bootstrap_server")
  62     test.p ("Removing " + path)
  63     while ((os.path.exists(path)) and (retries > 0)):
  64         shutil.rmtree ((path), False)
  65         time.sleep (1)
  66         retries -= 1
  67     if (os.path.exists(path)):
  68         test.p ("Failed to remove " + path)
  69
  70
  71     retries = 10
  72     path = os.path.join (tmp, "c_nat_client")
  73     test.p ("Removing " + path)
  74     while ((os.path.exists(path)) and (retries > 0)):
  75         shutil.rmtree ((path), False)
  76         time.sleep (1)
  77         retries -= 1
  78     if (os.path.exists(path)):
  79         test.p ("Failed to remove " + path)
  80
  81 def success_server_stop_cont (check):
  82     global success
  83     success = True;
  84
  85 def success_cont (check):
  86         server.stop()
  87
  88         check = Check (test)
  89         check.add (StatisticsCondition (client, 'transport', '# peers connected',0))
  90         check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',0))
  91         check.add (StatisticsCondition (client, 'core', '# peers connected',0))
  92         check.add (StatisticsCondition (client, 'topology', '# peers connected',0))
  93         check.add (StatisticsCondition (client, 'fs', '# peers connected',0))
  94
  95         check.run_blocking (check_timeout, success_server_stop_cont, fail_cont)
  96
  97 def fail_cont (check):
  98     global success
  99     success = False;
 100     check.evaluate(True)
 101
 102 def check ():
 103   check = Check (test)
 104   check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
 105   check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',1))
 106   check.add (StatisticsCondition (client, 'core', '# peers connected',1))
 107   check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
 108   check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
 109
 110
 111   check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
 112   check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',1))
 113   check.add (StatisticsCondition (server, 'core', '# peers connected',1))
 114   check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
 115   check.add (StatisticsCondition (server, 'fs', '# peers connected',1))
 116
 117   check.run_blocking (check_timeout, success_cont, fail_cont)
 118
 119 #
 120 # Test execution
 121 #
 122
 123 def run ():
 124         global success
 125         global test
 126         global server
 127         global client
 128
 129         success = False
 130         test = Test ('test_integration_bootstrap_and_connect.py', verbose)
 131
 132         server = Peer(test, './confs/c_bootstrap_server.conf');
 133         client = Peer(test, './confs/c_nat_client.conf');
 134
 135         assert (True == server.start());
 136         assert (True == client.start());
 137
 138         if ((client.started == True) and (server.started == True)):
 139                 test.p ('Peers started, running check')
 140                 time.sleep(5)
 141                 check ()
 142         server.stop ()
 143         client.stop ()
 144
 145         cleanup ()
 146
 147         if (success == False):
 148                 print ('Test failed')
 149                 return False
 150         else:
 151                 return True
 152
 153 try:
 154         run ()
 155 except (KeyboardInterrupt, SystemExit):
 156         print 'Test interrupted'
 157         server.stop ()
 158         client.stop ()
 159         cleanup ()
 160 if (success == False):
 161         sys.exit(1)
 162 else:
 163         sys.exit(0)
 164