流量測試

1.第三方工具:安測試和流量包應用

2.先用adb shell ps | grep 包名獲得進程id,adb shell cat /proc/"+pid+"/net/dev 獲得接收和發送數據流量值,操作被測試應用業務,再次輸入命令獲得接收和發送流量,兩次相減獲得消耗的流量

import csv
import os
import string
import time
import decimal


# 控制類
class Controller(object):
    def __init__(self, count):
        # 定義測試的次數
        self.counter = count
        # 定義收集數據的數組
        self.alldata = [("timestamp", "traffic")]

    # 單次測試過程
    def testprocess(self):
        # 執行獲取進程的命令
        result = os.popen("adb shell ps | findstr com.heytap.browser")
        # 獲取進程ID
        pid = result.readlines()[0].split(" ")[5]

        # 獲取進程ID使用的流量
        traffic = os.popen("adb shell cat /proc/" + pid + "/net/dev")
        for line in traffic:
            if "eth0" in line:
                # 將所有空行換成#
                line = "#".join(line.split())
                # 按#號拆分,獲取收到和發出的流量
                receive = line.split("#")[1]
                transmit = line.split("#")[9]
                print(receive)
                print(transmit)
            elif "eth1" in line:
                # 將所有空行換成#
                line =  "#".join(line.split())
                # 按#號拆分,獲取收到和發出的流量
                receive2= line.split("#")[1]
                transmit2 = line.split("#")[9]

        # #計算所有流量的之和
        alltraffic = int(receive) + int(transmit)
        print(alltraffic)
        # 按KB計算流量值
        alltraffic = alltraffic / 1024
        # 獲取當前時間
        currenttime = self.getCurrentTime()
        # 將獲取到的數據存到數組中
        self.alldata.append((currenttime, alltraffic))

    # 多次測試過程控制
    def run(self):
        while self.counter > 0:
            self.testprocess()
            self.counter = self.counter - 1
            # 每5秒鐘采集一次數據

            time.sleep(5)

    # 獲取當前的時間戳
    def getCurrentTime(self):
        currentTime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        return currentTime

    # 數據的存儲
    def SaveDataToCSV(self):
        csvfile = open('traffic.csv', 'w')
        writer = csv.writer(csvfile)
        writer.writerows(self.alldata)
        csvfile.close()


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