压力测试之python自动化实践

# encoding:utf-8
import csv
import os
import time
#APP类

class App(object):
    def __init__(self):
        self.content = ""
        self.startTime = 0

    def LaunchApp(self):
        #启动APP
        cmd = "adb shell am start  -W  -n  com.zgjky.app/.activity.welcomepage.Welcome_Activity "
        self.content= os.popen(cmd)

    def StopApp(self):
        #停止APP
        cmd = "adb shell am force-stop com.zgjky.app"
        os.popen(cmd)

    #获取启动时间
    def GetLaunchedTime(self):
        for line in self.content.readlines():
            if "ThisTime" in line:
                self.startTime = line.split(":")[1]
                break
        return self.startTime

#控制类
class Controller(object):
    def __init__(self,count):
        self.app = App()
        self.counter = count
        self.alldata = [("timestamp","elpasedtime")]

    #单次测试过程
    def testprocess(self):
        self.app.LaunchApp()
        time.sleep(5)
        elpasedtime = self.app.GetLaunchedTime()
        self.app.StopApp()
        time.sleep(3)
        currenttime = self.getCurrentTime()
        self.alldata.append((currenttime,elpasedtime))

    #多次测试过程
    def run(self):
        while self.counter > 0:
            self.testprocess()
            self.counter = self.counter - 1

     #获取当前时间戳
    def getCurrentTime(self):
        currentTime = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())
        return currentTime

    #数据的存贮
    def SaveDataToCSV(self):
        csvfile = file("startTime.csv","wb")
        writer = csv.writer(csvfile)
        writer.writerows(self.alldata)
        csvfile.close()

if __name__ == "__main__":
    controller = Controller(11)
    controller.run()
    controller.SaveDataToCSV()