-
Notifications
You must be signed in to change notification settings - Fork 0
/
flow_bkm.py
31 lines (25 loc) · 905 Bytes
/
flow_bkm.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import random
from locust import HttpLocust, TaskSequence, seq_task, task, TaskSet
import json
tokens = None
class Benchmark(TaskSet):
# def on_start(self):
# if len(tokens) > 0:
# self.token = random.choice(tokens)
# region EVENT
@task
def task1(self):
with self.client.request('GET', '/bpk-tv/HTV7HD/default/HTV7HD.isml/index.m3u8', headers={'User-Agent':'Locust'}, catch_response=True, timeout=1) as response:
if response.status_code == 200:
response.success('OK')
class Run(HttpLocust):
min_wait = 100
max_wait = 100
task_set = Benchmark
# def __init__(self):
# super(Run, self).__init__()
# global tokens
# if tokens is None:
# with open('token.txt', 'r') as lines:
# tokens = [line.strip()
# for line in lines if len(line.strip()) > 20]