- htons => htonl
[oweals/gnunet.git] / src / integration-tests / test_integration_clique.py.in
index 892f809312ab114873b8bbb33423a5face211531..12f99a837cd3606f84cdd7fddec1a28f2b9090fa 100755 (executable)
 #    Boston, MA 02111-1307, USA.
 #
 # 
+#
+# This test starts 3 peers (nated, server, no nat)and expects bootstrap 
+# and a connected clique
+#
+# Conditions for successful exit:
+# Both peers have 2 connected peers in transport, core, topology, fs and dht 
+
 import sys
+import signal
 import os
 import subprocess
 import re
 import shutil
-
-
-testname = "test_integration_clique"
-verbose = True
-
-srcdir = "../.."
-gnunet_pyexpect_dir = os.path.join (srcdir, "contrib")
-if gnunet_pyexpect_dir not in sys.path:
-  sys.path.append (gnunet_pyexpect_dir)
-
-from gnunet_pyexpect import pexpect
-
-if os.name == 'posix':
-  gnunetarm = 'gnunet-arm'
-elif os.name == 'nt':
-  gnunetarm = 'gnunet-arm.exe'
-
+import time
+from gnunet_testing import Peer
+from gnunet_testing import Test
+from gnunet_testing import Check
+from gnunet_testing import Condition
+from gnunet_testing import * 
 if os.name == "nt":
-  shutil.rmtree (os.path.join (os.getenv ("TEMP"), testname), True)
+  tmp = os.getenv ("TEMP")
 else:
-  shutil.rmtree ("/tmp/" + testname, True)
-
-if verbose == True:
-       print "Running " + testname
-
+  tmp = "/tmp"
 
+#definitions
 
+testname = "test_integration_clique"
+verbose = True
+check_timeout = 180
+
+def cleanup_onerror (function, path, excinfo):
+  import stat
+  if not os.path.exists (path):
+    pass
+  elif not os.access(path, os.W_OK):
+    # Is the error an access error ?
+    os.chmod (path, stat.S_IWUSR)
+    function (path)
+  else:
+    raise
+
+def cleanup ():
+    retries = 10
+    path = os.path.join (tmp, "c_bootstrap_server")  
+    test.p ("Removing " + path)      
+    while ((os.path.exists(path)) and (retries > 0)):
+        shutil.rmtree ((path), False, cleanup_onerror)
+        time.sleep (1)
+        retries -= 1
+    if (os.path.exists(path)):
+        test.p ("Failed to remove " + path) 
+    retries = 10
+    path = os.path.join (tmp, "c_no_nat_client")  
+    test.p ("Removing " + path)      
+    while ((os.path.exists(path)) and (retries > 0)):
+        shutil.rmtree ((path), False, cleanup_onerror)
+        time.sleep (1)
+        retries -= 1
+    if (os.path.exists(path)):
+        test.p ("Failed to remove " + path)     
+        retries = 10
+    path = os.path.join (tmp, "c_nat_client")  
+    test.p ("Removing " + path)      
+    while ((os.path.exists(path)) and (retries > 0)):
+        shutil.rmtree ((path), False, cleanup_onerror)
+        time.sleep (1)
+        retries -= 1
+    if (os.path.exists(path)):
+        test.p ("Failed to remove " + path) 
+
+
+def success_cont (check):
+    global success 
+    success = True;
+    print 'Connected clique successfully'
+
+def fail_cont (check):    
+    global success 
+    success= False;
+    check.evaluate(True)
+    print 'Failed to connect clique'
+
+def check_connect ():
+  check = Check (test)
+  check.add (StatisticsCondition (client, 'transport', '# peers connected',2))
+  check.add (StatisticsCondition (client, 'core', '# neighbour entries allocated',2))  
+  check.add (StatisticsCondition (client, 'core', '# peers connected',2))
+  check.add (StatisticsCondition (client, 'topology', '# peers connected',2))
+  check.add (StatisticsCondition (client, 'dht', '# peers connected',2))
+  check.add (StatisticsCondition (client, 'fs', '# peers connected',2))
+
+  check.add (StatisticsCondition (client_nat, 'transport', '# peers connected',2))
+  check.add (StatisticsCondition (client_nat, 'core', '# neighbour entries allocated',2))  
+  check.add (StatisticsCondition (client_nat, 'core', '# peers connected',2))
+  check.add (StatisticsCondition (client_nat, 'topology', '# peers connected',2))
+  check.add (StatisticsCondition (client_nat, 'dht', '# peers connected',2))
+  check.add (StatisticsCondition (client_nat, 'fs', '# peers connected',2))
+  
+  check.add (StatisticsCondition (server, 'transport', '# peers connected',2))
+  check.add (StatisticsCondition (server, 'core', '# neighbour entries allocated',2))  
+  check.add (StatisticsCondition (server, 'core', '# peers connected',2))
+  check.add (StatisticsCondition (server, 'topology', '# peers connected',2))
+  check.add (StatisticsCondition (server, 'dht', '# peers connected',2))
+  check.add (StatisticsCondition (server, 'fs', '# peers connected',2))  
+  
+  check.run_blocking (check_timeout, success_cont, fail_cont)
 
-exit ()
-# dummy copied from fs
-arm = subprocess.Popen ([gnunetarm, '-sq', '-c', 'test_gnunet_fs_ns_data.conf'])
-arm.communicate ()
+# 
+# Test execution
+# 
 
+def SigHandler(signum = None, frame = None):
+       global success  
+       global server
+       global client
+       global client_nat    
+       
+       print 'Test was aborted!'
+       if (None != server):
+               server.stop ()
+       if (None != client):            
+               client.stop ()
+       if (None != client_nat):                
+               client_nat.stop ()              
+       cleanup ()
+       sys.exit(success)
+
+def run ():
+       global success
+       global test
+       global server
+       global client
+       global client_nat       
+       
+       success = False
+       server = None
+       client = None
+       client_nat = None
+       test = Test ('test_integration_clique', verbose)
+       cleanup ()
+       
+       server = Peer(test, './confs/c_bootstrap_server.conf');
+       if (True != server.start()):
+               print 'Failed to start server'
+               if (None != server):
+                       server.stop ()
+               cleanup ()
+               sys.exit(success)
+       
+       # Server has to settle down
+       time.sleep(5)
+       
+       client = Peer(test, './confs/c_no_nat_client.conf');
+       if (True != client.start()):
+               print 'Failed to start client'
+               if (None != server):
+                       server.stop ()
+               if (None != client):
+                       client.stop ()                  
+               cleanup ()
+               sys.exit(success)
+       
+       # Server has to settle down
+       time.sleep(5)
+       
+       
+       client_nat = Peer(test, './confs/c_nat_client.conf');
+       if (True != client_nat.start()):
+               print 'Failed to start client_nat'
+               if (None != server):
+                       server.stop ()
+               if (None != client):
+                       client.stop ()
+               if (None != client_nat):
+                       client_nat.stop ()                                                              
+               cleanup ()
+               sys.exit(success)       
+       
+       if ((client.started == True) and (client_nat.started == True) and (server.started == True)):
+           test.p ('Peers started, running check')
+           check_connect ()
+           
+       server.stop ()    
+       client.stop ()
+       client_nat.stop ()
+       
+       cleanup ()
+       
+       if (success == False):
+               print ('Test failed')
+               return False 
+       else:
+               return True
+
+    
 try:
-  pseu = pexpect ()
-  pseu.spawn (None, [pseudonym, '-c', 'test_gnunet_fs_ns_data.conf', '-C', 'licenses', '-k', 'gplad', '-m', 'description:Free Software Licenses', '-R', 'myroot'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-  pseu.spawn (None, [pseudonym, '-c', 'test_gnunet_fs_ns_data.conf', '-o'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-  pseu.expect ("stdout", re.compile (r"licenses (.*)\r?\n"))
-
-  pub = pexpect ()
-  pub.spawn (None, [publish, '-c', 'test_gnunet_fs_ns_data.conf', '-k', 'licenses', '-P', 'licenses', '-u', 'gnunet://fs/chk/PC0M19QMQC0BPSHR6BGA228PP6INER1D610MGEMOMEM87222FN8HVUO7PQGO0O9HD2GVLHF2N5IDHEQUNK6LKE428FPO96SKQEA486O.PG7K85JGQ6N599MD5HEP3CHEVFPKQD9JB6NPSLVA3T1SKDS66CFI499VS6MGQ88B0QUAVT1282TCRD4GGFVUKDLGI8F0SPIANA3J2LG.35147', '-t', 'gpl', '-N', 'gpl3'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-
-  s = pexpect ()
-  s.spawn (None, [search, '-V', '-t', '1000', '-N', '1', '-c', 'test_gnunet_fs_ns_data.conf', 'gplad'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-  s.expect ("stdout", re.compile (r'#0:\r?\n'))
-  s.expect ("stdout", re.compile (r'gnunet-download gnunet://fs/sks/.*/myroot\r?\n'))
-  s.expect ("stdout", re.compile (r'\s*description: Free Software Licenses\r?\n'))
-
-  pseu = pexpect ()
-  pseu.spawn (None, [pseudonym, '-c', 'test_gnunet_fs_ns_data.conf'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-  pseu.expect ("stdout", re.compile (r'Free Software Licenses.*:\r?\n'))
+    run ()
+except (KeyboardInterrupt, SystemExit):    
+    print 'Test interrupted'
+    server.stop ()
+    client.stop ()
+    client_nat.stop ()
+    cleanup ()
+if (success == False):
+       sys.exit(1)   
+else:
+       sys.exit(0)    
+           
 
-finally:
-  arm = subprocess.Popen ([gnunetarm, '-eq', '-c', 'test_gnunet_fs_ns_data.conf'])
-  arm.communicate ()
-  if os.name == "nt":
-    shutil.rmtree (os.path.join (os.getenv ("TEMP"), "gnunet-test-fs-py-ns"), True)
-  else:
-    shutil.rmtree ("/tmp/gnunet-test-fs-py-ns", True)