Locust performance test 2 (interface request)

Have been to Loust With a basic understanding of , Next, you can make an interface request

One 、 Post request

The user login

According to business needs , The login interface is executed only once , stay on_start() Function to write login information

class MyTasks(TaskSet):
    url = '/login'
    headers = {
    'Content-Type': 'application/json'}

def on_start(self):
	self.data = {
    "username":"admin","password":"123456"}
	respon = self.client.post(self.url, headers = self.headers, data=json.dumps(self.data), name=' The user login ', verify=False, timeout=10)
	resp_dict = respon.json()
	
	# print(f' The response data is :{resp_dict}')
	if respon.status_code == 200:
		#  Interface success 
		print(resp_dict['msg'])
	else:
		#  request was aborted 
		respon.failure(resp_dict['msg'])

class User(HttpUser):
    tasks = [MyTasks]
    host = 'https://www.baidu.com'
    wait_time = between(2,2)	#  Every time 2 Once per second 

Two 、 Get request

Send mobile phone verification code

from locust import HttpUser, TaskSet, task
''' get Request interface  '''

class MyTasks(TaskSet):
    url = '/xxx/xxx/sendAuthCode'
    headers = {
    'Content-Type': 'application/json',
               'Authorization': 'xxxxx'
               }
               
    def on_start(self):
        print(" User initialization -- Sign in ")
        
     @task
    def get_login(self):
        print(' Send verification code ...')
        
        #  Define an object property 
        self.query_data = {
    'mobile': 1579715000}

        respon = self.client.get(self.url, params = self.query_data, headers = self.headers, name=' Send verification code ', verify=False, timeout=10)
        resp_dict = respon.json()

        print(f' The response data is :{
      resp_dict}')