mirror of https://github.com/JohnHammond/CTFd.git
Prevent accidental double unlocks (#1315)
* Prevent a hint from being unlocked twice * Closes #1301bulk-clear-sessions
parent
96f317293f
commit
04e6b2011f
|
@ -59,6 +59,16 @@ class UnlockList(Resource):
|
||||||
if response.errors:
|
if response.errors:
|
||||||
return {"success": False, "errors": response.errors}, 400
|
return {"success": False, "errors": response.errors}, 400
|
||||||
|
|
||||||
|
existing = Unlocks.query.filter_by(**req).first()
|
||||||
|
if existing:
|
||||||
|
return (
|
||||||
|
{
|
||||||
|
"success": False,
|
||||||
|
"errors": {"target": "You've already unlocked this this target"},
|
||||||
|
},
|
||||||
|
400,
|
||||||
|
)
|
||||||
|
|
||||||
db.session.add(response.data)
|
db.session.add(response.data)
|
||||||
|
|
||||||
award_schema = AwardSchema()
|
award_schema = AwardSchema()
|
||||||
|
|
|
@ -95,6 +95,27 @@ def test_api_hint_unlocked():
|
||||||
destroy_ctfd(app)
|
destroy_ctfd(app)
|
||||||
|
|
||||||
|
|
||||||
|
def test_api_hint_double_unlock():
|
||||||
|
"""Can a target hint be unlocked twice"""
|
||||||
|
app = create_ctfd()
|
||||||
|
with app.app_context():
|
||||||
|
chal = gen_challenge(app.db)
|
||||||
|
gen_hint(app.db, chal.id, content="This is a hint", cost=1, type="standard")
|
||||||
|
register_user(app)
|
||||||
|
# Give user points with an award
|
||||||
|
gen_award(app.db, 2)
|
||||||
|
client = login_as_user(app)
|
||||||
|
r = client.get("/api/v1/hints/1")
|
||||||
|
assert r.status_code == 200
|
||||||
|
r = client.post("/api/v1/unlocks", json={"target": 1, "type": "hints"})
|
||||||
|
assert r.status_code == 200
|
||||||
|
r = client.get("/api/v1/hints/1")
|
||||||
|
assert r.status_code == 200
|
||||||
|
r = client.post("/api/v1/unlocks", json={"target": 1, "type": "hints"})
|
||||||
|
assert r.status_code == 400
|
||||||
|
destroy_ctfd(app)
|
||||||
|
|
||||||
|
|
||||||
def test_api_hints_admin_access():
|
def test_api_hints_admin_access():
|
||||||
"""Can the users access /api/v1/hints if not admin"""
|
"""Can the users access /api/v1/hints if not admin"""
|
||||||
app = create_ctfd()
|
app = create_ctfd()
|
||||||
|
|
Loading…
Reference in New Issue