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