在上一篇文章中，介紹了運(yùn)行一個(gè)Locust測(cè)試腳本的完整過(guò)成，包括設(shè)置運(yùn)行場(chǎng)景、運(yùn)行命令、運(yùn)行中的數(shù)據(jù)監(jiān)控等內(nèi)容，接下來(lái)的本文中將會(huì)介紹一個(gè)完整的測(cè)試腳本如何編寫(xiě)。

　　首先，看一段完整的測(cè)試腳本：

　　import time

　　from locust import HttpUser, task, between

　　class QuickstartUser(HttpUser):

　　wait_time = between(1. 5)

　　@task

　　def hello_world(self):

　　self.client.get("/hello")

　　self.client.get("/world")

　　@task(3)

　　def view_items(self):

　　for item_id in range(10):

　　self.client.get(f"/item?id={item_id}", name="/item")

　　time.sleep(1)

　　def on_start(self):

　　self.client.post("/login", json={"username":"foo", "password":"bar"})

　　腳本的內(nèi)容對(duì)于新手來(lái)說(shuō)，可能真的還是有點(diǎn)難度的，不過(guò)也沒(méi)關(guān)系，接下來(lái)一一的解釋一番。

　　1、導(dǎo)入常用的包

　　Locust測(cè)試腳本也是正常的Python代碼，也需要導(dǎo)入各種程序包才能很好的運(yùn)行，所以，第一步就是導(dǎo)入包：

　　import time

　　from locust import HttpUser, task, between

　　以上的包就是time包、Locust包中的部分模塊。

　　HttpUser模塊的作用是模擬用戶(hù)發(fā)送HTTP請(qǐng)求;

　　Time模塊是為了使用休眠等時(shí)間控制函數(shù);

　　Task模塊是為了使用Locust框架的任務(wù)模塊

　　Between模塊是為了設(shè)置等待時(shí)間的區(qū)間的;。

　　2、定義測(cè)試類(lèi)

　　這里為將要模擬的用戶(hù)定義一個(gè)類(lèi)。它從HttpUser繼承，HttpUser給每個(gè)用戶(hù)一個(gè)client屬性，這是HttpSession的一個(gè)實(shí)例，可以用來(lái)向想要加載測(cè)試的目標(biāo)系統(tǒng)發(fā)出HTTP請(qǐng)求。當(dāng)測(cè)試開(kāi)始時(shí)，locust將為它模擬的每個(gè)用戶(hù)創(chuàng)建該類(lèi)的實(shí)例，每個(gè)用戶(hù)將開(kāi)始在自己的綠色gevent線(xiàn)程中運(yùn)行。

　　class QuickstartUser(HttpUser):

　　一個(gè)文件要成為一個(gè)有效的locustfile，它必須包含至少一個(gè)繼承自User的類(lèi)。

　　3、定義wait_time

　　它將使模擬用戶(hù)在每個(gè)任務(wù)執(zhí)行后等待1到5秒。

　　wait_time = between(1. 5)

　　接下來(lái)，用@task裝飾的方法是Locust文件的核心。對(duì)于每個(gè)運(yùn)行的用戶(hù)，Locust創(chuàng)建一個(gè)greenlet(微線(xiàn)程)，該線(xiàn)程將調(diào)用這些方法。

　　4、定義方法

　　通過(guò)用@task裝飾兩個(gè)方法，案例中已經(jīng)聲明了兩個(gè)任務(wù)，其中一個(gè)被賦予了更高的權(quán)重(3)。當(dāng)QuickstartUser運(yùn)行時(shí)，它會(huì)選擇一個(gè)已聲明的任務(wù)——在本例中是hello_world或view_items——并執(zhí)行它。任務(wù)是隨機(jī)挑選的，但可以給它們不同的權(quán)重。

　　上述配置將使Locust選擇view_items的可能性是hello_world的三倍。當(dāng)一個(gè)任務(wù)完成執(zhí)行時(shí)，User將在它的等待時(shí)間(在本例中是1到5秒)中休眠。在它的等待時(shí)間之后，它會(huì)選擇一個(gè)新任務(wù)并不斷重復(fù)。

　　@task

　　def hello_world(self):

　　...

　　@task

　　def hello_world(self):

　　self.client.get("/hello")

　　self.client.get("/world")

　　@task(3)

　　def view_items(self):

　　...

　　注意，只有裝飾了@task的方法才會(huì)被選中，所以可以任意定義自己的內(nèi)部助手方法。

　　5、發(fā)送請(qǐng)求

　　self.client.get("/hello")

　　self.client屬性使得發(fā)出將被Locust記錄的HTTP調(diào)用成為可能。案例中使用get方法發(fā)送請(qǐng)求。

　　HttpUser不是一個(gè)真正的瀏覽器，因此不會(huì)解析HTML響應(yīng)來(lái)加載資源或呈現(xiàn)頁(yè)面。但它會(huì)跟蹤cookie。

　　@task(3)

　　def view_items(self):

　　for item_id in range(10)

　　self.client.get(f"/item?id={item_id}", name="/item")

　　time.sleep(1)

　　6、設(shè)置多參數(shù)運(yùn)行

　　在view_items任務(wù)中，使用一個(gè)可變的查詢(xún)參數(shù)加載10個(gè)不同的url。為了不在Locust的統(tǒng)計(jì)數(shù)據(jù)中獲得10個(gè)單獨(dú)的條目(因?yàn)榻y(tǒng)計(jì)數(shù)據(jù)是根據(jù)URL分組的)，使用name參數(shù)將所有這些請(qǐng)求分組到一個(gè)名為“/item”的條目下。

　　def on_start(self):

　　self.client.post("/login", json={"username":"foo", "password":"bar"})

　　此外，案例還聲明了一個(gè)on_start方法。當(dāng)每個(gè)模擬用戶(hù)啟動(dòng)時(shí)，將調(diào)用具有此名稱(chēng)的方法。

　　更多關(guān)于軟件測(cè)試培訓(xùn)的問(wèn)題，歡迎咨詢(xún)千鋒教育在線(xiàn)名師，如果想要了解我們的師資、課程、項(xiàng)目實(shí)操的話(huà)可以點(diǎn)擊咨詢(xún)課程顧問(wèn)，獲取試聽(tīng)資格來(lái)試聽(tīng)我們的課程，在線(xiàn)零距離接觸千鋒教育大咖名師，讓你輕松從入門(mén)到精通。