def setup(): body = json.dumps({ 'username': e2e_users[0].username, 'password': e2e_users[0].password, 'cloud_fqdn': e2e_users[0].fqdn }) headers = { 'content-type': 'application/json', 'accept': 'application/json' } r = requests.post(splinter_tests.make_url('/api/login'), data=body, headers=headers) r.raise_for_status() global csrftoken, sessionid csrftoken = r.cookies['csrftoken'] sessionid = r.cookies['sessionid']
def do_request(corrupt_cookie=False, corrupt_header=False): """Perform a POST request against the server, and return that request object.""" token = (csrftoken + "_") if corrupt_cookie else csrftoken header = (csrftoken + "_") if corrupt_header else csrftoken cookies = { 'csrftoken': token, 'sessionid': sessionid } headers = { 'X-CSRFToken': header, 'content-type': 'application/json' } r = requests.post(splinter_tests.make_url('/api/devices'), data='{"mac": "00409D987654"}', cookies=cookies, headers=headers) return r
def teardown(): requests.get(splinter_tests.make_url('/logout'), cookies={'csrftoken': csrftoken, 'sessionid': sessionid}, headers={'X-CSRFToken': csrftoken})