使用STAF进行自动化安装测试

首先安装STAF

在实体机和目标机器中均需要安装，STAF拥有对Python支持的库函数，可以直接调用，比较方便，所以测试脚本决定用Python来写。

需要注意的是，在/staf/bin中有一个STAF.cfg文件需要修改server和client端的trustlevel = 5.注意不同的command的trust level的要求不一样，所以在我的测试环境里面直接设定成为5比较方便。

安装之后，需要将STAF在实验机中设定为开机自启动。以便可以直接进行连接操作，不过我是使用虚拟机进行测试，只要直接载入一个开机的snapshot就可以了。

具体可以参照这个网址的开机自启动设置。

点击打开链接

这次测试环境为windows，方法是目标机要从server上面检查并且下载一个安装文件，并且执行安装。所以，windows 必须执行静默安装，并且通过STAF执行ftp命令时候需要通过载入一个conf文件来执行下载任务。

同时在Python中需要操作exsi，这个API可以直接Google到。Python真的很方便额。。。

需要注意的是在操作时候，安装和check的代码之间需要有一个sleep等待，以保证client端执行成功。

#-------------------------------------------------------------------------------
# Name:        STAF_test
# Purpose:     Used for staf automation test. rollback a snapshot, download and
#              install a program from server
#
# Author:      Levin
#
# Requirement: You need to install the Staf on your local machine and add your
#              machine's ip to the */staf/bin/STAF.cfg. And add the */staf/bin
#              to your python path.
#
# Created:     25/08/2014
# Copyright:   (c) Levin 2014
# Licence:     <your licence>
#-------------------------------------------------------------------------------

from PySTAF import *
import pysphere
from xml.etree import ElementTree as ET
import sys
from time import sleep

##-----------Prepare the variables need to use-----------------##
nodeall = ET.parse('config.xml')
IPnode = nodeall.find('SETTING')
EXSIIP = IPnode.getchildren()[0].text
UserName = IPnode.getchildren()[1].text
PassWord = IPnode.getchildren()[2].text
ClientIP = IPnode.getchildren()[3].text
VirtualMachine = IPnode.getchildren()[4].text
Snapshot = IPnode.getchildren()[5].text

Pathnode = nodeall.find('WORKINGPATH')
ClientPath = Pathnode.getchildren()[0].text
LocalPath = Pathnode.getchildren()[1].text


##------------------installation process------------##
def staf_install():
    try:
        handle = STAFHandle("test")
    except STAFException, e:
        print "Error registering with STAF, RC: %d" % e.rc
        sys.exit(e.rc)

    while (True):
        result = handle.submit(ClientIP, "ping", "ping")
        if (result.rc == 0):
            break
        print "Error submitting request, RC: %d, Result: %s" % (result.rc, result.result)
    print "connected"

    result = handle.submit("local", "FS", "COPY File "+LocalPath+r"\ftpSample.conf TODIRECTORY "+ClientPath+" TOMACHINE "+ClientIP)
    if (result.rc == 0):
        print "Upload config file succeed!"
    else :
        print "Upload config file failed!"
        print "%s" % result.result
        sys.exit(1)
    result = handle.submit(ClientIP, "process", "start shell command" +  "\"ftp -s:ftpSample.conf\"" + "workdir " + ClientPath)
    if (result.rc == 0):
        print "Download install file succeed!"
    else :
        print "Download install file failed!"
        print "%s" % result.result
        sys.exit(2)
    print "Now wait 6 seconds"
    sleep(6)
    result = handle.submit(ClientIP,"process", "start shell command" + "\"ActivePerl-5.16.3.1604-MSWin32-x86-298023.msi/qn\""+ " workdir " + ClientPath)
    print "Now wait 23 seconds for installation complete"
    sleep(23)
    result = handle.submit(ClientIP,"fs", "list directory " + r"C:\Perl\bin")
    if ("perl.exe" in result.resultObj):
        print "Installation succeed!"
    else :
        print "Installation failed!"
        sys.exit(3)

##--------------reset the exsi to clean the environment-------##
def Reset_Virtualmachine():
    server = pysphere.VIServer()
    try :
        server.connect(EXSIIP, UserName, PassWord)
    except pysphere.VIException, e :
        print "Can not connect to exsi. Exit now!"
        sys.exit(e.rc)
    print "Connected to EXSI"
    vm = server.get_vm_by_name("win2k3-R2-32bits")
    vm.revert_to_named_snapshot(Snapshot)

if __name__ == '__main__':
    Reset_Virtualmachine()
    print "Enter to Staf install."
    staf_install()