X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=system%2FTestNode.py;h=4d684dcb359aa3eb51126e1557f61b7dbd61b55d;hb=1c87dd36c45ad0896491f51aa8bdf63f8668da25;hp=f9f39797db37d99f5864c754af623f29d414836f;hpb=eb4d7e9f9e7f77d6f2e49aab70dd715a43708230;p=tests.git diff --git a/system/TestNode.py b/system/TestNode.py index f9f3979..4d684dc 100644 --- a/system/TestNode.py +++ b/system/TestNode.py @@ -1,9 +1,13 @@ -import os, sys, time, base64 +# Thierry Parmentelat +# Copyright (C) 2010 INRIA +# +import sys, os, os.path, time, base64 import xmlrpclib import utils from TestUser import TestUser -from TestBox import TestBox +from TestBoxQemu import TestBoxQemu +from TestSsh import TestSsh class TestNode: @@ -47,12 +51,12 @@ class TestNode: utils.header("WARNING : qemu nodes need a host box") return 'localhost' - # this returns a TestBox instance - cached in .test_box_value + # this returns a TestBoxQemu instance - cached in .test_box_value def test_box (self): try: return self.test_box_value except: - self.test_box_value = TestBox (self.host_box(),self.buildname()) + self.test_box_value = TestBoxQemu (self.host_box(),self.buildname()) return self.test_box_value def create_node (self): @@ -62,33 +66,35 @@ class TestNode: userauth = test_user.auth() utils.header("node %s created by user %s"%(self.name(),test_user.name())) rootauth=self.test_plc.auth_root() - server = self.test_plc.server + server = self.test_plc.apiserver server.AddNode(userauth, self.test_site.site_spec['site_fields']['login_base'], self.node_spec['node_fields']) + server.SetNodePlainBootstrapfs(userauth, + self.node_spec['node_fields']['hostname'], + 'YES') # create as reinstall to avoid user confirmation - server.UpdateNode(userauth, self.name(), {'boot_state':'rins'}) + server.UpdateNode(userauth, self.name(), {'boot_state':'reinstall'}) # populate network interfaces - primary - server.AddNodeNetwork(userauth,self.name(), - self.node_spec['network_fields']) + server.AddInterface(userauth,self.name(), + self.node_spec['interface_fields']) # populate network interfaces - others if self.node_spec.has_key('extra_interfaces'): for interface in self.node_spec['extra_interfaces']: - server.AddNodeNetwork(userauth,self.name(), - interface['network_fields']) + server.AddInterface(userauth,self.name(), interface['interface_fields']) if interface.has_key('settings'): for (attribute,value) in interface['settings'].iteritems(): # locate node network - nn = server.GetNodeNetworks(userauth,{'ip':interface['network_fields']['ip']})[0] - nnid=nn['nodenetwork_id'] + interface = server.GetInterfaces(userauth,{'ip':interface['interface_fields']['ip']})[0] + interface_id=interface['interface_id'] # locate or create node network attribute type try: - nnst = server.GetNodeNetworkSettingTypes(userauth,{'name':attribute})[0] + interface_tagtype = server.GetTagTypes(userauth,{'name':attribute})[0] except: - nnst = server.AddNodeNetworkSettingType(rootauth,{'category':'test', - 'name':attribute}) + interface_tagtype = server.AddTagType(rootauth,{'category':'test', + 'tagname':attribute}) # attach value - server.AddNodeNetworkSetting(userauth,nnid,attribute,value) + server.AddInterfaceTag(userauth,interface_id,attribute,value) def delete_node (self): # uses the right auth as far as poss. @@ -99,87 +105,190 @@ class TestNode: auth = test_user.auth() except: auth=self.test_plc.auth_root() - self.test_plc.server.DeleteNode(auth,self.name()) + self.test_plc.apiserver.DeleteNode(auth,self.name()) # Do most of the stuff locally - will be pushed on host_box - *not* the plc - later if needed - def prepare_area(self): + def qemu_local_init(self): utils.system("rm -rf %s"%self.nodedir()) utils.system("mkdir %s"%self.nodedir()) - #create the tar log file - utils.system("rm -rf nodeslogs && mkdir nodeslogs") - if self.is_qemu(): - utils.system("rsync -v -a --exclude .svn template-qemu/ %s/"%self.nodedir()) + if not self.is_qemu(): + return True + return utils.system("rsync -v -a --exclude .svn template-qemu/ %s/"%self.nodedir())==0 - def create_boot_cd(self): + def bootcd(self): utils.header("Calling GetBootMedium for %s"%self.name()) options = [] if self.is_qemu(): - options=['serial'] - encoded=self.test_plc.server.GetBootMedium(self.test_plc.auth_root(), self.name(), 'node-iso', '', options) + options.append('serial') + options.append('no-hangcheck') + encoded=self.test_plc.apiserver.GetBootMedium(self.test_plc.auth_root(), + self.name(), 'node-iso', '', options) if (encoded == ''): raise Exception, 'GetBootmedium failed' filename="%s/%s.iso"%(self.nodedir(),self.name()) utils.header('Storing boot medium into %s'%filename) - file(filename,'w').write(base64.b64decode(encoded)) + if self.test_plc.options.dry_run: + print "Dry_run: skipped writing of iso image" + return True + else: + file(filename,'w').write(base64.b64decode(encoded)) + return True + + def nodestate_reinstall (self): + self.test_plc.apiserver.UpdateNode(self.test_plc.auth_root(), + self.name(),{'boot_state':'reinstall'}) + return True + + def nodestate_safeboot (self): + self.test_plc.apiserver.UpdateNode(self.test_plc.auth_root(), + self.name(),{'boot_state':'safeboot'}) + return True - def configure_qemu(self): + def nodestate_boot (self): + self.test_plc.apiserver.UpdateNode(self.test_plc.auth_root(), + self.name(),{'boot_state':'boot'}) + return True + + def nodestate_show (self): + if self.test_plc.options.dry_run: + print "Dry_run: skipped getting current node state" + return True + state=self.test_plc.apiserver.GetNodes(self.test_plc.auth_root(), self.name(), ['boot_state'])[0]['boot_state'] + print self.name(),':',state + return True + + def qemu_local_config(self): if not self.is_qemu(): return - mac=self.node_spec['network_fields']['mac'] + mac=self.node_spec['interface_fields']['mac'] hostname=self.node_spec['node_fields']['hostname'] + ip=self.node_spec['interface_fields']['ip'] + auth=self.test_plc.auth_root() + target_arch=self.test_plc.apiserver.GetPlcRelease(auth)['build']['target-arch'] conf_filename="%s/qemu.conf"%(self.nodedir()) + if self.test_plc.options.dry_run: + print "dry_run: skipped actual storage of qemu.conf" + return True utils.header('Storing qemu config for %s in %s'%(self.name(),conf_filename)) file=open(conf_filename,'w') file.write('MACADDR=%s\n'%mac) file.write('NODE_ISO=%s.iso\n'%self.name()) file.write('HOSTNAME=%s\n'%hostname) + file.write('IP=%s\n'%ip) + file.write('TARGET_ARCH=%s\n'%target_arch) file.close() + return True + def qemu_export (self): # if relevant, push the qemu area onto the host box - if ( not self.test_box().is_local()): - utils.header ("Transferring configuration files for node %s onto %s"%(self.name(),self.host_box())) -# self.test_box().clean_dir(self.buildname()) - self.test_box().mkdir("nodeslogs") - self.test_box().copy(self.nodedir(),recursive=True) - + if self.test_box().is_local(): + return True + utils.header ("Cleaning any former sequel of %s on %s"%(self.name(),self.host_box())) + self.test_box().run_in_buildname("rm -rf %s"%self.nodedir()) + utils.header ("Transferring configuration files for node %s onto %s"%(self.name(),self.host_box())) + return self.test_box().copy(self.nodedir(),recursive=True)==0 - def start_node (self,options): + def qemu_start (self): model=self.node_spec['node_fields']['model'] #starting the Qemu nodes before if self.is_qemu(): - self.start_qemu(options) + self.start_qemu() else: - utils.header("TestNode.start_node : %s model %s taken as real node"%(self.name(),model)) + utils.header("TestNode.qemu_start : %s model %s taken as real node"%(self.name(),model)) + return True + + def timestamp_qemu (self): + test_box = self.test_box() + test_box.run_in_buildname("mkdir -p %s"%self.nodedir()) + now=int(time.time()) + return test_box.run_in_buildname("echo %d > %s/timestamp"%(now,self.nodedir()))==0 - def start_qemu (self, options): + def start_qemu (self): test_box = self.test_box() utils.header("Starting qemu node %s on %s"%(self.name(),test_box.hostname())) - test_box.run_in_buildname("qemu-%s/env-qemu start >> nodeslogs/%s.log"%(self.name(),self.name())) + test_box.run_in_buildname("%s/qemu-bridge-init start >> %s/log.txt"%(self.nodedir(),self.nodedir())) # kick it off in background, as it would otherwise hang - test_box.run_in_buildname("qemu-%s/start-qemu-node 2>&1 >> nodeslogs/%s.log &"%(self.name(),self.name()),True) + test_box.run_in_buildname("%s/qemu-start-node 2>&1 >> %s/log.txt"%(self.nodedir(),self.nodedir())) def list_qemu (self): utils.header("Listing qemu for host %s on box %s"%(self.name(),self.test_box().hostname())) - command="qemu-%s/kill-qemu-node -l %s"%(self.name(),self.name()) + command="%s/qemu-kill-node -l %s"%(self.nodedir(),self.name()) self.test_box().run_in_buildname(command) return True def kill_qemu (self): #Prepare the log file before killing the nodes test_box = self.test_box() - if(not test_box.tar_logs()): - utils.header("Failed to get the nodes log files") # kill the right processes - utils.header("Stopping qemu for host %s on box %s"%(self.name(),self.test_box().hostname())) - command="qemu-%s/kill-qemu-node %s"%(self.name(),self.name()) + utils.header("Stopping qemu for node %s on box %s"%(self.name(),self.test_box().hostname())) + command="%s/qemu-kill-node %s"%(self.nodedir(),self.name()) self.test_box().run_in_buildname(command) return True def gather_qemu_logs (self): - utils.header("WARNING - Incomplete logs gathering TestNodes.gather_qemu_logs") + if not self.is_qemu(): + return True + remote_log="%s/log.txt"%self.nodedir() + local_log="logs/node.qemu.%s.txt"%self.name() + self.test_box().test_ssh.fetch(remote_log,local_log) + + def keys_clear_known_hosts (self): + TestSsh(self.name()).clear_known_hosts() + return True + + def create_test_ssh(self): + # get the plc's keys for entering the node + vservername=self.test_plc.vservername +### # assuming we've run testplc.fetch_keys() +### key = "keys/%(vservername)s.rsa"%locals() + # fetch_keys doesn't grab the root key anymore + key = "keys/key1.rsa" + return TestSsh(self.name(), buildname=self.buildname(), key=key) - def gather_var_logs (self): - utils.header("WARNING - Incomplete logs gathering TestNodes.gather_var_logs") + def check_hooks (self): + extensions = [ 'py','pl','sh' ] + path='hooks/node' + scripts=utils.locate_hooks_scripts ('node '+self.name(), path,extensions) + overall = True + for script in scripts: + if not self.check_hooks_script (script): + overall = False + return overall + def check_hooks_script (self,local_script): + # push the script on the node's root context + script_name=os.path.basename(local_script) + utils.header ("NODE hook %s (%s)"%(script_name,self.name())) + test_ssh=self.create_test_ssh() + test_ssh.copy_home(local_script) + if test_ssh.run("./"+script_name) != 0: + utils.header ("WARNING: node hooks check script %s FAILED (ignored)"%script_name) + #return False + return True + else: + utils.header ("SUCCESS: node hook %s OK"%script_name) + return True + + def has_libvirt (self): + test_ssh=self.create_test_ssh() + return test_ssh.run ("rpm -q libvirt-client")==0 + + def check_systemslice (self, slicename,dry_run=False): + sitename=self.test_plc.plc_spec['PLC_SLICE_PREFIX'] + vservername="%s_%s"%(sitename,slicename) + test_ssh=self.create_test_ssh() + if self.has_libvirt(): + return test_ssh.run("virsh --connect lxc:// list | grep -q ' %s '"%vservername, + dry_run=dry_run)==0 + else: + (retcod,output)=utils.output_of(test_ssh.actual_command("cat /vservers/%s/etc/slicefamily 2> /dev/null")%vservername) + # get last line only as ssh pollutes the output + slicefamily=output.split("\n")[-1] + utils.header("system slice %s has slicefamily %s"%(slicename, slicefamily)) + if retcod != 0: + return False + return test_ssh.run("vserver-stat | grep %s"%vservername,dry_run=dry_run)==0 + +