如何在用户类中初始化参数

Question

我用一些init参数创建了一个User类，但在执行时得到了TypeError，如下所示：“hocust-f stress.py--headless-u 100-r 100-t 1m”

Traceback（最近一次调用最后一次）：文件src/gevent/greenlet.py，第906行，在gevent._gevent_cgreenlet。Greenlet.run文件/库/框架/Python.framework/Versions/3.7/lib/python3.7/site-packages/locust/runners.py，第406行，在lambda：超级（LocalRunner，自我）. start（user_count，spawn_rate，等待=等待）文件/库/框架/Python.framework/Versions/3.7/lib/python3.7/site-packages/locust/runners.py，第314行，在startself.spawn_users（user_count，spawn_rate=spawn_rate，等待=等待）文件/库/框架/Python.framework/Versions/3.7/lib/python3.7/site-packages/locust/runners.py，第206行，在spawn_users产卵（）文件/库/框架/Python.framework/Versions/3.7/lib/python3.7/site-packages/locust/runners.py，第199行，在产卵new_user=user_class（self.environment）打字错误：init（）缺少4个必需的位置参数：uid、room_id、action_uid和appkey

class JoinRoomUser(HttpUser):
    wait_time = between(1, 2.5)
    host = ''

    def __init__(self, env, uid, room_id, action_uid, appkey):
        self.env = env
        self.uid = uid
        self.room_id =room_id
        self.action_id = action_uid
        self.appkey = appkey

如何启动这些参数？

蝗虫脚本中的任务如下。

@task
def join_room(self):
host = 'xxxxxxx'
body = {
    "room_id": self.room_id,
    "userid": self.action_uid,
    "uid": self.action_uid
}
rsp = self.client.post(url, json=body, headers=get_header(self.appkey))

如果我执行“hocust-f stress test/stress.py'Qa'123456'101013'686868'eos'--headless-u 100-r 100--run-time1m”，然后抛出“bogon/ERROR/hocust.main:未知用户：Qa，686868，123456，101013，eos”错误。

Answer 1

尝试使用*args，**kwargs和超级（）从父类继承时。对于这种特殊情况，您不会在初始化步骤中提供uid、room_id、action_uid、appkey变量，因为HttpUser继承自User，它只有一个1参数环境

class SimpleSendRequest(User):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

对于用户类更建议从用户类文档继承

如果确实需要将变量设置为子对象，可以在初始化步骤进行设置，如下所示：

def __init__(self, env, *args, **kwargs):
    super().__init__(env, *args, **kwargs)
    self.uid = your_map['uid']
    self.room_id = your_map['room_id']
    self.action_id = your_map['action_uid']
    self.appkey = your_map['appkey']

唯一的问题是，你将如何使用它们

要添加自定义参数，请通过CLI检查github：

    @events.init_command_line_parser.add_listener
def init_parser(parser):
    parser.add_argument("--customarg", type=str, env_var="LOCUST_MY_ARGUMENT", default="1234", help="It's working")

@events.init.add_listener
def _(environment, **kw):
    print("Custom argument supplied: %s" % environment.parsed_options.customarg)

要在类JoinRoomUser init中获取它，请使用：

print(self.environment.parsed_options.customarg)

像这样跑：

locust -f locustfile.py --customarg CUSTOM_VALUE --headless -u 10 -r 10 --run-time 5s