腳本之家服務(wù)器常用軟件

快捷導(dǎo)航

淺析Jmeter多用戶token使用問(wèn)題

更新時(shí)間：2021年10月18日 08:33:47 作者：Github

這篇文章主要介紹了Jmeter多用戶token使用問(wèn)題,通過(guò)具體的例子給大家介紹了Jmeter多用戶token使用場(chǎng)景接口分析，需要的朋友可以參考下

背景

在測(cè)試的時(shí)候，經(jīng)常會(huì)有模擬用戶登錄，拿到用戶 token 后再去請(qǐng)求接口的場(chǎng)景。

這個(gè)模擬用戶登錄就會(huì)分為兩種，一種是單用戶，另一種是多用戶。

日常自動(dòng)化測(cè)試的時(shí)候可能一個(gè)用戶對(duì)應(yīng) n 個(gè)用例就可以滿足大多數(shù)場(chǎng)景；

如果是在壓力測(cè)試的場(chǎng)景下面，可能就會(huì)略顯單調(diào)，也無(wú)法滿足一些真實(shí)業(yè)務(wù)場(chǎng)景。

對(duì)于單用戶的情況下，和我們常規(guī)的多接口有依賴的測(cè)試其實(shí)沒(méi)什么太大的差別。

所以這里主要講的是多用戶產(chǎn)生多個(gè) token 的情況。

下面來(lái)看一個(gè)具體的例子來(lái)了解一下。

場(chǎng)景接口

在這里的話，只有兩個(gè)接口，一個(gè)是登錄拿 token，一個(gè)是有 token 才能請(qǐng)求的。

下面是各接口定義

登錄接口

請(qǐng)求：

POST http://localhost:8532/MultiToken/Login
Content-Type: application/json

{
	"UserName":"catcherwong-1",
	"Password":"123"
}

響應(yīng)：

{"code":0,"msg":"ok","data":"catcherwong-1-token"}

業(yè)務(wù)接口

請(qǐng)求：

GET http://localhost:8532//MultiToken/do?account=xxx
Content-Type: application/json
token: catcherwong-1-token

響應(yīng)：

{"code":0,"msg":"ok","data":"catcherwong-1-token"}

登錄接口處理

登錄接口屬于預(yù)請(qǐng)求，所以我們一般會(huì)選擇把它放在 setUp 線程組里面。

我們需要準(zhǔn)備一個(gè) csv 文件，里面用來(lái)存放需要登錄的用戶名和密碼。

接下來(lái)就是把這個(gè) csv 配置好，定義了兩個(gè)變量 account 和 pwd

然后是把登錄的 HTTP 請(qǐng)求配置好：

由于后面要用到 token，所以要先把 token 提取出來(lái)，這里用的是 JSON Extractor。

到這里就要開(kāi)始注意了！?。?！

由于我們會(huì)有多個(gè)用戶進(jìn)行登錄，但是這一個(gè)提取操作每次都會(huì)把 token 賦值到 access_token 這個(gè)變量上面，是覆蓋的操作。

換句話就是說(shuō)，每登錄一個(gè)用戶，這個(gè) access_token 的值就會(huì)是最后一個(gè)登錄的用戶的 token，。

換個(gè)思路，每次它會(huì)覆蓋，那么把這些 token 存到一個(gè)地方，然后業(yè)務(wù)接口去這個(gè)地方取就可以了。

如果沒(méi)有用戶登錄這一步，給的直接是 token，那么我們也是直接把這個(gè) token 放到 csv 文件里面，然后讓 jmeter 去循環(huán)使用里面的 token。

那么要做的東西其實(shí)就很確定了，就是在提取到 token 后，把這個(gè) token 寫(xiě)到一個(gè) csv 文件里面。

要想做到這一步，需要在登錄接口后面加一個(gè)后置的處理。

String p1 = System.getProperty("user.dir");
String p2 = System.getProperty("file.separator");
String p3 = "user_token.csv";
String path = p1 + p2 + p3;

FileWriter fileWriter = new FileWriter(new File(path), true);
BufferedWriter writer = new BufferedWriter(fileWriter);
writer.append(vars.get("accout")+","+vars.get("access_token")+"\n");
writer.close();
fileWriter.close();

這段代碼的意思是，把用戶名和提取到的 access_token 寫(xiě)進(jìn)到 csv 文件里面，這個(gè)文件在的位置是 jmeter 的目錄。

這里是對(duì)文件路徑做了處理，可以適配所有操作系統(tǒng)的。不會(huì)出現(xiàn)說(shuō)指定了一個(gè) windows 系統(tǒng)的路徑，然后放到 linux 系統(tǒng)下面就跑不了了。

還有最重要的一個(gè)是，要修改 setUp 線程組的屬性，把循環(huán)次數(shù)改成 3 。因?yàn)榍懊娴?csv 文件里面有 3 個(gè)用戶，這樣它才會(huì)觸發(fā)三次登錄。

業(yè)務(wù)接口處理

業(yè)務(wù)接口要放到正常的線程組里面，獨(dú)立于 setUp 線程組。

前面提到，登錄后會(huì)有一個(gè) csv 文件，所以這里第一個(gè)要做的是把 csv 配置好。

上面的截圖用的是 ${__P(user.dir,)}${__P(file.separator,)}user_token.csv 這個(gè)文件路徑，這個(gè)在本地環(huán)境的 Jmeter 是可以通過(guò)的，不過(guò)在一些云服務(wù)上面是不行的，如阿里云 PTS 。

這里可以忽略前面的路徑，直接填寫(xiě) user_token.csv 即可，填這兩個(gè)，得到的文件路徑是一樣的，一個(gè)是絕對(duì)路徑一個(gè)是相對(duì)路徑。

然后就是配置 HTTP 請(qǐng)求了

PS：不要忘記把請(qǐng)求頭也配置了，這里就不截圖了。

這里試跑兩次，可以發(fā)現(xiàn)業(yè)務(wù)請(qǐng)求的接口，它的 token 請(qǐng)求頭每次都是不一樣的，在交替變化，這個(gè)是符合預(yù)期的。

但是會(huì)發(fā)現(xiàn)生成 csv 文件里面的數(shù)據(jù)會(huì)重復(fù)，沒(méi)有自動(dòng)清理掉上一次產(chǎn)生的數(shù)據(jù)。如果上一次產(chǎn)生的 token 過(guò)期了，那么用了這些過(guò)期的 token === 涼涼。

所以這里還有必要加一步 tearDown 線程組，每次跑完腳本把這個(gè)文件刪除掉。

String p1 = System.getProperty("user.dir");
String p2 = System.getProperty("file.separator");
String p3 = "user_token.csv";
String path = p1 + p2 + p3;

log.info("準(zhǔn)備刪除文件:" + path);

File file = new File(path);
if (!file.exists()) {
  log.info("刪除文件失敗:" + path + "不存在！");
} else {
  file.delete();
}

這個(gè)時(shí)候跑腳本就基本沒(méi)什么問(wèn)題了。