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 = self.create_test_ssh()
15 def get_privateKey(self):
16 slice_spec=self.test_slice.slice_spec
18 (found,privatekey)=self.test_slice.locate_key()
19 return (found,privatekey)
23 def create_test_ssh(self):
24 (found,privatekey) = self.get_privateKey()
26 raise Exception,"Cannot find a valid key for slice %s"%self.test_slice.name()
27 return TestSsh (self.test_node.name(),key=privatekey,username=self.test_slice.name(),
28 # so that copies end up in the home dir
32 return "%s@%s"%(self.test_slice.name(),self.test_node.name())
34 def check_initscript(self,initscript):
35 slice_spec=self.test_slice.slice_spec
36 initscript = slice_spec['initscriptname']
37 utils.header("Checking initscript %s on sliver %s"%(initscript,self.name()))
38 return self.test_ssh.run("ls -l /tmp/%s.stamp"%initscript)==0
40 def run_tcp_server (self,port,timeout=10):
41 server_command = "tcptest.py server -p %d -t %d"%(port,timeout)
42 return self.test_ssh.copy("tcptest.py")==0 and self.test_ssh.run(server_command)==0
44 def run_tcp_client (self,servername,port):
45 client_command="tcptest.py client -a %s -p %d"%(servername,port)
46 return self.test_ssh.copy("tcptest.py")==0 and self.test_ssh.run(client_command)==0
48 def tar_var_logs (self):
49 return self.test_ssh.actual_command("sudo tar -C /var/log -cf - .")