Тестирование API Flask при сбое запроса

Question

Я пишу приложение Flask и использую Postman для проверки правильности написания моих конечных точек. После тестирования я понял, что единственным тестом, который я не смог выполнить, был тест, когда запросы к базе данных не выполнялись (поэтому я должен поднять код состояния HTTP 500).

Итак, мой вопрос: как я могу проверить, что запрос не выполняется должным образом, если происходит сбой какой-либо внешней службы (например, базы данных)? В идеале я мог бы проверить это непосредственно в Postman, но я не знаю, буду ли я издеваться над сбойным сервисом.

Answer 1

Вы можете использовать try / except блоки для исключений, ниже приведен пример кода:

@app.route('/url')
def func():
    try:
        cur = mysql.connection.cursor()
        cur.execute(sqlQuery)
        mysql.connection.commit()
        cur.close()
    except:
        return "Failure"
    return "Success"

if __name__ == '__main__':
    app.run()

Примечание: помните, что код - это просто фрагмент кода, где я использовал MySQL подключение и возврат нормального статуса ответа.