improved test framework
[oweals/gnunet.git] / src / integration-tests / test_integration_bootstrap_and_connect.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 import pexpect
28 from gnunet_testing import Peer
29 from gnunet_testing import Test
30 from gnunet_testing import Check
31 from gnunet_testing import Condition
32 from gnunet_testing import * 
33  
34
35 #
36 # This test tests if a fresh peer bootstraps from a hostlist server and then
37 # successfully connects to the server 
38 #
39 # Conditions for successful exit:
40 # Both peers have 1 connected peer in transport, core, topology, fs
41
42 #definitions
43
44 testname = "test_integration_bootstrap_and_connect"
45 verbose = True
46 success = False
47 timeout = 100
48
49
50 def cleanup ():
51         if os.name == "nt":
52             shutil.rmtree (os.path.join (os.getenv ("TEMP"), "gnunet-test-fs-py-ns"), True)
53             shutil.rmtree (os.path.join (os.getenv ("TEMP"), "c_no_nat_client"), True)
54         else:
55             shutil.rmtree ("/tmp/c_bootstrap_server/", True)
56             shutil.rmtree ("/tmp/c_no_nat_client/", True)    
57
58 def success ():
59     global success
60     success = True;
61 def fail ():    
62     global success 
63     success= False;
64
65 def check ():
66   
67   check = Check ()
68   check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
69   check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',1))  
70   check.add (StatisticsCondition (client, 'core', '# entries in session map',1))
71   check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
72   check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
73   
74
75   check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
76   check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',1))  
77   check.add (StatisticsCondition (server, 'core', '# entries in session map',1))
78   check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
79   check.add (StatisticsCondition (server, 'fs', '# peers connected',1))  
80   
81   check.run_blocking (10, success, fail)
82
83
84 # Test execution
85
86
87 test = Test ('test_integration_bootstrap_and_connect.py', verbose)
88
89 server = Peer(test, './confs/c_bootstrap_server.conf');
90 server.start();
91
92 client = Peer(test, './confs/c_no_nat_client.conf');
93 client.start();
94
95 if ((client.started == True) and (server.started == True)):
96     check ()
97         
98 server.stop ()
99 client.stop ()
100
101 cleanup ()
102
103 if (success == False):
104         print ('Test failed')
105         exit (1)
106 else:
107         exit (0)