在上一篇文章中，介紹了運行一個Locust測試腳本的完整過成，包括設置運行場景、運行命令、運行中的數據監(jiān)控等內容，接下來的本文中將會介紹一個完整的測試腳本如何編寫。

　　首先，看一段完整的測試腳本：

　　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"})

　　腳本的內容對于新手來說，可能真的還是有點難度的，不過也沒關系，接下來一一的解釋一番。

　　1、導入常用的包

　　Locust測試腳本也是正常的Python代碼，也需要導入各種程序包才能很好的運行，所以，第一步就是導入包：

　　import time

　　from locust import HttpUser, task, between

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

　　HttpUser模塊的作用是模擬用戶發(fā)送HTTP請求;

　　Time模塊是為了使用休眠等時間控制函數;

　　Task模塊是為了使用Locust框架的任務模塊

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

　　2、定義測試類

　　這里為將要模擬的用戶定義一個類。它從HttpUser繼承，HttpUser給每個用戶一個client屬性，這是HttpSession的一個實例，可以用來向想要加載測試的目標系統發(fā)出HTTP請求。當測試開始時，locust將為它模擬的每個用戶創(chuàng)建該類的實例，每個用戶將開始在自己的綠色gevent線程中運行。

　　class QuickstartUser(HttpUser):

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

　　3、定義wait_time

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

　　wait_time = between(1. 5)

　　接下來，用@task裝飾的方法是Locust文件的核心。對于每個運行的用戶，Locust創(chuàng)建一個greenlet(微線程)，該線程將調用這些方法。

　　4、定義方法

　　通過用@task裝飾兩個方法，案例中已經聲明了兩個任務，其中一個被賦予了更高的權重(3)。當QuickstartUser運行時，它會選擇一個已聲明的任務——在本例中是hello_world或view_items——并執(zhí)行它。任務是隨機挑選的，但可以給它們不同的權重。

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

　　@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的方法才會被選中，所以可以任意定義自己的內部助手方法。

　　5、發(fā)送請求

　　self.client.get("/hello")

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

　　HttpUser不是一個真正的瀏覽器，因此不會解析HTML響應來加載資源或呈現頁面。但它會跟蹤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、設置多參數運行

　　在view_items任務中，使用一個可變的查詢參數加載10個不同的url。為了不在Locust的統計數據中獲得10個單獨的條目(因為統計數據是根據URL分組的)，使用name參數將所有這些請求分組到一個名為“/item”的條目下。

　　def on_start(self):

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

　　此外，案例還聲明了一個on_start方法。當每個模擬用戶啟動時，將調用具有此名稱的方法。

　　更多關于軟件測試培訓的問題，歡迎咨詢千鋒教育在線名師，如果想要了解我們的師資、課程、項目實操的話可以點擊咨詢課程顧問，獲取試聽資格來試聽我們的課程，在線零距離接觸千鋒教育大咖名師，讓你輕松從入門到精通。