mirror of https://github.com/JohnHammond/CTFd.git
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from flask.testing import FlaskClient
|
|
from tests.helpers import create_ctfd, destroy_ctfd, login_as_user
|
|
|
|
|
|
def test_api_csrf_failure():
|
|
"""Test that API requests require the CSRF-Token header"""
|
|
app = create_ctfd()
|
|
app.test_client_class = FlaskClient
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as client:
|
|
r = client.post(
|
|
"/api/v1/challenges",
|
|
json={
|
|
"name": "chal",
|
|
"category": "cate",
|
|
"description": "desc",
|
|
"value": "100",
|
|
"state": "hidden",
|
|
"type": "standard",
|
|
},
|
|
)
|
|
assert r.status_code == 403
|
|
|
|
with client.session_transaction() as sess:
|
|
nonce = sess.get("nonce")
|
|
|
|
r = client.post(
|
|
"/api/v1/challenges",
|
|
headers={"CSRF-Token": nonce},
|
|
json={
|
|
"name": "chal",
|
|
"category": "cate",
|
|
"description": "desc",
|
|
"value": "100",
|
|
"state": "hidden",
|
|
"type": "standard",
|
|
},
|
|
)
|
|
assert r.status_code == 200
|
|
destroy_ctfd(app)
|