init CTFd source

tests/api/v1/teams/test_teams.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from CTFd.models import Teams, Users, db
+from CTFd.utils.crypto import verify_password
+from tests.helpers import (
+    create_ctfd,
+    destroy_ctfd,
+    gen_team,
+    login_as_user,
+    register_user,
+)
+
+
+def test_api_can_query_by_team_emails():
+    """Can an admin user query /api/v1/teams using a teams's email address"""
+    app = create_ctfd(user_mode="teams")
+    with app.app_context():
+        gen_team(app.db, email="team@findme.com")
+        register_user(app, name="testuser", email="user@findme.com")
+        with login_as_user(app, "testuser") as client:
+            r = client.get("/api/v1/teams?field=email&q=findme", json=True)
+            assert r.status_code == 400
+            assert r.get_json()["errors"].get("field")
+        with login_as_user(app, "admin") as client:
+            r = client.get("/api/v1/teams?field=email&q=findme", json=True)
+            assert r.status_code == 200
+            assert r.get_json()["data"][0]["id"] == 1
+            assert r.get_json()["data"][0]["name"] == "team_name"
+    destroy_ctfd(app)
+
+
+def test_api_team_can_update_password_if_none_not_if_set():
+    app = create_ctfd(user_mode="teams")
+    with app.app_context():
+        # Create a user with a null password. Use raw SQL to bypass SQLAlchemy validates
+        gen_team(app.db, name="testteam", email="team@examplectf.com")
+        db.session.execute("UPDATE teams SET password=NULL WHERE name='testteam'")
+        team = Teams.query.filter_by(id=1).first()
+        db.session.commit()
+        assert team.password is None
+
+        # Login and test that we are authed
+        captain = Users.query.filter_by(id=2).first()
+        normal_user = Users.query.filter_by(id=3).first()
+        with login_as_user(app, captain.name) as client:
+            r = client.get("/api/v1/teams/me", json=True)
+            assert r.get_json()["data"]["id"] == team.id
+            assert r.status_code == 200
+
+            # Patch the team's password from NULL
+            team = Teams.query.filter_by(name="testteam").first()
+            assert team.password is None
+            data = {"password": "12345", "confirm": "password"}
+            r = client.patch("/api/v1/teams/me", json=data)
+            assert r.status_code == 200
+
+            # Verify password is now set
+            team = Teams.query.filter_by(name="testteam").first()
+            assert verify_password(plaintext="12345", ciphertext=team.password)
+
+            # Verify that password cannot be changed without valid password
+            data = {"password": "noset", "confirm": "noset"}
+            r = client.patch("/api/v1/teams/me", json=data)
+            resp = r.get_json()
+            assert resp["errors"]["confirm"] == ["Your previous password is incorrect"]
+            assert r.status_code == 400
+
+            # Verify that a normal user cannot change the team password
+            with login_as_user(app, normal_user.name) as client:
+                # Try changing the password for the team
+                data = {"password": "newpassword", "confirm": "12345"}
+                r = client.patch("/api/v1/teams/me", json=data)
+                assert r.status_code == 403
+
+                # Verify that team password has not changed
+                team = Teams.query.filter_by(name="testteam").first()
+                assert verify_password(plaintext="12345", ciphertext=team.password)
+
+            # Create a new team
+            new_team = gen_team(app.db, name="newteam", email="newteam@examplectf.com")
+            new_captain = Users.query.filter_by(id=new_team.captain_id).first()
+
+            # Verify that the captain from the new team cannot change the password of the original team
+            with login_as_user(app, new_captain.name) as client:
+                data = {"password": "newpassword", "confirm": "12345"}
+                r = client.patch("/api/v1/teams/1", json=data)
+                assert r.status_code == 403
+
+                # Verify that old test team password has not changed
+                team = Teams.query.filter_by(name="testteam").first()
+                assert verify_password(plaintext="12345", ciphertext=team.password)