1 import os, sys, time, base64
6 from TestUser import TestUser
10 def __init__ (self,test_plc,test_site,node_spec):
11 self.test_plc=test_plc
12 self.test_site=test_site
13 self.node_spec=node_spec
16 return self.node_spec['node_fields']['hostname']
18 def create_node (self):
19 ownername = self.node_spec['owner']
20 user_spec = self.test_site.locate_user(ownername)
21 test_user = TestUser(self.test_plc,self.test_site,user_spec)
22 auth = test_user.auth()
23 utils.header("node %s created by user %s"%(self.name(),test_user.name()))
24 filter={'boot_state':'rins'}
25 self.test_plc.server.AddNode(auth,
26 self.test_site.site_spec['site_fields']['login_base'],
27 self.node_spec['node_fields'])
28 self.test_plc.server.AddNodeNetwork(auth,self.name(),
29 self.node_spec['network_fields'])
30 self.test_plc.server.UpdateNode(auth, self.name(), filter)
32 def conffile(self,image,hostname,path):
33 template='%s/template-vmplayer/node.vmx'%(path)
34 actual='%s/vmplayer-%s/node.vmx'%(path,hostname)
35 sed_command="sed -e s,@BOOTCD@,%s,g %s > %s"%(image,template,actual)
36 utils.header('Creating %s from %s'%(actual,template))
37 os.system('set -x; ' + sed_command)
39 def create_boot_cd(self,path):
40 node_spec=self.node_spec
41 hostname=node_spec['node_fields']['hostname']
42 utils.header('Initializing vmplayer area for node %s'%hostname)
43 clean_dir="rm -rf %s/vmplayer-%s"%(path,hostname)
44 mkdir_command="mkdir -p %s/vmplayer-%s"%(path,hostname)
45 tar_command="tar -C %s/template-vmplayer -cf - . | tar -C %s/vmplayer-%s -xf -"%(path,path,hostname)
46 os.system('set -x; ' +clean_dir + ';' + mkdir_command + ';' + tar_command);
47 utils.header('Creating boot medium for node %s'%hostname)
48 encoded=self.test_plc.server.GetBootMedium(self.test_plc.auth_root(), hostname, 'node-iso', '')
50 raise Exception, 'boot.iso not found'
51 file=open(path+'/vmplayer-'+hostname+'/boot_file.iso','w')
52 file.write(base64.b64decode(encoded))
54 utils.header('boot cd created for %s'%hostname)
55 self.conffile('boot_file.iso',hostname, path)
57 def start_node (self,options):
58 model=self.node_spec['node_fields']['model']
59 if model.find("vmware") >= 0:
60 self.start_vmware(options)
61 elif model.find("qemu") >= 0:
62 self.start_qemu(options)
64 utils.header("TestNode.start_node : ignoring model %s"%model)
66 def start_vmware (self,options):
67 hostname=self.node_spec['node_fields']['hostname']
69 display=options.display
70 utils.header('Starting vmplayer for node %s on %s'%(hostname,display))
71 os.system('set -x; cd %s/vmplayer-%s ; DISPLAY=%s vmplayer node.vmx < /dev/null >/dev/null 2>/dev/null &'%(path,hostname,display))
73 def start_qemu (self, options):
74 utils.header ("TestNode.start_qemu: not implemented yet")