def test_scoring_logic():
"""Test that scoring logic is correct"""
app = create_ctfd()
with app.app_context():
admin = login_as_user(app, name="admin", password="******")
register_user(app,
name="user1",
email="*****@*****.**",
password="******")
client1 = login_as_user(app, name="user1", password="******")
register_user(app,
name="user2",
email="*****@*****.**",
password="******")
client2 = login_as_user(app, name="user2", password="******")
chal1 = gen_challenge(app.db)
gen_flag(app.db, challenge_id=chal1.id, content="flag")
chal1_id = chal1.id
chal2 = gen_challenge(app.db)
gen_flag(app.db, challenge_id=chal2.id, content="flag")
chal2_id = chal2.id
# user1 solves chal1
with freeze_time("2017-10-3 03:21:34"):
with client1.session_transaction():
data = {"submission": "flag", "challenge_id": chal1_id}
client1.post("/api/v1/challenges/attempt", json=data)
# user1 is now on top
scores = get_scores(admin)
assert scores[0]["name"] == "user1"
# user2 solves chal1 and chal2
with freeze_time("2017-10-4 03:30:34"):
with client2.session_transaction():
# solve chal1
data = {"submission": "flag", "challenge_id": chal1_id}
client2.post("/api/v1/challenges/attempt", json=data)
# solve chal2
data = {"submission": "flag", "challenge_id": chal2_id}
client2.post("/api/v1/challenges/attempt", json=data)
# user2 is now on top
scores = get_scores(admin)
assert scores[0]["name"] == "user2"
# user1 solves chal2
with freeze_time("2017-10-5 03:50:34"):
with client1.session_transaction():
data = {"submission": "flag", "challenge_id": chal2_id}
client1.post("/api/v1/challenges/attempt", json=data)
# user2 should still be on top because they solved chal2 first
scores = get_scores(admin)
assert scores[0]["name"] == "user2"
destroy_ctfd(app)
def test_hidden_users_should_not_influence_scores():
app = create_ctfd()
with app.app_context():
register_user(app,
name="user1",
email="*****@*****.**",
password="******")
register_user(app,
name="user2",
email="*****@*****.**",
password="******")
register_user(app,
name="user3",
email="*****@*****.**",
password="******")
user = Users.query.filter_by(name="user3").first()
user.hidden = True
app.db.session.commit()
client1 = login_as_user(app, name="user1", password="******")
# User 1 solves 1st challenge
chal1 = gen_challenge(app.db)
gen_solve(app.db, user_id=2, challenge_id=chal1.id)
# User 2 solves 2nd challenge
chal2 = gen_challenge(app.db)
gen_solve(app.db, user_id=3, challenge_id=chal2.id)
# User 3 solves both
gen_solve(app.db, user_id=4, challenge_id=chal1.id)
gen_solve(app.db, user_id=4, challenge_id=chal2.id)
scores = get_scores(client1)
for entry in scores:
assert entry["name"] != "user3"
user1 = Users.query.filter_by(name="user1").first()
assert user1.place == "1st"
user2 = Users.query.filter_by(name="user2").first()
assert user2.place == "2nd"
destroy_ctfd(app)
def test_scoring_logic_with_zero_point_challenges():
"""Test that scoring logic is correct with zero point challenges. Zero point challenges should not tie break"""
app = create_ctfd()
with app.app_context():
admin = login_as_user(app, name="admin", password="******")
register_user(app,
name="user1",
email="*****@*****.**",
password="******")
client1 = login_as_user(app, name="user1", password="******")
register_user(app,
name="user2",
email="*****@*****.**",
password="******")
client2 = login_as_user(app, name="user2", password="******")
chal1 = gen_challenge(app.db)
gen_flag(app.db, challenge_id=chal1.id, content='flag')
chal1_id = chal1.id
chal2 = gen_challenge(app.db)
gen_flag(app.db, challenge_id=chal2.id, content='flag')
chal2_id = chal2.id
# A 0 point challenge shouldn't influence the scoreboard (see #577)
chal0 = gen_challenge(app.db, value=0)
gen_flag(app.db, challenge_id=chal0.id, content='flag')
chal0_id = chal0.id
# user1 solves chal1
with freeze_time("2017-10-3 03:21:34"):
with client1.session_transaction():
data = {"submission": 'flag', "challenge_id": chal1_id}
client1.post('/api/v1/challenges/attempt', json=data)
# user1 is now on top
scores = get_scores(admin)
assert scores[0]['name'] == 'user1'
# user2 solves chal1 and chal2
with freeze_time("2017-10-4 03:30:34"):
with client2.session_transaction():
# solve chal1
data = {"submission": 'flag', "challenge_id": chal1_id}
client2.post('/api/v1/challenges/attempt'.format(chal1_id),
json=data)
# solve chal2
data = {"submission": 'flag', "challenge_id": chal2_id}
client2.post('/api/v1/challenges/attempt'.format(chal2_id),
json=data)
# user2 is now on top
scores = get_scores(admin)
assert scores[0]['name'] == 'user2'
# user1 solves chal2
with freeze_time("2017-10-5 03:50:34"):
with client1.session_transaction():
data = {"submission": 'flag', "challenge_id": chal2_id}
client1.post('/api/v1/challenges/attempt'.format(chal2_id),
json=data)
# user2 should still be on top because they solved chal2 first
scores = get_scores(admin)
assert scores[0]['name'] == 'user2'
# user2 solves a 0 point challenge
with freeze_time("2017-10-5 03:55:34"):
with client2.session_transaction():
data = {"submission": 'flag', "challenge_id": chal0_id}
client2.post('/api/v1/challenges/attempt'.format(chal0_id),
json=data)
# user2 should still be on top because 0 point challenges should not tie break
scores = get_scores(admin)
assert scores[0]['name'] == 'user2'
destroy_ctfd(app)