mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 14:04:20 +01:00
@@ -1,42 +1,36 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from tests.helpers import *
|
||||
from CTFd.models import Notifications
|
||||
from tests.helpers import (create_ctfd,
|
||||
destroy_ctfd,
|
||||
register_user,
|
||||
login_as_user,
|
||||
gen_challenge,
|
||||
gen_notification)
|
||||
|
||||
|
||||
def test_api_notifications_get():
|
||||
"""Can the users get /api/v1/notifications"""
|
||||
app = create_ctfd()
|
||||
with app.app_context():
|
||||
register_user(app)
|
||||
gen_notification(app.db)
|
||||
with login_as_user(app) as client:
|
||||
# test_api_notifications_get
|
||||
"""Can the users get /api/v1/notifications"""
|
||||
r = client.get('/api/v1/notifications', json="")
|
||||
assert r.status_code == 200
|
||||
assert len(r.get_json()['data']) == 1
|
||||
destroy_ctfd(app)
|
||||
|
||||
|
||||
def test_api_get_notification_detail():
|
||||
app = create_ctfd()
|
||||
with app.app_context():
|
||||
register_user(app)
|
||||
gen_notification(app.db)
|
||||
with login_as_user(app) as client:
|
||||
# test_api_get_notification_detail
|
||||
r = client.get('/api/v1/notifications/1', json="")
|
||||
assert r.status_code == 200
|
||||
resp = r.get_json()
|
||||
assert resp['data']['title'] == 'title'
|
||||
assert resp['data']['content'] == 'content'
|
||||
destroy_ctfd(app)
|
||||
|
||||
|
||||
def test_api_notifications_post_non_admin():
|
||||
"""Can the users post /api/v1/notifications if not admin"""
|
||||
app = create_ctfd()
|
||||
with app.app_context():
|
||||
register_user(app)
|
||||
with login_as_user(app) as client:
|
||||
# test_api_notifications_post_non_admin
|
||||
"""Can the users post /api/v1/notifications if not admin"""
|
||||
r = client.post('/api/v1/notifications', json="")
|
||||
assert r.status_code == 403
|
||||
destroy_ctfd(app)
|
||||
|
||||
Reference in New Issue
Block a user