24 lines
680 B
Python
24 lines
680 B
Python
from CTFd.models import (
|
|
Challenges,
|
|
Comments,
|
|
Files,
|
|
Solves,
|
|
Submissions,
|
|
get_class_by_tablename,
|
|
)
|
|
from tests.helpers import create_ctfd, destroy_ctfd
|
|
|
|
|
|
def test_get_class_by_tablename():
|
|
"""
|
|
Test that get_class_by_tablename() returns the correct table
|
|
"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
assert get_class_by_tablename("solves") == Solves
|
|
assert get_class_by_tablename("comments") == Comments
|
|
assert get_class_by_tablename("files") == Files
|
|
assert get_class_by_tablename("submissions") == Submissions
|
|
assert get_class_by_tablename("challenges") == Challenges
|
|
destroy_ctfd(app)
|