system/TestSliver.py

   1 # Thierry Parmentelat <thierry.parmentelat@inria.fr>
   2 # Copyright (C) 2010 INRIA
   3 #
   4 import utils
   5 import os, os.path
   6 import datetime
   7 import time
   8 from TestSsh import TestSsh
   9
  10 class TestSliver:
  11
  12     def __init__ (self,test_plc,test_node,test_slice):
  13         self.test_plc=test_plc
  14         self.test_node=test_node
  15         self.test_slice=test_slice
  16         self.test_ssh = self.create_test_ssh()
  17
  18     def get_privateKey(self):
  19         slice_spec=self.test_slice.slice_spec
  20         try:
  21             (found,privatekey)=self.test_slice.locate_key()
  22             return (found,privatekey)
  23         except Exception,e:
  24             print str(e)
  25
  26     def create_test_ssh(self):
  27         private_key = self.test_slice.locate_private_key()
  28         if not private_key:
  29             raise Exception,"Cannot find the private key for slice %s"%self.test_slice.name()
  30         return TestSsh (self.test_node.name(),key=private_key,username=self.test_slice.name(),
  31                         # so that copies end up in the home dir
  32                         buildname=".")
  33
  34     def name(self):
  35         return "%s@%s"%(self.test_slice.name(),self.test_node.name())
  36
  37     def check_initscript_stamp(self, stamp):
  38         utils.header("Checking for initscript stamp %s on sliver %s"%(stamp, self.name()))
  39         return self.test_ssh.run("ls -l /var/tmp/%s.stamp"%stamp)==0
  40
  41     def check_tcp_ready (self, port):
  42         ready_command = "./tcptest.py ready -p %d"%(port)
  43         return self.test_ssh.copy("tcptest.py") == 0 and \
  44             self.test_ssh.run(ready_command) == 0
  45
  46     def run_tcp_server (self, port, timeout=10):
  47         server_command = "./tcptest.py server -p %d -t %d"%(port, timeout)
  48         return self.test_ssh.copy("tcptest.py") == 0 and \
  49             self.test_ssh.run(server_command, background=True)==0
  50
  51     def run_tcp_client (self, servername, port, retry=5):
  52         client_command="./tcptest.py client -a %s -p %d"%(servername, port)
  53         if self.test_ssh.copy("tcptest.py") != 0:
  54             return False
  55         if self.test_ssh.run(client_command) == 0:
  56             return True
  57         return False
  58
  59     # use the node's main ssh root entrance, as the slice entrance might be down
  60     #def tar_var_logs (self):
  61     #    return self.test_ssh.actual_command("sudo tar -C /var/log -cf - .")
  62     def tar_var_logs (self):
  63         test_ssh=self.test_node.create_test_ssh()
  64         dir_to_tar="/vservers/%s/var/log"%self.test_slice.name()
  65         return test_ssh.actual_command("tar -C %s -cf - ."%dir_to_tar)
  66
  67     def check_hooks (self):
  68         print 'NOTE: slice hooks check scripts NOT (yet?) run in sudo'
  69         extensions = [ 'py','pl','sh' ]
  70         path='hooks/slice/'
  71         scripts=utils.locate_hooks_scripts ('sliver '+self.name(), path,extensions)
  72         overall = True
  73         for script in scripts:
  74             if not self.check_hooks_script (script):
  75                 overall = False
  76         return overall
  77
  78     def check_hooks_script (self,local_script):
  79         script_name=os.path.basename(local_script)
  80         utils.header ("SLIVER hook %s (%s)"%(script_name,self.name()))
  81         test_ssh=self.create_test_ssh()
  82         test_ssh.copy_home(local_script)
  83         if test_ssh.run("./"+script_name) != 0:
  84             utils.header ("WARNING: hooks check script %s FAILED (ignored)"%script_name)
  85             #return False
  86             return True
  87         else:
  88             utils.header ("SUCCESS: sliver hook %s OK"%script_name)
  89             return True
  90