pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add referer headers to load tests for csrf

Browse Source
This commit is contained in:
tomdds 2019-11-18 14:30:34 -05:00
parent 84e5fa9bd1
commit a4c106ee0f
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
load-test/locustfile.py
View File

@ -111,7 +111,11 @@ def create_new_app(l, portfolio_id):
"csrf_token": get_csrf_token(new_app_form), "csrf_token": get_csrf_token(new_app_form),
} }
create_app_response = l.client.post(create_app_url, create_app_body) create_app_response = l.client.post(
create_app_url,
create_app_body,
headers={"Referer": l.parent.host + create_app_url},
)
application_id = extract_id(create_app_response.url) application_id = extract_id(create_app_response.url)
create_environments_body = { create_environments_body = {
@ -122,11 +126,13 @@ def create_new_app(l, portfolio_id):
"csrf_token": get_csrf_token(create_app_response), "csrf_token": get_csrf_token(create_app_response),
} }
create_environments_url = ( create_environments_url = f"/applications/{application_id}/new/step_2"
f"/applications/{application_id}/new/step_2?portfolio_id={portfolio_id}"
)
l.client.post(create_environments_url, create_environments_body) l.client.post(
create_environments_url + f"?portfolio_id={portfolio_id}",
create_environments_body,
headers={"Referer": l.parent.host + create_environments_url},
)
return f"/applications/{application_id}/settings" return f"/applications/{application_id}/settings"
@ -147,7 +153,11 @@ def create_portfolio(l):
"csrf_token": get_csrf_token(new_portfolio_form), "csrf_token": get_csrf_token(new_portfolio_form),
} }
response = l.client.post("/portfolios", new_portfolio_body) response = l.client.post(
"/portfolios",
new_portfolio_body,
headers={"Referer": l.parent.host + "/portfolios"},
)
return urlparse(response.url).path return urlparse(response.url).path