1 from locust import HttpLocust, task, TaskSet
 2 import os
 3 
 4 
 5 '''
 6 -f 指定效能測試指令碼檔案
 7 --host 指定被測試應用的URL的地址
 8 --port 埠設定，預設8089
 9 --no-web 表示不使用Web介面執行測試
10 -c 設定虛擬使用者數
11 -r 設定每秒啟動虛擬使用者數
12 -t 設定執行時間
13 '''
14 
15 
16 # ----------------使用locust進行效能測試----------------------
17 
18 
19 class
 
 UserBehavior(TaskSet):  # 定義一個使用者行為， 繼承TaskSet類
20 
21     @task  # @task裝飾該方法為一個事務
22     def test_czth(self):  # 表示一個使用者行為
23         print('--訪問拼多多主頁--')
24         url = '/'
25         with self.client.post(url, catch_response=True) as response:  # client.get()用於指請求的路徑“/”,首頁指定根路徑
26             print
 
(response.status_code)
27             if response.status_code != 200:
28                 response.failure('Failed!')
29             else:
30                 response.success()
31 
32     @task
33     def test_czth_login(self):
34         print('--訪問拼多多商品展示頁--')
35         payload = {'username': 'admin'
 
,
36                    'password': '123456',
37                    'validcode': '999999'}
38         url = '/home'
39 
40         with self.client.post(url=url, data=payload) as response:
41             print(response.status_code)
42             print(response.text)
43             assert '歡迎您, 系統管理員' in response.text
44 
45 
46 class WebsiteUser(HttpLocust):  # 用於設定效能測試， 繼承Httplocust類
47     task_set = UserBehavior  # 指向一個定義的使用者行為類
48     min_wait = 1000  # 使用者等待時間的下界（毫秒）
49     max_wait = 3000  # 使用者等待時間的上界
50     host = '192.168.106.211:8100'  # 指壓測地址
51 
52 
53 if __name__ == '__main__':
54     # os.system('locust -f locust_xn.py --port=8999')
55     os.system('locust -f locust_xn.py  --port=8999 --no-web -c 10000 -r 100 -t 1h')
56     # os.system('locust -f locust_xn.py  --master-bind-port=8999 --expect-slaves=X --slave ')