init CTFd source

tests/models/test_model_utils.py

@@ -0,0 +1,23 @@
+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)