# $Id$
-import os
+import os, os.path
+import datetime
+import time
import sys
-import xmlrpclib
import datetime
-import pprint
import traceback
+from types import StringTypes
+
import utils
from TestSite import TestSite
from TestNode import TestNode
from TestUser import TestUser
+from TestKey import TestKey
+from TestSlice import TestSlice
+from TestSliver import TestSliver
+from TestBox import TestBox
+from TestSsh import TestSsh
+from TestApiserver import TestApiserver
+
+# step methods must take (self) and return a boolean (options is a member of the class)
+
+def standby(minutes,dry_run):
+ utils.header('Entering StandBy for %d mn'%minutes)
+ if dry_run:
+ print 'dry_run'
+ else:
+ time.sleep(60*minutes)
+ return True
+
+def standby_generic (func):
+ def actual(self):
+ minutes=int(func.__name__.split("_")[1])
+ return standby(minutes,self.options.dry_run)
+ return actual
+
+def node_mapper (method):
+ def actual(self):
+ overall=True
+ node_method = TestNode.__dict__[method.__name__]
+ for site_spec in self.plc_spec['sites']:
+ test_site = TestSite (self,site_spec)
+ for node_spec in site_spec['nodes']:
+ test_node = TestNode (self,test_site,node_spec)
+ if not node_method(test_node): overall=False
+ return overall
+ return actual
-# step methods must take (self, options) and return a boolean
+def slice_mapper_options (method):
+ def actual(self):
+ overall=True
+ slice_method = TestSlice.__dict__[method.__name__]
+ for slice_spec in self.plc_spec['slices']:
+ site_spec = self.locate_site (slice_spec['sitename'])
+ test_site = TestSite(self,site_spec)
+ test_slice=TestSlice(self,test_site,slice_spec)
+ if not slice_method(test_slice,self.options): overall=False
+ return overall
+ return actual
class TestPlc:
- def __init__ (self,plc_spec):
+ def __init__ (self,plc_spec,options):
self.plc_spec=plc_spec
- self.url="https://%s:443/PLCAPI/"%plc_spec['hostname']
- self.server=xmlrpclib.Server(self.url,allow_none=True)
- self.path=os.path.dirname(sys.argv[0])
+ self.options=options
+ self.test_ssh=TestSsh(self.plc_spec['hostname'],self.options.buildname)
try:
- self.vserver=plc_spec['vserver']
+ self.vserverip=plc_spec['vserverip']
+ self.vservername=plc_spec['vservername']
+ self.url="https://%s:443/PLCAPI/"%plc_spec['vserverip']
+ self.vserver=True
except:
- self.vserver=None
+ self.vserver=False
+ self.url="https://%s:443/PLCAPI/"%plc_spec['hostname']
+# utils.header('Using API url %s'%self.url)
+ self.apiserver=TestApiserver(self.url,options.dry_run)
def name(self):
- return self.plc_spec['name']
+ name=self.plc_spec['name']
+ if self.vserver:
+ return name+".vserver.%s"%self.vservername
+ else:
+ return name+".chroot"
+ def hostname(self):
+ return self.plc_spec['hostname']
+
+ def is_local (self):
+ return self.test_ssh.is_local()
+
+ # define the API methods on this object through xmlrpc
+ # would help, but not strictly necessary
def connect (self):
- # tricky : define les methodes de l'API sur cet object
pass
+
+ def actual_command_in_guest (self,command):
+ return self.test_ssh.actual_command(self.host_to_guest(command))
+ def run_in_guest (self,command):
+ return utils.system(self.actual_command_in_guest(command))
+
+ def run_in_host (self,command):
+ return self.test_ssh.run_in_buildname(command)
+
+ #command gets run in the chroot/vserver
+ def host_to_guest(self,command):
+ if self.vserver:
+ return "vserver %s exec %s"%(self.vservername,command)
+ else:
+ return "chroot /plc/root %s"%TestSsh.backslash_shell_specials(command)
+
+ # copy a file to the myplc root image - pass in_data=True if the file must go in /plc/data
+ def copy_in_guest (self, localfile, remotefile, in_data=False):
+ if in_data:
+ chroot_dest="/plc/data"
+ else:
+ chroot_dest="/plc/root"
+ if self.is_local():
+ if not self.vserver:
+ utils.system("cp %s %s/%s"%(localfile,chroot_dest,remotefile))
+ else:
+ utils.system("cp %s /vservers/%s/%s"%(localfile,self.vservername,remotefile))
+ else:
+ if not self.vserver:
+ utils.system("scp %s %s:%s/%s"%(localfile,self.hostname(),chroot_dest,remotefile))
+ else:
+ utils.system("scp %s %s@/vservers/%s/%s"%(localfile,self.hostname(),self.vservername,remotefile))
+
+
+ # xxx quick n dirty
+ def run_in_guest_piped (self,local,remote):
+ return utils.system(local+" | "+self.test_ssh.actual_command(self.host_to_guest(remote)))
+
def auth_root (self):
return {'Username':self.plc_spec['PLC_ROOT_USER'],
'AuthMethod':'password',
'AuthString':self.plc_spec['PLC_ROOT_PASSWORD'],
'Role' : self.plc_spec['role']
}
- def display_results(self, test_case_name, status, timers):
- timers=datetime.datetime.now()
- fileHandle = open (self.path+'/results.txt', 'a' )
- fileHandle.write ( str(test_case_name)+'\t' +str(status)+'\t'+str(timers)+'\n')
- fileHandle.close()
-
- def init_node (self,test_site,node_spec,path):
-
- test_node = TestNode(self, test_site, node_spec)
- test_node.create_node (test_site.locate_user("pi"))
- test_node.create_node ("tech")
- test_node.create_boot_cd(path)
- return test_node
-
def locate_site (self,sitename):
for site in self.plc_spec['sites']:
if site['site_fields']['name'] == sitename:
return site
raise Exception,"Cannot locate site %s"%sitename
+ def locate_node (self,nodename):
+ for site in self.plc_spec['sites']:
+ for node in site['nodes']:
+ if node['name'] == nodename:
+ return (site,node)
+ raise Exception,"Cannot locate node %s"%nodename
+
+ def locate_hostname (self,hostname):
+ for site in self.plc_spec['sites']:
+ for node in site['nodes']:
+ if node['node_fields']['hostname'] == hostname:
+ return (site,node)
+ raise Exception,"Cannot locate hostname %s"%hostname
+
def locate_key (self,keyname):
for key in self.plc_spec['keys']:
if key['name'] == keyname:
return key
raise Exception,"Cannot locate key %s"%keyname
-
- def kill_all_vmwares(self):
- utils.header('Killing any running vmware or vmplayer instance')
- os.system('pgrep vmware | xargs -r kill')
- os.system('pgrep vmplayer | xargs -r kill ')
- os.system('pgrep vmware | xargs -r kill -9')
- os.system('pgrep vmplayer | xargs -r kill -9')
-
- # step methods
- def uninstall(self,options):
- os.system('set -x; service plc safestop')
+
+ def locate_slice (self, slicename):
+ for slice in self.plc_spec['slices']:
+ if slice['slice_fields']['name'] == slicename:
+ return slice
+ raise Exception,"Cannot locate slice %s"%slicename
+
+ # all different hostboxes used in this plc
+ def gather_hostBoxes(self):
+ # maps on sites and nodes, return [ (host_box,test_node) ]
+ tuples=[]
+ for site_spec in self.plc_spec['sites']:
+ test_site = TestSite (self,site_spec)
+ for node_spec in site_spec['nodes']:
+ test_node = TestNode (self, test_site, node_spec)
+ if not test_node.is_real():
+ tuples.append( (test_node.host_box(),test_node) )
+ # transform into a dict { 'host_box' -> [ hostnames .. ] }
+ result = {}
+ for (box,node) in tuples:
+ if not result.has_key(box):
+ result[box]=[node]
+ else:
+ result[box].append(node)
+ return result
+
+ # a step for checking this stuff
+ def show_boxes (self):
+ for (box,nodes) in self.gather_hostBoxes().iteritems():
+ print box,":"," + ".join( [ node.name() for node in nodes ] )
+ return True
+
+ # make this a valid step
+ def kill_all_qemus(self):
+ for (box,nodes) in self.gather_hostBoxes().iteritems():
+ # this is the brute force version, kill all qemus on that host box
+ TestBox(box,self.options.buildname).kill_all_qemus()
+ return True
+
+ # make this a valid step
+ def list_all_qemus(self):
+ for (box,nodes) in self.gather_hostBoxes().iteritems():
+ # this is the brute force version, kill all qemus on that host box
+ TestBox(box,self.options.buildname).list_all_qemus()
+ return True
+
+ # kill only the right qemus
+ def list_qemus(self):
+ for (box,nodes) in self.gather_hostBoxes().iteritems():
+ # the fine-grain version
+ for node in nodes:
+ node.list_qemu()
+ return True
+
+ # kill only the right qemus
+ def kill_qemus(self):
+ for (box,nodes) in self.gather_hostBoxes().iteritems():
+ # the fine-grain version
+ for node in nodes:
+ node.kill_qemu()
+ return True
+
+ #################### step methods
+
+ ### uninstall
+ def uninstall_chroot(self):
+ self.run_in_host('service plc safestop')
#####detecting the last myplc version installed and remove it
- os.system('set -x; rpm -e myplc')
+ self.run_in_host('rpm -e myplc')
##### Clean up the /plc directory
- os.system('set -x; rm -rf /plc/data')
+ self.run_in_host('rm -rf /plc/data')
+ ##### stop any running vservers
+ self.run_in_host('for vserver in $(ls -d /vservers/* | sed -e s,/vservers/,,) ; do case $vserver in vtest*) echo Shutting down vserver $vserver ; vserver $vserver stop ;; esac ; done')
return True
-
- def install(self,options):
- utils.header('Installing from %s'%options.myplc_url)
- url=options.myplc_url
- os.system('set -x; rpm -Uvh '+url)
- os.system('set -x; service plc mount')
+
+ def uninstall_vserver(self):
+ self.run_in_host("vserver --silent %s delete"%self.vservername)
+ return True
+
+ def uninstall(self):
+ # if there's a chroot-based myplc running, and then a native-based myplc is being deployed
+ # it sounds safer to have the former uninstalled too
+ # now the vserver method cannot be invoked for chroot instances as vservername is required
+ if self.vserver:
+ self.uninstall_vserver()
+ self.uninstall_chroot()
+ else:
+ self.uninstall_chroot()
return True
- def configure(self,options):
- tmpname='/tmp/plc-config-tty-%d'%os.getpid()
+ ### install
+ def install_chroot(self):
+ # nothing to do
+ return True
+
+ def install_vserver(self):
+ # we need build dir for vtest-init-vserver
+ if self.is_local():
+ # a full path for the local calls
+ build_dir=os.path(sys.argv[0])+"/build"
+ else:
+ # use a standard name - will be relative to HOME
+ build_dir="options.buildname"
+ # run checkout in any case - would do an update if already exists
+ build_checkout = "svn checkout %s %s"%(self.options.build_url,build_dir)
+ if self.run_in_host(build_checkout) != 0:
+ return False
+ # the repo url is taken from myplc-url
+ # with the last two steps (i386/myplc...) removed
+ repo_url = self.options.myplc_url
+ for level in [ 'rpmname','arch' ]:
+ repo_url = os.path.dirname(repo_url)
+ create_vserver="%s/vtest-init-vserver.sh %s %s -- --interface eth0:%s"%\
+ (build_dir,self.vservername,repo_url,self.vserverip)
+ return self.run_in_host(create_vserver) == 0
+
+ def install(self):
+ if self.vserver:
+ return self.install_vserver()
+ else:
+ return self.install_chroot()
+
+ ### install_rpm - make this an optional step
+ def cache_rpm(self):
+ url = self.options.myplc_url
+ rpm = os.path.basename(url)
+ cache_fetch="pwd;if [ -f %(rpm)s ] ; then echo Using cached rpm %(rpm)s ; else echo Fetching %(url)s ; curl -O %(url)s; fi"%locals()
+ return self.run_in_host(cache_fetch)==0
+
+ def install_rpm_chroot(self):
+ url = self.options.myplc_url
+ rpm = os.path.basename(url)
+ if not self.cache_rpm():
+ return False
+ utils.header('Installing the : %s'%rpm)
+ return self.run_in_host('rpm -Uvh '+rpm)==0 and self.run_in_host('service plc mount')==0
+
+ def install_rpm_vserver(self):
+ return self.run_in_guest("yum -y install myplc-native")==0
+
+ def install_rpm(self):
+ if self.vserver:
+ return self.install_rpm_vserver()
+ else:
+ return self.install_rpm_chroot()
+
+ ###
+ def configure(self):
+ tmpname='%s.plc-config-tty'%(self.name())
fileconf=open(tmpname,'w')
for var in [ 'PLC_NAME',
'PLC_ROOT_PASSWORD',
fileconf.write('w\n')
fileconf.write('q\n')
fileconf.close()
- os.system('set -x ; cat %s'%tmpname)
- os.system('set -x ; chroot /plc/root plc-config-tty < %s'%tmpname)
- os.system('set -x ; service plc start')
- os.system('set -x; service sendmail stop')
- os.system('set -x; chroot /plc/root service sendmail restart')
- os.system('set -x; rm %s'%tmpname)
+ utils.system('cat %s'%tmpname)
+ self.run_in_guest_piped('cat %s'%tmpname,'plc-config-tty')
+ utils.system('rm %s'%tmpname)
+ return True
+
+ # the chroot install is slightly different to this respect
+ def start(self):
+ if self.vserver:
+ self.run_in_guest('service plc start')
+ else:
+ self.run_in_host('service plc start')
return True
- def sites (self,options):
- return self.do_sites(options)
+ def stop(self):
+ if self.vserver:
+ self.run_in_guest('service plc stop')
+ else:
+ self.run_in_host('service plc stop')
+ return True
+
+ # could use a TestKey class
+ def store_keys(self):
+ for key_spec in self.plc_spec['keys']:
+ TestKey(self,key_spec).store_key()
+ return True
+
+ def clean_keys(self):
+ utils.system("rm -rf %s/keys/"%os.path(sys.argv[0]))
+
+ def sites (self):
+ return self.do_sites()
- def clean_sites (self,options):
- return self.do_sites(options,"delete")
+ def clean_sites (self):
+ return self.do_sites(action="delete")
- def do_sites (self,options,action="add"):
+ def do_sites (self,action="add"):
for site_spec in self.plc_spec['sites']:
test_site = TestSite (self,site_spec)
- if (action == "delete"):
+ if (action != "add"):
utils.header("Deleting site %s in %s"%(test_site.name(),self.name()))
test_site.delete_site()
# deleted with the site
test_site.create_users()
return True
- def nodes (self, options):
+ def nodes (self):
+ return self.do_nodes()
+ def clean_nodes (self):
+ return self.do_nodes(action="delete")
+
+ def do_nodes (self,action="add"):
for site_spec in self.plc_spec['sites']:
test_site = TestSite (self,site_spec)
- utils.header("Creating nodes for site %s in %s"%(test_site.name(),self.name()))
- for node_spec in site_spec['nodes']:
- utils.header('Creating node %s'%node_spec)
- pprint.PrettyPrinter(indent=4).pprint(node_spec)
- test_node = TestNode (self,test_site,node_spec)
- test_node.create_node ()
+ if action != "add":
+ utils.header("Deleting nodes in site %s"%test_site.name())
+ for node_spec in site_spec['nodes']:
+ test_node=TestNode(self,test_site,node_spec)
+ utils.header("Deleting %s"%test_node.name())
+ test_node.delete_node()
+ else:
+ utils.header("Creating nodes for site %s in %s"%(test_site.name(),self.name()))
+ for node_spec in site_spec['nodes']:
+ utils.pprint('Creating node %s'%node_spec,node_spec)
+ test_node = TestNode (self,test_site,node_spec)
+ test_node.create_node ()
return True
- def bootcd (self, options):
+ # create nodegroups if needed, and populate
+ # no need for a clean_nodegroups if we are careful enough
+ def nodegroups (self):
+ # 1st pass to scan contents
+ groups_dict = {}
for site_spec in self.plc_spec['sites']:
test_site = TestSite (self,site_spec)
for node_spec in site_spec['nodes']:
test_node=TestNode (self,test_site,node_spec)
- test_node.create_boot_cd(options.path)
+ if node_spec.has_key('nodegroups'):
+ nodegroupnames=node_spec['nodegroups']
+ if isinstance(nodegroupnames,StringTypes):
+ nodegroupnames = [ nodegroupnames ]
+ for nodegroupname in nodegroupnames:
+ if not groups_dict.has_key(nodegroupname):
+ groups_dict[nodegroupname]=[]
+ groups_dict[nodegroupname].append(test_node.name())
+ auth=self.auth_root()
+ for (nodegroupname,group_nodes) in groups_dict.iteritems():
+ try:
+ self.apiserver.GetNodeGroups(auth,{'name':nodegroupname})[0]
+ except:
+ self.apiserver.AddNodeGroup(auth,{'name':nodegroupname})
+ for node in group_nodes:
+ self.apiserver.AddNodeToNodeGroup(auth,node,nodegroupname)
+ return True
+
+ def all_hostnames (self) :
+ hostnames = []
+ for site_spec in self.plc_spec['sites']:
+ hostnames += [ node_spec['node_fields']['hostname'] \
+ for node_spec in site_spec['nodes'] ]
+ return hostnames
+
+ # gracetime : during the first <gracetime> minutes nothing gets printed
+ def do_nodes_booted (self, minutes, gracetime=2):
+ if self.options.dry_run:
+ print 'dry_run'
+ return True
+ # compute timeout
+ timeout = datetime.datetime.now()+datetime.timedelta(minutes=minutes)
+ graceout = datetime.datetime.now()+datetime.timedelta(minutes=gracetime)
+ # the nodes that haven't checked yet - start with a full list and shrink over time
+ tocheck = self.all_hostnames()
+ utils.header("checking nodes %r"%tocheck)
+ # create a dict hostname -> status
+ status = dict ( [ (hostname,'undef') for hostname in tocheck ] )
+ while tocheck:
+ # get their status
+ tocheck_status=self.apiserver.GetNodes(self.auth_root(), tocheck, ['hostname','boot_state' ] )
+ # update status
+ for array in tocheck_status:
+ hostname=array['hostname']
+ boot_state=array['boot_state']
+ if boot_state == 'boot':
+ utils.header ("%s has reached the 'boot' state"%hostname)
+ else:
+ # if it's a real node, never mind
+ (site_spec,node_spec)=self.locate_hostname(hostname)
+ if TestNode.is_real_model(node_spec['node_fields']['model']):
+ utils.header("WARNING - Real node %s in %s - ignored"%(hostname,boot_state))
+ # let's cheat
+ boot_state = 'boot'
+ if datetime.datetime.now() > graceout:
+ utils.header ("%s still in '%s' state"%(hostname,boot_state))
+ graceout=datetime.datetime.now()+datetime.timedelta(1)
+ status[hostname] = boot_state
+ # refresh tocheck
+ tocheck = [ hostname for (hostname,boot_state) in status.iteritems() if boot_state != 'boot' ]
+ if not tocheck:
+ return True
+ if datetime.datetime.now() > timeout:
+ for hostname in tocheck:
+ utils.header("FAILURE due to %s in '%s' state"%(hostname,status[hostname]))
+ return False
+ # otherwise, sleep for a while
+ time.sleep(15)
+ # only useful in empty plcs
+ return True
+
+ def nodes_booted(self):
+ return self.do_nodes_booted(minutes=0)
+
+
+ def do_nodes_ssh(self,minutes):
+ # compute timeout
+ timeout = datetime.datetime.now()+datetime.timedelta(minutes=minutes)
+ tocheck = self.all_hostnames()
+# self.scan_publicKeys(tocheck)
+ utils.header("checking Connectivity on nodes %r"%tocheck)
+ while tocheck:
+ for hostname in tocheck:
+ # try to ssh in nodes
+ node_test_ssh = TestSsh (hostname,key="/etc/planetlab/root_ssh_key.rsa")
+ access=self.run_in_guest(node_test_ssh.actual_command("date"))
+ if not access:
+ utils.header('The node %s is sshable -->'%hostname)
+ # refresh tocheck
+ tocheck.remove(hostname)
+ else:
+ # we will have tried real nodes once, in case they're up - but if not, just skip
+ (site_spec,node_spec)=self.locate_hostname(hostname)
+ if TestNode.is_real_model(node_spec['node_fields']['model']):
+ utils.header ("WARNING : check ssh access into real node %s - skipped"%hostname)
+ tocheck.remove(hostname)
+ if not tocheck:
+ return True
+ if datetime.datetime.now() > timeout:
+ for hostname in tocheck:
+ utils.header("FAILURE to ssh into %s"%hostname)
+ return False
+ # otherwise, sleep for a while
+ time.sleep(15)
+ # only useful in empty plcs
return True
-
- def initscripts (self, options):
+
+ def nodes_ssh(self):
+ return self.do_nodes_ssh(minutes=2)
+
+ @node_mapper
+ def init_node (self): pass
+ @node_mapper
+ def bootcd (self): pass
+ @node_mapper
+ def configure_qemu (self): pass
+
+ def do_check_initscripts(self):
+ overall = True
+ for slice_spec in self.plc_spec['slices']:
+ if not slice_spec.has_key('initscriptname'):
+ continue
+ initscript=slice_spec['initscriptname']
+ for nodename in slice_spec['nodenames']:
+ (site,node) = self.locate_node (nodename)
+ # xxx - passing the wrong site - probably harmless
+ test_site = TestSite (self,site)
+ test_slice = TestSlice (self,test_site,slice_spec)
+ test_node = TestNode (self,test_site,node)
+ test_sliver = TestSliver (self, test_node, test_slice)
+ if not test_sliver.check_initscript(initscript):
+ overall = False
+ return overall
+
+ def check_initscripts(self):
+ return self.do_check_initscripts()
+
+ def initscripts (self):
for initscript in self.plc_spec['initscripts']:
- utils.header('Adding Initscript %s in plc %s'%\
- (initscript['name'],self.plc_spec['name']))
- pprint.PrettyPrinter(indent=4).pprint(initscript)
- self.server.AddInitScript(self.auth_root(),initscript['initscript_fields'])
+ utils.pprint('Adding Initscript in plc %s'%self.plc_spec['name'],initscript)
+ self.apiserver.AddInitScript(self.auth_root(),initscript['initscript_fields'])
return True
- def slices (self, options):
+ def slices (self):
return self.do_slices()
- def clean_slices (self, options):
+ def clean_slices (self):
return self.do_slices("delete")
- ### would need a TestSlice class
- def do_slices (self, add_or_delete="add"):
+ def do_slices (self, action="add"):
for slice in self.plc_spec['slices']:
site_spec = self.locate_site (slice['sitename'])
test_site = TestSite(self,site_spec)
- owner_spec = test_site.locate_user(slice['owner'])
- auth = TestUser(self,test_site,owner_spec).auth()
- slice_fields = slice['slice_fields']
- slice_name = slice_fields['name']
- if (add_or_delete == "delete"):
- self.server.DeleteSlice(auth,slice_fields['name'])
- utils.header("Deleted slice %s"%slice_fields['name'])
- continue
- pprint.PrettyPrinter(indent=4).pprint(slice_fields)
- self.server.AddSlice(auth,slice_fields)
- utils.header('Created Slice %s'%slice_fields['name'])
- for username in slice['usernames']:
- user_spec=test_site.locate_user(username)
- test_user=TestUser(self,test_site,user_spec)
- self.server.AddPersonToSlice(auth, test_user.name(), slice_name)
-
- hostnames=[]
- for nodename in slice['nodenames']:
- node_spec=test_site.locate_node(nodename)
- test_node=TestNode(self,test_site,node_spec)
- hostnames += [test_node.name()]
- utils.header("Adding %r in %s"%(hostnames,slice_name))
- self.server.AddSliceToNodes(auth, slice_name, hostnames)
- if slice.has_key('initscriptname'):
- isname=slice['initscriptname']
- utils.header("Adding initscript %s in %s"%(isname,slice_name))
- self.server.AddSliceAttribute(self.auth_root(), slice_name,
- 'initscript',isname)
+ test_slice=TestSlice(self,test_site,slice)
+ if action != "add":
+ utils.header("Deleting slices in site %s"%test_site.name())
+ test_slice.delete_slice()
+ else:
+ utils.pprint("Creating slice",slice)
+ test_slice.create_slice()
+ utils.header('Created Slice %s'%slice['slice_fields']['name'])
return True
- def start_nodes (self, options):
- self.kill_all_vmwares()
- utils.header("Starting vmware nodes")
+ @slice_mapper_options
+ def check_slice(self): pass
+
+ @node_mapper
+ def clear_known_hosts (self): pass
+
+ def start_nodes (self):
+ utils.header("Starting nodes")
for site_spec in self.plc_spec['sites']:
- TestSite(self,site_spec).start_nodes (options)
+ TestSite(self,site_spec).start_nodes (self.options)
return True
- def stop_nodes (self, options):
- self.kill_all_vmwares ()
+ def locate_first_sliver (self):
+ slice_spec = self.plc_spec['slices'][0]
+ slicename = slice_spec['slice_fields']['name']
+ nodename = slice_spec['nodenames'][0]
+ return self.locate_sliver_obj(nodename,slicename)
+
+ def locate_sliver_obj (self,nodename,slicename):
+ (site,node) = self.locate_node(nodename)
+ slice = self.locate_slice (slicename)
+ # build objects
+ test_site = TestSite (self, site)
+ test_node = TestNode (self, test_site,node)
+ # xxx the slice site is assumed to be the node site - mhh - probably harmless
+ test_slice = TestSlice (self, test_site, slice)
+ return TestSliver (self, test_node, test_slice)
+
+ def check_tcp (self):
+ specs = self.plc_spec['tcp_test']
+ overall=True
+ for spec in specs:
+ port = spec['port']
+ # server side
+ s_test_sliver = self.locate_sliver_obj (spec['server_node'],spec['server_slice'])
+ if not s_test_sliver.run_tcp_server(port,timeout=10):
+ overall=False
+ break
+
+ # idem for the client side
+ c_test_sliver = self.locate_sliver_obj(spec['server_node'],spec['server_slice'])
+ if not c_test_sliver.run_tcp_client(s_test_sliver.test_node.name(),port):
+ overall=False
+ return overall
+
+
+ def gather_logs (self):
+ # (1) get the plc's /var/log and store it locally in logs/<plcname>-var-log/*
+ # (2) get all the nodes qemu log and store it as logs/<node>-qemu.log
+ # (3) get the nodes /var/log and store is as logs/<node>-var-log/*
+ # (4) as far as possible get the slice's /var/log as logs/<slice>-<node>-var-log/*
+ # (1)
+ print "-------------------- TestPlc.gather_logs : PLC's /var/log"
+ self.gather_var_logs ()
+ # (2)
+ print "-------------------- TestPlc.gather_logs : nodes's QEMU logs"
+ for site_spec in self.plc_spec['sites']:
+ test_site = TestSite (self,site_spec)
+ for node_spec in site_spec['nodes']:
+ test_node=TestNode(self,test_site,node_spec)
+ test_node.gather_qemu_logs()
+ # (3)
+ print "-------------------- TestPlc.gather_logs : nodes's /var/log"
+ self.gather_nodes_var_logs()
+ # (4)
+ print "-------------------- TestPlc.gather_logs : sample sliver's /var/log"
+ self.gather_first_sliver_logs()
return True
- def db_dump(self, options):
+ def gather_first_sliver_logs(self):
+ try:
+ test_sliver = self.locate_first_sliver()
+ remote = test_sliver.tar_var_logs()
+ utils.system("mkdir -p logs/%s-var-log"%test_sliver.name())
+ command = remote + " | tar -C logs/%s-var-log -xf -"%test_sliver.name()
+ utils.system(command)
+ except Exception,e:
+ print 'Cannot locate first sliver - giving up',e
+ return True
+
+ def gather_var_logs (self):
+ to_plc = self.actual_command_in_guest("tar -C /var/log/ -cf - .")
+ command = to_plc + "| tar -C logs/%s-var-log -xf -"%self.name()
+ utils.system("mkdir -p logs/%s-var-log"%self.name())
+ utils.system(command)
+
+ def gather_nodes_var_logs (self):
+ for site_spec in self.plc_spec['sites']:
+ test_site = TestSite (self,site_spec)
+ for node_spec in site_spec['nodes']:
+ test_node=TestNode(self,test_site,node_spec)
+ test_ssh = TestSsh (test_node.name(),key="/etc/planetlab/root_ssh_key.rsa")
+ to_plc = self.actual_command_in_guest ( test_ssh.actual_command("tar -C /var/log -cf - ."))
+ command = to_plc + "| tar -C logs/%s-var-log -xf -"%test_node.name()
+ utils.system("mkdir -p logs/%s-var-log"%test_node.name())
+ utils.system(command)
+
+
+ # returns the filename to use for sql dump/restore, using options.dbname if set
+ def dbfile (self, database):
# uses options.dbname if it is found
try:
- name=options.dbname
+ name=self.options.dbname
+ if not isinstance(name,StringTypes):
+ raise Exception
except:
t=datetime.datetime.now()
d=t.date()
name=str(d)
- dump='/data/%s.sql'%name
- os.system('chroot /plc/root pg_dump -U pgsqluser planetlab4 -f '+ dump)
+ return "/root/%s-%s.sql"%(database,name)
+
+ def db_dump(self):
+ dump=self.dbfile("planetab4")
+ self.run_in_guest('pg_dump -U pgsqluser planetlab4 -f '+ dump)
utils.header('Dumped planetlab4 database in %s'%dump)
return True
+ def db_restore(self):
+ dump=self.dbfile("planetab4")
+ ##stop httpd service
+ self.run_in_guest('service httpd stop')
+ # xxx - need another wrapper
+ self.run_in_guest_piped('echo drop database planetlab4','psql --user=pgsqluser template1')
+ self.run_in_guest('createdb -U postgres --encoding=UNICODE --owner=pgsqluser planetlab4')
+ self.run_in_guest('psql -U pgsqluser planetlab4 -f '+dump)
+ ##starting httpd service
+ self.run_in_guest('service httpd start')
+
+ utils.header('Database restored from ' + dump)
+
+ @standby_generic
+ def standby_1(): pass
+ @standby_generic
+ def standby_2(): pass
+ @standby_generic
+ def standby_3(): pass
+ @standby_generic
+ def standby_4(): pass
+ @standby_generic
+ def standby_5(): pass
+ @standby_generic
+ def standby_6(): pass
+ @standby_generic
+ def standby_7(): pass
+ @standby_generic
+ def standby_8(): pass
+ @standby_generic
+ def standby_9(): pass
+ @standby_generic
+ def standby_10(): pass
+ @standby_generic
+ def standby_11(): pass
+ @standby_generic
+ def standby_12(): pass
+ @standby_generic
+ def standby_13(): pass
+ @standby_generic
+ def standby_14(): pass
+ @standby_generic
+ def standby_15(): pass
+ @standby_generic
+ def standby_16(): pass
+ @standby_generic
+ def standby_17(): pass
+ @standby_generic
+ def standby_18(): pass
+ @standby_generic
+ def standby_19(): pass
+ @standby_generic
+ def standby_20(): pass
+