"""
Description:
This file is used to setup the running environment
Include Download code,setup environment variable
Set onos running config
Set user name/password
Onos-push-keys and so on
lanqinglong@huawei.com
"""
import os
import time
import pexpect
import re
import sys
import pxssh
from connection import connection
class environment( connection ):
def __init__( self ):
connection.__init__( self )
self.loginfo = connection( )
self.masterhandle = ''
self.home = ''
def DownLoadCode( self, handle, codeurl ):
"""
Download Code use 'git clone'
parameters:
handle: current working handle
codeurl: clone code url
"""
print "Now loading test codes! Please wait in patient..."
originalfolder = sys.path[0]
print originalfolder
gitclone = handle
gitclone.sendline( "git clone " + codeurl )
index = 0
increment = 0
while index != 1 or index != 4:
index = gitclone.expect ( ['already exists', 'esolving deltas: 100%', \
'eceiving objects', 'Already up-to-date', \
'npacking objects: 100%', pexpect.EOF] )
filefolder = self.home + '/' + codeurl.split('/')[-1].split('.')[0]
if index == 0 :
os.chdir( filefolder )
os.system( 'git pull' )
os.chdir( originalfolder )
self.loginfo.log( 'Download code success!' )
break
elif index == 1 or index == 4:
self.loginfo.log( 'Download code success!' )
gitclone.sendline( "mkdir onos" )
gitclone.prompt( )
gitclone.sendline( "cp -rf " + filefolder+ "/tools onos/" )
gitclone.prompt( )
break
elif index == 2 :
os.write(1, gitclone.before)
sys.stdout.flush()
else :
self.loginfo.log( 'Download code failed!' )
self.loginfo.log( 'Information before' + gitclone.before )
break
gitclone.prompt( )
def InstallDefaultSoftware( self, handle ):
"""
Install default software
parameters:
handle(input): current working handle
"""
print "Now Cleaning test environment"
handle.sendline("sudo apt-get install -y mininet")
handle.prompt( )
handle.sendline("sudo pip install configobj")
handle.prompt( )
handle.sendline("sudo apt-get install -y sshpass")
handle.prompt( )
handle.sendline("OnosSystemTest/TestON/bin/cleanup.sh")
handle.prompt( )
time.sleep(5)
self.loginfo.log( 'Clean environment success!' )
def OnosPushKeys(self, handle, cmd, password):
"""
Using onos-push-keys to make ssh device without password
parameters:
handle(input): working handle
cmd(input): onos-push-keys xxx(xxx is device)
password(input): login in password
"""
print "Now Pushing Onos Keys:"+cmd
Pushkeys = handle
Pushkeys.sendline( cmd )
Result = 0
while Result != 2:
Result = Pushkeys.expect( ["(yes/no)", "assword:", "PEXPECT]#", \
pexpect.EOF, pexpect.TIMEOUT])
if ( Result == 0 ):
Pushkeys.sendline( "yes" )
if ( Result == 1 ):
Pushkeys.sendline( password )
if ( Result == 2 ):
self.loginfo.log( "ONOS Push keys Success!" )
break
if ( Result == 3 ):
self.loginfo.log( "ONOS Push keys Error!" )
break
time.sleep(2)
Pushkeys.prompt( )
print "Done!"
def SetOnosEnvVar( self, handle, masterpass, agentpass):
"""
Setup onos pushkeys to all devices(3+2)
parameters:
handle(input): current working handle
masterpass: scripts running server's password
agentpass: onos cluster&compute node password
"""
print "Now Setting test environment"
for host in self.hosts:
print "try to connect " + str(host)
result = self.CheckSshNoPasswd(host)
if not result:
print "ssh lgin failed,try to copy master publickey to agent " + str(host)
self.CopyPublicKey(host)
self.OnosPushKeys( handle, "onos-push-keys " + self.OCT, masterpass)
self.OnosPushKeys( handle, "onos-push-keys " + self.OC1, agentpass)
self.OnosPushKeys( handle, "onos-push-keys " + self.OC2, agentpass)
self.OnosPushKeys( handle, "onos-push-keys " + self.OC3, agentpass)
self.OnosPushKeys( handle, "onos-push-keys " + self.OCN, agentpass)
self.OnosPushKeys( handle, "onos-push-keys " + self.OCN2, agentpass)
def CheckSshNoPasswd( self, host):
"""
Check master can connect agent with no password
"""
login = pexpect.spawn( "ssh " + str(host))
index = 4
while index == 4:
index = login.expect(['(yes/no)','>|#|\$', \
pexpect.EOF, pexpect.TIMEOUT] )
if index == 0:
login.sendline( "yes" )
index = 4
if index == 1:
self.loginfo.log("ssh connect to " + str(host) + " success,no need to copy ssh public key" )
return True
login.interact()
return False
def ChangeOnosName( self, user, password):
"""
Change onos name in envDefault file
Because some command depend on this
parameters:
user: onos&compute node user
password: onos&compute node password
"""
print "Now Changing ONOS name&password"
filepath = self.home + '/onos/tools/build/envDefaults'
line = open(filepath, 'r').readlines()
lenall = len(line) - 1
for i in range(lenall):
if "ONOS_USER=" in line[i]:
line[i]=line[i].replace("sdn",user)
if "ONOS_GROUP" in line[i]:
line[i]=line[i].replace("sdn",user)
if "ONOS_PWD" in line[i]:
line[i]=line[i].replace("rocks",password)
NewFile = open(filepath, 'w')
NewFile.writelines(line)
NewFile.close
print "Done!"
def ChangeTestCasePara(self, testcase, user, password):
"""
When running test script, there's something need \
to change in every test folder's *.param & *.topo files
user: onos&compute node user
password: onos&compute node password
"""
print "Now Changing " + testcase + " name&password"
if self.masterusername == 'root':
filepath = '/root/'
else :
filepath = '/home/' + self.masterusername + '/'
filepath = filepath +"OnosSystemTest/TestON/tests/" + testcase + "/" + \
testcase + ".topo"
line = open(filepath,'r').readlines()
lenall = len(line)-1
for i in range(lenall-2):
if ("localhost" in line[i]) or ("OCT" in line[i]):
line[i+1]=re.sub(">\w+",">"+user,line[i+1])
line[i+2]=re.sub(">\w+",">"+password,line[i+2])
if "OC1" in line [i] \
or "OC2" in line [i] \
or "OC3" in line [i] \
or "OCN" in line [i] \
or "OCN2" in line[i]:
line[i+1]=re.sub(">\w+",">root",line[i+1])
line[i+2]=re.sub(">\w+",">root",line[i+2])
NewFile = open(filepath,'w')
NewFile.writelines(line)
NewFile.close
def SSHlogin ( self, ipaddr, username, password ) :
"""
SSH login provide a connection to destination.
parameters:
ipaddr: ip address
username: login user name
password: login password
return: handle
"""
login = pxssh.pxssh( )
login.login ( ipaddr, username, password, original_prompt='[$#>]')
#send command ls -l
login.sendline ('ls -l')
#match prompt
login.prompt()
print ("SSH login " + ipaddr + " success!")
return login
def SSHRelease( self, handle ):
#Release ssh
handle.logout()
def CopyOnostoTestbin( self ):
sourcefile = self.cipath + '/dependencies/onos'
destifile = self.home + '/onos/tools/test/bin/'
os.system( 'pwd' )
runcommand = 'cp ' + sourcefile + ' ' + destifile
os.system( runcommand )
def CopyPublicKey( self, host ):
output = os.popen( 'cat /root/.ssh/id_rsa.pub' )
publickey = output.read().strip('\n')
tmphandle = self.SSHlogin( self.installer_master, self.installer_master_username, self.installer_master_password )
tmphandle.sendline("ssh "+ host + " -T \'echo " + str(publickey) + ">>/root/.ssh/authorized_keys\'" )
tmphandle.prompt()
self.SSHRelease(tmphandle)
print "Add OCT PublicKey to " + host + " success"
def OnosEnvSetup( self, handle ):
"""
Onos Environment Setup function
"""
self.Gensshkey( handle )
self.home = self.GetEnvValue( handle, 'HOME' )
self.AddKnownHost( handle, self.OC1, "karaf", "karaf" )
self.AddKnownHost( handle, self.OC2, "karaf", "karaf" )
self.AddKnownHost( handle, self.OC3, "karaf", "karaf" )
self.DownLoadCode( handle, 'https://github.com/sunyulin/OnosSystemTest.git' )
#self.DownLoadCode( handle, 'https://gerrit.onosproject.org/onos' )
if self.masterusername == 'root':
filepath = '/root/'
else :
filepath = '/home/' + self.masterusername + '/'
self.OnosRootPathChange( filepath )
self.CopyOnostoTestbin()
self.ChangeOnosName(self.agentusername,self.agentpassword)
self.InstallDefaultSoftware( handle )
self.SetOnosEnvVar(handle, self.masterpassword,self.agentpassword)