Improve the tests, add a new debug endpoint

This commit is contained in:
Thomas Sileo
2018-05-27 11:50:09 +02:00
parent 25a75a9cef
commit cf242b2d84
4 changed files with 52 additions and 3 deletions

20
app.py
View File

@@ -21,6 +21,7 @@ from flask import redirect
from flask import Response
from flask import render_template
from flask import session
from flask import jsonify as flask_jsonify
from flask import url_for
from html2text import html2text
from itsdangerous import JSONWebSignatureSerializer
@@ -47,6 +48,8 @@ from config import OBJECT_SERVICE
from config import PASS
from config import HEADERS
from config import VERSION
from config import DEBUG_MODE
from config import _drop_db
from config import custom_cache_purge_hook
from utils.httpsig import HTTPSigAuth, verify_request
from utils.key import get_secret_key
@@ -710,6 +713,23 @@ def inbox():
)
@app.route('/api/debug', methods=['GET', 'DELETE'])
@api_required
def api_debug():
"""Endpoint used/needed for testing, only works in DEBUG_MODE."""
if not DEBUG_MODE:
return flask_jsonify(message='DEBUG_MODE is off')
if request.method == 'DELETE':
_drop_db()
return flask_jsonify(message='DB dropped')
return flask_jsonify(
inbox=DB.inbox.count(),
outbox=DB.outbox.count(),
)
@app.route('/api/upload', methods=['POST'])
@api_required
def api_upload():