mirror of https://github.com/JohnHammond/CTFd.git
The place indicator on the team page now excludes counting hidden teams (#1094)
* The place indicator in Team Mode on the team pages and user pages now excludes counting hidden teams. * Updated `flask-marshmallow` to 0.10.1, `marshmallow-sqlalchemy` to 0.17.0 * Pinned `marshmallow` to 2.20.2 * Closes #1093 * Closes #1088selenium-screenshot-testing
parent
75a9a5a697
commit
941ca8f506
|
@ -168,6 +168,7 @@ def get_team_standings(count=None, admin=False):
|
||||||
db.session.query(Teams.id.label("team_id"))
|
db.session.query(Teams.id.label("team_id"))
|
||||||
.join(sumscores, Teams.id == sumscores.columns.team_id)
|
.join(sumscores, Teams.id == sumscores.columns.team_id)
|
||||||
.filter(Teams.banned == False)
|
.filter(Teams.banned == False)
|
||||||
|
.filter(Teams.hidden == False)
|
||||||
.order_by(sumscores.columns.score.desc(), sumscores.columns.id)
|
.order_by(sumscores.columns.score.desc(), sumscores.columns.id)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -24,6 +24,7 @@ gevent==1.4.0
|
||||||
python-dotenv==0.9.1
|
python-dotenv==0.9.1
|
||||||
flask-restplus==0.12.1
|
flask-restplus==0.12.1
|
||||||
pathlib2==2.3.2
|
pathlib2==2.3.2
|
||||||
flask-marshmallow==0.9.0
|
flask-marshmallow==0.10.1
|
||||||
marshmallow-sqlalchemy==0.15.0
|
marshmallow-sqlalchemy==0.17.0
|
||||||
boto3==1.9.39
|
boto3==1.9.39
|
||||||
|
marshmallow==2.20.2
|
||||||
|
|
|
@ -0,0 +1,92 @@
|
||||||
|
from tests.helpers import (
|
||||||
|
create_ctfd,
|
||||||
|
destroy_ctfd,
|
||||||
|
login_as_user,
|
||||||
|
gen_challenge,
|
||||||
|
gen_flag,
|
||||||
|
gen_user,
|
||||||
|
gen_team,
|
||||||
|
)
|
||||||
|
from CTFd.models import Teams
|
||||||
|
|
||||||
|
from CTFd.utils.scores import get_standings, get_team_standings
|
||||||
|
|
||||||
|
|
||||||
|
def setup_app(app):
|
||||||
|
user1 = gen_user(app.db, name="user1", email="user1@ctfd.io")
|
||||||
|
team1 = gen_team(app.db, name="team1", email="team1@ctfd.io")
|
||||||
|
user1.team_id = team1.id
|
||||||
|
team1.members.append(user1)
|
||||||
|
team1.hidden = True
|
||||||
|
|
||||||
|
user2 = gen_user(app.db, name="user2", email="user2@ctfd.io")
|
||||||
|
team2 = gen_team(app.db, name="team2", email="team2@ctfd.io")
|
||||||
|
user2.team_id = team2.id
|
||||||
|
team2.members.append(user2)
|
||||||
|
|
||||||
|
gen_challenge(app.db)
|
||||||
|
gen_flag(app.db, 1)
|
||||||
|
|
||||||
|
app.db.session.commit()
|
||||||
|
|
||||||
|
with login_as_user(app, name="user1") as client:
|
||||||
|
flag = {"challenge_id": 1, "submission": "flag"}
|
||||||
|
client.post("/api/v1/challenges/attempt", json=flag)
|
||||||
|
|
||||||
|
with login_as_user(app, name="user2") as client:
|
||||||
|
flag = {"challenge_id": 1, "submission": "flag"}
|
||||||
|
client.post("/api/v1/challenges/attempt", json=flag)
|
||||||
|
|
||||||
|
|
||||||
|
def test_standings():
|
||||||
|
app = create_ctfd(user_mode="teams")
|
||||||
|
|
||||||
|
with app.app_context():
|
||||||
|
setup_app(app)
|
||||||
|
|
||||||
|
standings = get_standings()
|
||||||
|
|
||||||
|
assert standings[0].name == "team2"
|
||||||
|
assert standings[0].score == 100
|
||||||
|
|
||||||
|
destroy_ctfd(app)
|
||||||
|
|
||||||
|
|
||||||
|
def test_team_standings():
|
||||||
|
app = create_ctfd(user_mode="teams")
|
||||||
|
|
||||||
|
with app.app_context():
|
||||||
|
setup_app(app)
|
||||||
|
|
||||||
|
team_standings = get_team_standings()
|
||||||
|
|
||||||
|
first_team = Teams.query.filter_by(id=team_standings[0].team_id).first_or_404()
|
||||||
|
|
||||||
|
assert first_team.name == "team2"
|
||||||
|
assert first_team.score == 100
|
||||||
|
|
||||||
|
|
||||||
|
def test_admin_standings():
|
||||||
|
app = create_ctfd(user_mode="teams")
|
||||||
|
|
||||||
|
with app.app_context():
|
||||||
|
setup_app(app)
|
||||||
|
|
||||||
|
standings = get_standings(admin=True)
|
||||||
|
|
||||||
|
assert standings[0].name == "team1"
|
||||||
|
assert standings[0].score == 100
|
||||||
|
|
||||||
|
|
||||||
|
def test_admin_team_standings():
|
||||||
|
app = create_ctfd(user_mode="teams")
|
||||||
|
|
||||||
|
with app.app_context():
|
||||||
|
setup_app(app)
|
||||||
|
|
||||||
|
team_standings = get_team_standings(admin=True)
|
||||||
|
|
||||||
|
first_team = Teams.query.filter_by(id=team_standings[0].team_id).first_or_404()
|
||||||
|
|
||||||
|
assert first_team.name == "team1"
|
||||||
|
assert first_team.score == 100
|
Loading…
Reference in New Issue