Fix CTF resets in mysql (#785)

* Extend test for CTFd resets and fix issue in MySQL
* Expand reset test and fix helper for fails and tracking
This commit is contained in:
Kevin Chung
2018-12-06 20:43:14 -05:00
committed by GitHub
parent 547fe61870
commit afdfaa15da
3 changed files with 19 additions and 10 deletions

View File

@@ -1,5 +1,6 @@
from CTFd.models import Users, Challenges
from tests.helpers import *
import random
def test_get_admin_config():
@@ -16,15 +17,20 @@ def test_reset():
app = create_ctfd()
with app.app_context():
base_user = 'user'
for x in range(10):
user = base_user + str(x)
user_email = user + "@ctfd.io"
gen_user(app.db, name=user, email=user_email)
for x in range(10):
chal = gen_challenge(app.db, name='chal_name{}'.format(x))
gen_flag(app.db, challenge_id=chal.id, content='flag')
for x in range(10):
user = base_user + str(x)
user_email = user + "@ctfd.io"
user_obj = gen_user(app.db, name=user, email=user_email)
gen_award(app.db, user_id=user_obj.id)
gen_solve(app.db, user_id=user_obj.id, challenge_id=random.randint(1, 10))
gen_fail(app.db, user_id=user_obj.id, challenge_id=random.randint(1, 10))
gen_tracking(app.db, user_id=user_obj.id)
assert Users.query.count() == 11 # 11 because of the first admin user
assert Challenges.query.count() == 10
@@ -39,4 +45,7 @@ def test_reset():
assert Users.query.count() == 0
assert Challenges.query.count() == 10
assert Solves.query.count() == 0
assert Fails.query.count() == 0
assert Tracking.query.count() == 0
destroy_ctfd(app)