def __call__(self, r: PreparedRequest):
data = {
"bk_app_code": self.bk_app_code,
"bk_app_secret": self.bk_app_secret,
"bk_username": self.bk_username,
"operator": self.operator,
}
if self.bk_supplier_account:
data["bk_supplier_account"] = self.bk_supplier_account
r.body = update_request_body(r.body, data)
return r
def __call__(self, r: PreparedRequest):
# 针对get请求，添加auth参数到url中; 针对post请求，添加auth参数到body体中
auth_params = {
"app_code": self.app_code,
"app_secret": self.app_secret,
"bk_username": self.bk_username
}
if r.method in ["GET"]:
r.url = update_url_parameters(r.url, auth_params)
elif r.method in ["POST"]:
r.body = update_request_body(r.body, auth_params)
return r
def test_update_request_body(body, params, expected_body):
updated_body = update_request_body(body, params)
assert updated_body == expected_body