5 from TestSsh import TestSsh
9 def __init__ (self,test_plc,test_node,test_slice):
10 self.test_plc=test_plc
11 self.test_node=test_node
12 self.test_slice=test_slice
13 self.test_ssh =TestSsh(self.test_plc.test_ssh)
16 return self.test_ssh.is_local()
18 def host_to_guest(self,command):
19 return self.test_plc.host_to_guest(self.test_plc.vserver,self.test_plc.vservername,command)
21 def get_privateKey(self,slice_spec):
23 (found,remote_privatekey)=self.test_slice.locate_key(slice_spec)
24 return (found,remote_privatekey)
28 def get_initscript(self,slice_spec):
29 (found,remote_privatekey)=self.get_privateKey(slice_spec)
31 raise Exception,"Cannot find a valid key for slice %s"%self.test_slice.name()
32 for hostname in slice_spec['nodenames']:
33 utils.header("Checking initiscript %s on the slice %s@%s"
34 %(slice_spec['initscriptname'],self.test_slice.name(),hostname))
35 init_file=self.test_plc.run_in_guest('ssh -i %s %s@%s ls -l /tmp/init* '
36 %(remote_privatekey,self.test_slice.name(),hostname))
42 def run_tcpcheck(self,peer_spec,remote_privatekey):
43 if peer_spec['peer_name']=="server":
44 tcp_command="ssh -i %s %s@%s ./tcptest.py server -t 10"%(remote_privatekey, peer_spec['slice_name'],
45 peer_spec['server_name'])
46 return self.test_plc.run_in_guest(tcp_command)
49 tcp_command="ssh -i %s %s@%s ./tcptest.py client -a %s -p %d"%(remote_privatekey, peer_spec['slice_name'],
50 peer_spec['client_name'],peer_spec['peer_server'],
51 peer_spec['server_port'])
52 return self.test_plc.run_in_guest(tcp_command)
56 def do_check_tcp(self,tcp_param,options):
57 for tcp_spec in tcp_param:
58 #copy the tcptest file under the chroot
59 localfile=remotefile="tcptest.py"
60 self.test_plc.copy_in_guest(localfile, remotefile, False)
61 peer_param=tcp_spec['tcp_fields']
62 if (tcp_spec['tcp_fields']['peer_name']=='server'):
64 utils.header("Transfert the tcp script to the server at %s@%s"%(peer_param['slice_name'],
65 peer_param['server_name']))
66 slice_spec=self.test_slice.get_slice(peer_param['slice_name'])
67 (found,remote_privatekey)=self.get_privateKey(slice_spec)
68 cp_server_command="scp -i %s ./tcptest.py %s@%s:"%(remote_privatekey,peer_param['slice_name'],
69 peer_param['server_name'])
70 self.test_plc.run_in_guest(cp_server_command)
71 serv_status=self.run_tcpcheck(peer_param,remote_privatekey)
73 utils.header("FAILED to check loop Connexion on the %s server side"%peer_param['server_name'])
77 utils.header("Transfert the tcp script to the client at %s@%s" %(peer_param['slice_name'],
78 peer_param['client_name']))
79 slice_spec=self.test_slice.get_slice(peer_param['slice_name'])
80 (found,remote_privatekey)=self.get_privateKey(slice_spec)
81 cp_client_command="scp -i %s ./tcptest.py %s@%s:"%(remote_privatekey, peer_param['slice_name'],
82 peer_param['client_name'])
83 self.test_plc.run_in_guest(cp_client_command)
84 client_status=self.run_tcpcheck(peer_param,remote_privatekey)
86 utils.header("FAILED to Contact the server %s from the client side %s"%(peer_param['peer_server'],
87 peer_param['client_name']))
91 self.test_plc.run_in_guest("rm -rf tcptest.py")