Files
RSSHub-python/tests/test_errors.py
2019-12-15 21:52:58 +08:00

31 lines
975 B
Python

from flask import current_app, abort
from tests.base import BaseTestCase
class ErrorsTestCase(BaseTestCase):
def test_400(self):
@current_app.route('/400')
def bad_request():
abort(400)
response = self.client.get('/400')
data = response.get_data(as_text=True)
self.assertIn('400 Bad Request', data)
self.assertEqual(response.status_code, 400)
def test_404(self):
response = self.client.get('/nothing')
data = response.get_data(as_text=True)
self.assertIn('404 Not Found', data)
self.assertEqual(response.status_code, 404)
def test_500(self):
@current_app.route('/500')
def internal_server_error_for_test():
abort(500)
response = self.client.get('/500')
data = response.get_data(as_text=True)
self.assertIn('服务器出错', data)
self.assertEqual(response.status_code, 500)