腳本之家服務器常用軟件

快捷導航

淺析Jmeter多用戶token使用問題

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

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

背景

在測試的時候，經(jīng)常會有模擬用戶登錄，拿到用戶 token 后再去請求接口的場景。

這個模擬用戶登錄就會分為兩種，一種是單用戶，另一種是多用戶。

日常自動化測試的時候可能一個用戶對應 n 個用例就可以滿足大多數(shù)場景；

如果是在壓力測試的場景下面，可能就會略顯單調(diào)，也無法滿足一些真實業(yè)務場景。

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

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

下面來看一個具體的例子來了解一下。

場景接口

在這里的話，只有兩個接口，一個是登錄拿 token，一個是有 token 才能請求的。

下面是各接口定義

登錄接口

請求：

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

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

響應：

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

業(yè)務接口

請求：

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

響應：

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

登錄接口處理

登錄接口屬于預請求，所以我們一般會選擇把它放在 setUp 線程組里面。

我們需要準備一個 csv 文件，里面用來存放需要登錄的用戶名和密碼。

接下來就是把這個 csv 配置好，定義了兩個變量 account 和 pwd

然后是把登錄的 HTTP 請求配置好：

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

到這里就要開始注意了?。。。?/p>

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

換句話就是說，每登錄一個用戶，這個 access_token 的值就會是最后一個登錄的用戶的 token，。

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

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

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

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

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 寫進到 csv 文件里面，這個文件在的位置是 jmeter 的目錄。

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

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

業(yè)務接口處理

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

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

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

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

然后就是配置 HTTP 請求了

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

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

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

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

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

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

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

這個時候跑腳本就基本沒什么問題了。