system/TestSliver.py

   1 import utils
   2 import os, os.path
   3 import datetime
   4 import time
   5 from TestSsh import TestSsh
   6
   7 class TestSliver:
   8
   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()
  14
  15     def get_privateKey(self):
  16         slice_spec=self.test_slice.slice_spec
  17         try:
  18             (found,privatekey)=self.test_slice.locate_key()
  19             return (found,privatekey)
  20         except Exception,e:
  21             print str(e)
  22
  23     def create_test_ssh(self):
  24         (found,privatekey) = self.get_privateKey()
  25         if not found:
  26             raise Exception,"Cannot find the private 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
  29                         buildname=".")
  30
  31     def name (self):
  32         return "%s@%s"%(self.test_slice.name(),self.test_node.name())
  33
  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
  39
  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 \
  43             self.test_ssh.run(server_command,background=True)==0
  44
  45     def run_tcp_client (self,servername,port):
  46         client_command="./tcptest.py client -a %s -p %d"%(servername,port)
  47         return self.test_ssh.copy("tcptest.py")==0 and \
  48             self.test_ssh.run(client_command,background=True)==0
  49
  50     def tar_var_logs (self):
  51         return self.test_ssh.actual_command("sudo tar -C /var/log -cf - .")
  52
  53     def check_sanity (self):
  54         print 'WARNING: slice sanity check scripts NOT (yet?) run in sudo'
  55         extensions = [ 'py','pl','sh' ]
  56         path='tests/qaapi/qa/tests/slice/'
  57         scripts=utils.locate_sanity_scripts ('sliver '+self.name(), path,extensions)
  58         overall = True
  59         for script in scripts:
  60             if not self.check_sanity_script (script):
  61                 overall = False
  62         return overall
  63
  64     def check_sanity_script (self,local_script):
  65         ssh_handle=self.create_test_ssh()
  66         ssh_handle.copy_home(local_script)
  67         if ssh_handle.run("./"+os.path.basename(local_script)) != 0:
  68             print "WARNING: sanity check script %s FAILED"
  69             # xxx - temporary : ignore result and always return true for now
  70             #return False
  71         return True
  72