2 # This file is part of GNUnet.
3 # (C) 2010 Christian Grothoff (and other contributing authors)
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.
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.
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.
20 # Functions for integration testing
26 from gnunet_pyexpect import pexpect
29 def __init__(self, test):
30 self.fulfilled = False
31 self.conditions = list()
33 def add (self, condition):
34 self.conditions.append(condition)
39 for c in self.conditions:
40 if (False == c.check ()):
46 def run_blocking (self, timeout, pos_cont, neg_cont):
49 while ((False == res) and (execs < timeout)):
53 if ((False == res) and (execs >= timeout)):
54 print ('Check had timeout after ' +str(timeout)+ ' seconds')
56 elif ((False == res) and (execs >= timeout)):
61 def run_once (self, pos_cont, neg_cont):
65 if ((res == False) and (neg_cont != None)):
67 if ((res == True) and (pos_cont != None)):
70 def evaluate (self, failed_only):
73 for c in self.conditions:
74 if (False == c.evaluate (failed_only)):
78 print (str(pos) +' out of '+ str (pos+neg) + ' conditions fulfilled')
81 self.fulfilled = False
82 for c in self.conditions:
87 self.fulfilled = False
89 def __init__(self, type):
90 self.fulfilled = False
94 def evaluate (self, failed_only):
95 if ((self.fulfilled == False) and (failed_only == True)):
96 print str(self.type) + 'condition for was ' + str(self.fulfilled)
97 elif (failed_only == False):
98 print str(self.type) + 'condition for was ' + str(self.fulfilled)
101 class FileExistCondition (Condition):
102 def __init__(self, file):
103 self.fulfilled = False
107 if (self.fulfilled == False):
108 res = os.path.isfile(self.file)
110 self.fulfilled = True
116 def evaluate (self, failed_only):
117 if ((self.fulfilled == False) and (failed_only == True)):
118 print str(self.type) + 'condition for file '+self.file+' was ' + str(self.fulfilled)
119 elif (failed_only == False):
120 print str(self.type) + 'condition for file '+self.file+' was ' + str(self.fulfilled)
121 return self.fulfilled
123 class StatisticsCondition (Condition):
124 def __init__(self, peer, subsystem, name, value):
125 self.fulfilled = False
126 self.type = 'statistics'
128 self.subsystem = subsystem;
133 if (self.fulfilled == False):
134 self.result = self.peer.get_statistics_value (self.subsystem, self.name)
135 if (str(self.result) == str(self.value)):
136 self.fulfilled = True
142 def evaluate (self, failed_only):
143 if (self.result == -1):
146 res = str(self.result)
147 if (self.fulfilled == False):
153 if (((self.fulfilled == False) and (failed_only == True)) or (failed_only == False)):
154 print self.peer.id[:4] + " " +self.peer.cfg + " " + str(self.type) + ' condition in subsystem "' + self.subsystem.ljust(12) +'" : "' + self.name.ljust(30) +'" : (expected/real value) ' + str(self.value) + op + res + fail
155 return self.fulfilled
157 # Specify two statistic values and check if they are equal
158 class EqualStatisticsCondition (Condition):
159 def __init__(self, peer, subsystem, name, peer2, subsystem2, name2):
160 self.fulfilled = False
161 self.type = 'equalstatistics'
163 self.subsystem = subsystem;
167 self.subsystem2 = subsystem2;
171 if (self.fulfilled == False):
172 self.result = self.peer.get_statistics_value (self.subsystem, self.name);
173 self.result2 = self.peer2.get_statistics_value (self.subsystem2, self.name2);
174 if (str(self.result) == str(self.result2)):
175 self.fulfilled = True
181 def evaluate (self, failed_only):
182 if (self.result == -1):
185 res = str(self.result)
186 if (self.result2 == -1):
189 res2 = str(self.result2)
190 if (self.fulfilled == False):
196 if (((self.fulfilled == False) and (failed_only == True)) or (failed_only == False)):
197 print self.peer.id[:4] + ' "' + self.subsystem.ljust(12) + '" "' + self.name.ljust(30) + '" == ' + str(self.result) +" " + self.peer2.id[:4] + ' "' + self.subsystem2.ljust(12) + '" '+ self.name2.ljust(30) + '" ' + str(self.result2)
198 return self.fulfilled
201 def __init__(self, testname, verbose):
203 self.verbose = verbose;
204 self.name = testname;
206 gnunet_pyexpect_dir = os.path.join (srcdir, "contrib")
207 if gnunet_pyexpect_dir not in sys.path:
208 sys.path.append (gnunet_pyexpect_dir)
210 self.gnunetstatistics = ''
211 if os.name == 'posix':
212 self.gnunetarm = 'gnunet-arm'
213 self.gnunetstatistics = 'gnunet-statistics'
214 self.gnunetpeerinfo = 'gnunet-peerinfo'
215 elif os.name == 'nt':
216 self.gnunetarm = 'gnunet-arm.exe'
217 self.gnunetstatistics = 'gnunet-statistics.exe'
218 self.gnunetpeerinfo = 'gnunet-peerinfo.exe'
220 shutil.rmtree (os.path.join (os.getenv ("TEMP"), testname), True)
222 shutil.rmtree ("/tmp/" + testname, True)
223 def add_peer (self, peer):
224 self.peers.append(peer)
226 if (self.verbose == True):
230 def __init__(self, test, cfg_file):
231 if (False == os.path.isfile(cfg_file)):
232 print ("Peer cfg " + cfg_file + ": FILE NOT FOUND")
238 if (self.started == True):
239 print 'ERROR! Peer using cfg ' + self.cfg + ' was not stopped'
242 print 'ERROR! Peer using cfg ' + self.cfg + ' could not be stopped'
248 self.test.p ("Starting peer using cfg " + self.cfg)
250 server = subprocess.Popen ([self.test.gnunetarm, '-sq', '-c', self.cfg])
251 server.communicate ()
253 print "Can not start peer"
260 server.spawn (None, [self.test.gnunetpeerinfo, '-c', self.cfg ,'-s'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
261 test = server.read("stdout", 1024)
263 print "Can not get peer identity"
264 test = (test.split('`')[1])
265 self.id = test.split('\'')[0]
268 if (self.started == False):
270 self.test.p ("Stopping peer using cfg " + self.cfg)
272 server = subprocess.Popen ([self.test.gnunetarm, '-eq', '-c', self.cfg])
273 server.communicate ()
275 print "Can not stop peer"
279 def get_statistics_value (self, subsystem, name):
281 server.spawn (None, [self.test.gnunetstatistics, '-c', self.cfg ,'-q','-n', name, '-s', subsystem ], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
282 #server.expect ("stdout", re.compile (r""))
283 test = server.read("stdout", 10240)
284 tests = test.partition('\n')
285 # On W32 GNUnet outputs with \r\n, rather than \n
286 if os.name == 'nt' and tests[1] == '\n' and tests[0][-1] == '\r':
287 tests = (tests[0][:-1], tests[1], tests[2])
289 if (tests.isdigit() == True):