[ Locust ] Locust 사용법(windows)

1. python 3.6버젼 이상 설치

2. locust 설치

python -m pip install locust

설치확인

locust --version

3. locust 실행

locust를 실행할 경로에 locustfile.py파일 생성

import time,re,urllib3
from locust import HttpUser, task, between, TaskSet, SequentialTaskSet

class SequenceOfTasks(SequentialTaskSet):
    wait_time = between(1, 1)
    #session_text = ''


class QuickstartUser(HttpUser):
    wait_time = between(1, 1)

    def on_start(self):
        # 로그인시 https url(https://g-yangsan.gen.es.kr:451/login.php)이 호출되는데 실제 ssl인증서를 통한 로그인이 아니여서 self.client.proxies 세팅을 "https":"http://localhost:8887"로 했을때
        # fiddler에 display가 정상적으로 되었음.
        #self.client.proxies = {"http": "http://localhost:8887", "https": "https://localhost:8887"}
        #self.client.proxies = {"http": "http://localhost:8887"}
        #self.client.proxies = {"https": "http://localhost:8887"}
        #self.client.verify = False

        response = self.client.get("/")
        print(response.headers["Set-Cookie"])
        # print(response.content)
        # print(response.text)
        p = re.compile('SESSIONID=(.+?);')
        session = p.search(response.headers["Set-Cookie"]).group(1)
        print(session)
        self.session_text = 'SESSIONID=' + session
        print(self.session_text)

        print(self.session_text)
        self.client.headers.update({'Cookie': self.session_text})
        response = self.client.get("/login.php?id=81")
        p = re.compile('csrf_token" value="(.+?)"')
        self.token = p.search(response.text).group(1)
        urllib3.disable_warnings()
        print(self.token)

        self.client.headers.update({'Cookie': self.session_text})
        self.client.headers.update({'Referer': 'http://g-yangsan.gen.es.kr/login.php?id=81'})
        response = self.client.post("https://g-yangsan.gen.es.kr:451/login.php", {'mode': 'login', 'id': '81','path':'','csrf_token':self.token,'userid':'vlfdus99','passwd':'!a5711327'})

    @task
    def mypage(self):
        self.client.headers.update({'Cookie': self.session_text})
        self.client.headers.update({'Referer': 'http://g-yangsan.gen.es.kr/login.php?id=81'})
        response = self.client.get("/join.php?id=88&mode=info")
        print(response.headers)
        print(response.text)

locust 실행

locust