Python多线程协同(C/P)完成设备[反复重启]疲劳测试

Python多线程协同(C/P)完成设备[反复重启]疲劳测试

发布于：2014-4-8 15:03:26来自【群】                 测试部                     编辑

上周，看到部门同事在做设备反复重启的qtp脚本，很显然，在ui层面完成硬件设备的反复重启是极其痛苦的，原因有如下几点：

1. 在ui层完成对设备的重启本身具有高难度，启用的都是低级录制，失败率达到90%。

2. 硬件设备反复重启这个事情如果在ui层做有一个最大的弊端就是，永远无法知道设备什么时候起来了，我要继续重启它。

ui层的设备反复重启？！简直就是坑爹，为此，我们必须改造这个非常笨拙的测试方法。

反复重启设备，从代码角度来说，简直就是一个活脱脱的多线程协同模型（c/p－消费者，生产者）

我的思路如下：

1.开启两个线程，一个线程专门用来嗅探设备状态（并全局一个state）

2.两个线程互相通知，分别负责连接设备和重启设备

3.线程间共享一个state

4.利用python强大的paramiko模块

使用方法：

1.直接复制下面的代码到xxx.py

2.运行方式python xxx.py

使用前需要安装：

paramiko，由于我是在mac下配置的，这里不给出教程了，paramiko安装前需要手动编译安装pycrypto。

代码如下：

# -*- coding:utf-8 -*-

#About

__author__ = 'archer'

__version__=1.0.0

'''

设备重启长拷验证

'''

from threading import Condition,Thread

from time import sleep as wait

import paramiko

#condition需要全局共享,以保证线程之间顺利工作

condition=Condition()

#sshclient

devicecon=paramiko.SSHClient()

#设备默认状态为关闭

devicestate=False

class MyError(paramiko.SSHException):

    pass

#设备通知线程

class DeviceNotify(Thread):

    def run(self):

        global devicestate

        while True:

            #拿锁

            condition.acquire()

            #如果设备处于开启状态,则通知DeviceDoReboot

            if devicestate:

                condition.notify()

                condition.release()

            else:

                print [Thread-Notify]设备状态:关闭;正在等待设备重启......

                try:

                    #set_missing_host_Key

                    devicecon.set_missing_host_key_policy(paramiko.AutoAddPolicy())

                    devicecon.connect('172.16.121.140',2222,root,kedacom,timeout=6000)

                except:

                    pass

                #等待时间

                wait(10)

                #将设备置为开并释放锁

                devicestate=True

                condition.release()

class DeviceDoReboot(Thread):

    def run(self):

        global devicestate

        while True:

            #拿锁

            condition.acquire()

            #如果设备处于开启状态,则进行关闭,

            if devicestate:

                print [Thread-DeviceReboot]线程已获取通知,正在重启设备......

                #发送重启命令

                try:

                    devicecon.exec_command(reboot)

                except:

                    pass

                #重启命令发送之后等待30秒,可以手动控制长拷频率

                wait(30)

                #将设备状态置为关闭

                devicestate=False

                #释放

                condition.release()

            else:

                #通知DeviceNotify线程

                condition.notify()

                condition.release()

if __name__==__main__:

    DeviceNotify().start()

    DeviceDoReboot().start()