mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 14:04:20 +01:00
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from CTFd.utils import get_config
|
|
from tests.helpers import create_ctfd, destroy_ctfd, login_as_user
|
|
|
|
|
|
def test_api_configs_get_non_admin():
|
|
"""Can a user get /api/v1/configs if not admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with app.test_client() as client:
|
|
r = client.get('/api/v1/configs')
|
|
assert r.status_code == 302
|
|
|
|
# test_api_configs_post_non_admin
|
|
"""Can a user post /api/v1/configs if not admin"""
|
|
r = client.post('/api/v1/configs', json="")
|
|
assert r.status_code == 403
|
|
|
|
# test_api_configs_patch_non_admin
|
|
"""Can a user patch /api/v1/configs if not admin"""
|
|
r = client.patch('/api/v1/configs', json="")
|
|
assert r.status_code == 403
|
|
|
|
# test_api_config_get_non_admin
|
|
"""Can a user get /api/v1/configs/<config_key> if not admin"""
|
|
r = client.get('/api/v1/configs/ctf_name')
|
|
assert r.status_code == 302
|
|
|
|
# test_api_config_patch_non_admin
|
|
"""Can a user patch /api/v1/configs/<config_key> if not admin"""
|
|
r = client.patch('/api/v1/configs/ctf_name', json="")
|
|
assert r.status_code == 403
|
|
|
|
# test_api_config_delete_non_admin
|
|
"""Can a user delete /api/v1/configs/<config_key> if not admin"""
|
|
r = client.delete('/api/v1/configs/ctf_name', json="")
|
|
assert r.status_code == 403
|
|
assert get_config('ctf_name') == 'CTFd'
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_configs_get_admin():
|
|
"""Can a user get /api/v1/configs if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.get('/api/v1/configs')
|
|
assert r.status_code == 200
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_configs_post_admin():
|
|
"""Can a user post /api/v1/configs if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.post('/api/v1/configs', json={"value": "9.9.9", "key": "test"})
|
|
assert r.status_code == 200
|
|
assert get_config('test') == "9.9.9"
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_configs_patch_admin():
|
|
"""Can a user patch /api/v1/configs if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.patch('/api/v1/configs', json={"ctf_name": "Changed_Name"})
|
|
assert r.status_code == 200
|
|
assert get_config('ctf_name') == "Changed_Name"
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_config_get_admin():
|
|
"""Can a user get /api/v1/configs/<config_key> if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.get('/api/v1/configs/ctf_name')
|
|
assert r.status_code == 200
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_config_patch_admin():
|
|
"""Can a user patch /api/v1/configs/<config_key> if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.patch('/api/v1/configs/ctf_name', json={"value": "Changed_Name"})
|
|
assert r.status_code == 200
|
|
assert get_config('ctf_name') == "Changed_Name"
|
|
destroy_ctfd(app)
|
|
|
|
|
|
def test_api_config_delete_admin():
|
|
"""Can a user delete /api/v1/configs/<config_key> if admin"""
|
|
app = create_ctfd()
|
|
with app.app_context():
|
|
with login_as_user(app, "admin") as admin:
|
|
r = admin.delete('/api/v1/configs/ctf_name', json="")
|
|
assert r.status_code == 200
|
|
assert get_config('ctf_name') is None
|
|
destroy_ctfd(app)
|