improved tests
[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 = False
46 timeout = 100
47
48
49 def cleanup ():
50         if os.name == "nt":
51             shutil.rmtree (os.path.join (os.getenv ("TEMP"), "gnunet-test-fs-py-ns"), True)
52             shutil.rmtree (os.path.join (os.getenv ("TEMP"), "c_no_nat_client"), True)
53         else:
54             shutil.rmtree ("/tmp/c_bootstrap_server/", True)
55             shutil.rmtree ("/tmp/c_no_nat_client/", True)    
56
57 def success_cont (check):
58     global success
59     success = True;
60     
61 def fail_cont (check):    
62     global success 
63     success= False;
64     check.eval(True)
65
66 def check ():
67   
68   check = Check (test)
69   check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
70   check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',1))  
71   check.add (StatisticsCondition (client, 'core', '# entries in session map',1))
72   check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
73   check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
74   
75
76   check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
77   check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',1))  
78   check.add (StatisticsCondition (server, 'core', '# entries in session map',1))
79   check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
80   check.add (StatisticsCondition (server, 'fs', '# peers connected',1))  
81   
82   check.run_blocking (10, success_cont, fail_cont)
83
84
85 # Test execution
86
87 success = False
88
89 test = Test ('test_integration_bootstrap_and_connect.py', verbose)
90
91 server = Peer(test, './confs/c_bootstrap_server.conf');
92 server.start();
93
94 client = Peer(test, './confs/c_no_nat_client.conf');
95 client.start();
96
97 if ((client.started == True) and (server.started == True)):
98     test.p ('Peers started, running check')
99     check ()
100         
101 server.stop ()
102 client.stop ()
103
104 cleanup ()
105
106 if (success == False):
107         print ('Test failed')
108         exit (1)
109 else:
110         exit (0)