+# Thierry Parmentelat <thierry.parmentelat@inria.fr>
+# Copyright (C) 2010 INRIA
+#
import utils
import os, os.path
import datetime
import time
-
+from TestSsh import TestSsh
class TestSliver:
self.test_plc=test_plc
self.test_node=test_node
self.test_slice=test_slice
-
- def get_privateKey(self,slice_spec):
+ self.test_ssh = self.create_test_ssh()
+
+ def get_privateKey(self):
+ slice_spec=self.test_slice.slice_spec
try:
- (found,remote_privatekey)=self.test_slice.locate_key(slice_spec)
- return (found,remote_privatekey)
+ (found,privatekey)=self.test_slice.locate_key()
+ return (found,privatekey)
except Exception,e:
print str(e)
- def get_initscript(self,slice_spec):
- (found,remote_privatekey)=self.get_privateKey(slice_spec)
- if not found :
- raise Exception,"Cannot find a valid key for slice %s"%self.test_slice.name()
- for hostname in slice_spec['nodenames']:
- utils.header("Checking initiscript %s on the slice %s@%s"
- %(slice_spec['initscriptname'],self.test_slice.name(),hostname))
- init_file=self.test_plc.run_in_guest('ssh -i %s %s@%s ls -l /tmp/init* '
- %(remote_privatekey,self.test_slice.name(),hostname))
- if ( init_file):
- return False
-
- return True
-
- def run_tcpcheck(self,peer_spec,remote_privatekey):
- if peer_spec['peer_name']=="server":
- tcp_command="ssh -i %s %s@%s ./tcptest.py server -t 10"%(remote_privatekey, peer_spec['slice_name'],
- peer_spec['server_name'])
- return self.test_plc.run_in_guest(tcp_command)
-
- else:
- tcp_command="ssh -i %s %s@%s ./tcptest.py client -a %s -p %d"%(remote_privatekey, peer_spec['slice_name'],
- peer_spec['client_name'],peer_spec['peer_server'],
- peer_spec['server_port'])
- return self.test_plc.run_in_guest(tcp_command)
+ def create_test_ssh(self):
+ private_key = self.test_slice.locate_private_key()
+ if not private_key:
+ raise Exception,"Cannot find the private key for slice %s"%self.test_slice.name()
+ return TestSsh (self.test_node.name(),key=private_key,username=self.test_slice.name(),
+ # so that copies end up in the home dir
+ buildname=".")
+ def name(self):
+ return "%s@%s"%(self.test_slice.name(),self.test_node.name())
+ def check_initscript_stamp(self, stamp):
+ utils.header("Checking for initscript stamp %s on sliver %s"%(stamp, self.name()))
+ return self.test_ssh.run("ls -l /var/tmp/%s.stamp"%stamp)==0
+
+ def check_tcp_ready (self, port):
+ ready_command = "./tcptest.py ready -p %d"%(port)
+ return self.test_ssh.copy("tcptest.py") == 0 and \
+ self.test_ssh.run(ready_command) == 0
- def do_check_tcp(self,tcp_param,options):
- for tcp_spec in tcp_param:
- #copy the tcptest file under the chroot
- localfile=remotefile="tcptest.py"
- self.test_plc.copy_in_guest(localfile, remotefile, False)
- peer_param=tcp_spec['tcp_fields']
- if (tcp_spec['tcp_fields']['peer_name']=='server'):
- #server instruction
- utils.header("Transfert the tcp script to the server at %s@%s"%(peer_param['slice_name'],
- peer_param['server_name']))
- slice_spec=self.test_slice.get_slice(peer_param['slice_name'])
- (found,remote_privatekey)=self.get_privateKey(slice_spec)
- cp_server_command="scp -i %s ./tcptest.py %s@%s:"%(remote_privatekey,peer_param['slice_name'],
- peer_param['server_name'])
- self.test_plc.run_in_guest(cp_server_command)
- serv_status=self.run_tcpcheck(peer_param,remote_privatekey)
- if (serv_status):
- utils.header("FAILED to check loop Connexion on the %s server side"%peer_param['server_name'])
- return False
- else:
- #Client instruction
- utils.header("Transfert the tcp script to the client at %s@%s" %(peer_param['slice_name'],
- peer_param['client_name']))
- slice_spec=self.test_slice.get_slice(peer_param['slice_name'])
- (found,remote_privatekey)=self.get_privateKey(slice_spec)
- cp_client_command="scp -i %s ./tcptest.py %s@%s:"%(remote_privatekey, peer_param['slice_name'],
- peer_param['client_name'])
- self.test_plc.run_in_guest(cp_client_command)
- client_status=self.run_tcpcheck(peer_param,remote_privatekey)
- if ( serv_status):
- utils.header("FAILED to Contact the server %s from the client side %s"%(peer_param['peer_server'],
- peer_param['client_name']))
- return False
-
+ def run_tcp_server (self, port, timeout=10):
+ server_command = "./tcptest.py server -p %d -t %d"%(port, timeout)
+ return self.test_ssh.copy("tcptest.py") == 0 and \
+ self.test_ssh.run(server_command, background=True)==0
- self.test_plc.run_in_guest("rm -rf tcptest.py")
- return True
+ def run_tcp_client (self, servername, port, retry=5):
+ client_command="./tcptest.py client -a %s -p %d"%(servername, port)
+ if self.test_ssh.copy("tcptest.py") != 0:
+ return False
+ if self.test_ssh.run(client_command) == 0:
+ return True
+ return False
+ # use the node's main ssh root entrance, as the slice entrance might be down
+ #def tar_var_logs (self):
+ # return self.test_ssh.actual_command("sudo tar -C /var/log -cf - .")
+ def tar_var_logs (self):
+ test_ssh=self.test_node.create_test_ssh()
+ dir_to_tar="/vservers/%s/var/log"%self.test_slice.name()
+ return test_ssh.actual_command("tar -C %s -cf - ."%dir_to_tar)
+
+ def check_hooks (self):
+ print 'NOTE: slice hooks check scripts NOT (yet?) run in sudo'
+ extensions = [ 'py','pl','sh' ]
+ path='hooks/slice/'
+ scripts=utils.locate_hooks_scripts ('sliver '+self.name(), path,extensions)
+ overall = True
+ for script in scripts:
+ if not self.check_hooks_script (script):
+ overall = False
+ return overall
+ def check_hooks_script (self,local_script):
+ script_name=os.path.basename(local_script)
+ utils.header ("SLIVER hook %s (%s)"%(script_name,self.name()))
+ test_ssh=self.create_test_ssh()
+ test_ssh.copy_home(local_script)
+ if test_ssh.run("./"+script_name) != 0:
+ utils.header ("WARNING: hooks check script %s FAILED (ignored)"%script_name)
+ #return False
+ return True
+ else:
+ utils.header ("SUCCESS: sliver hook %s OK"%script_name)
+ return True
+
git://git.onelab.eu / tests.git / blobdiff