mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 14:04:20 +01:00
Prevent accidental double unlocks (#1315)
* Prevent a hint from being unlocked twice * Closes #1301
This commit is contained in:
@@ -59,6 +59,16 @@ class UnlockList(Resource):
|
||||
if response.errors:
|
||||
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)
|
||||
|
||||
award_schema = AwardSchema()
|
||||
|
||||
@@ -95,6 +95,27 @@ def test_api_hint_unlocked():
|
||||
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():
|
||||
"""Can the users access /api/v1/hints if not admin"""
|
||||
app = create_ctfd()
|
||||
|
||||
Reference in New Issue
Block a user