快捷導(dǎo)航

jmeter執(zhí)行python腳本的實現(xiàn)示例

更新時間：2022年05月30日 08:37:24 作者：咲之

本文主要介紹了jmeter執(zhí)行python腳本的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

概要

本文主要介紹一個通過jmeter執(zhí)行python腳本的測試場景。主要過程如下：

在jmeter中設(shè)置參數(shù)；
通過jmeter修改py模板文件，并創(chuàng)建后續(xù)需執(zhí)行的py腳本；
通過jmeter執(zhí)行py腳本文件，獲得py文件的執(zhí)行輸出結(jié)果，并設(shè)置為變量；
使用變量進(jìn)行后續(xù)測試。

提示：以下為本篇文章正文內(nèi)容，案例僅供參考。

一、環(huán)境準(zhǔn)備

1. 測試工具

jmeter、python，與版本無關(guān)，任意版本均可。

2. 測試資源

py模板文件：
1）自定義需要在py模板文件中修改的參數(shù)，比如：我們想在jmeter中修改py腳本中某些變量的初始值，我們可以在采用特殊標(biāo)記[變量名]的方式標(biāo)記其為可替換變量；
2）使用print()方法打印輸出目標(biāo)參數(shù)。也可以采用其他方法，比如將多個變量值寫入到csv文件中，后續(xù)可以使用csv文件進(jìn)行參數(shù)化。
圖-變量設(shè)置：[times]、[modelName]、[modelType]
圖-輸出設(shè)置：print(filepath)

在這里插入圖片描述

二、實現(xiàn)過程

1. 在jmeter中設(shè)置參數(shù)

如圖，可以選擇添加“前置處理器>>用戶參數(shù)”，用于設(shè)置相關(guān)變量的初始值；

在這里插入圖片描述

2. 通過jmeter修改py模板

添加“取樣器>>BeanShell取樣器”，編寫讀取文件、寫入文件代碼。主要思路：讀取模板文件內(nèi)容，替換目標(biāo)字符后，寫入到新文件中；

// 模板文件位置
String read_path = vars.get("dir") + File.separator + "files\\mkxml\\template_demo.py";
// 輸出文件位置
String write_path = vars.get("dir") + File.separator + "files\\mkxml\\demo.py";

// 讀取模板文件內(nèi)容
StringBuffer res = new StringBuffer();
String line = null;
File read_file = new File(read_path);
BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(read_file), "UTF-8"));
while ((line = reader.readLine()) != null) {
	res.append(line + "\n");
}
reader.close();

// 處理模板內(nèi)容，替換相關(guān)參數(shù)
String content = res.toString();
content = content.replace("[times]", vars.get("times"));
content = content.replace("[modelName]", vars.get("modelName"));
content = content.replace("[modelType]", vars.get("modelType"));

// 寫入內(nèi)容到輸出文件位置
File write_file = new File(write_path);
BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(write_file), "UTF-8"));
writer.write(content);
writer.flush();
writer.close();

在這里插入圖片描述

3. 通過jmeter執(zhí)行py腳本

添加“取樣器>>BeanShell取樣器”，通過cmd命令執(zhí)行py腳本，并獲得窗口輸出信息；

// 以cmd命令行方式調(diào)用py腳本，&&為多組命令連接符
String cmd = "cmd /c cd /d "+ vars.get("dir") + File.separator + "files\\mkxml && python demo.py";
Runtime rt = Runtime.getRuntime();
Process pr = rt.exec(cmd);

pr.waitFor();

BufferedReader br = new BufferedReader(new InputStreamReader(pr.getInputStream()));
String line = "";
StringBuilder response = new StringBuilder();

while((line = br.readLine()) != null) {
	response.append(line);
}

br.close();

// 設(shè)置成可在jmeter中使用的變量，filepath為目標(biāo)XML文件所在位置
vars.put("filepath", response.toString());

在這里插入圖片描述