root
/
flask
mirror of https://github.com/pallets/flask.git
1
0
Fork 0
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Expand get_flashed_messages tests. 

Ready to pull request #336.
http://github.com/mitsuhiko/flask/pull/336
This commit is contained in:
Ron DuPlain 2012-01-16 22:22:04 -05:00
parent ed40eacf42
commit b9907b4969
1 changed files with 44 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
flask/testsuite/basic.py
View File

@ -321,22 +321,61 @@ class BasicFunctionalityTestCase(FlaskTestCase):
@app.route('/test')
def test():
messages = flask.get_flashed_messages()
self.assert_equal(len(messages), 3)
self.assert_equal(messages[0], u'Hello World')
self.assert_equal(messages[1], u'Hello World')
self.assert_equal(messages[2], flask.Markup(u'<em>Testing</em>'))
return ''
@app.route('/test_with_categories')
def test_with_categories():
messages = flask.get_flashed_messages(with_categories=True)
self.assert_equal(len(messages), 3)
self.assert_equal(messages[0], ('message', u'Hello World'))
self.assert_equal(messages[1], ('error', u'Hello World'))
self.assert_equal(messages[2], ('warning', flask.Markup(u'<em>Testing</em>')))
return ''
messages = flask.get_flashed_messages()
self.assert_equal(len(messages), 3)
@app.route('/test_filter')
def test_filter():
messages = flask.get_flashed_messages(category_filter=['message'], with_categories=True)
self.assert_equal(len(messages), 1)
self.assert_equal(messages[0], ('message', u'Hello World'))
return ''
@app.route('/test_filters')
def test_filters():
messages = flask.get_flashed_messages(category_filter=['message', 'warning'], with_categories=True)
self.assert_equal(len(messages), 2)
self.assert_equal(messages[0], ('message', u'Hello World'))
self.assert_equal(messages[1], ('warning', flask.Markup(u'<em>Testing</em>')))
return ''
@app.route('/test_filters_without_returning_categories')
def test_filters():
messages = flask.get_flashed_messages(category_filter=['message', 'warning'])
self.assert_equal(len(messages), 2)
self.assert_equal(messages[0], u'Hello World')
self.assert_equal(messages[1], u'Hello World')
self.assert_equal(messages[2], flask.Markup(u'<em>Testing</em>'))
self.assert_equal(messages[1], flask.Markup(u'<em>Testing</em>'))
return ''
c = app.test_client()
c.get('/')
c.get('/') # Flash some messages.
c.get('/test')
c.get('/') # Flash more messages.
c.get('/test_with_categories')
c.get('/') # Flash more messages.
c.get('/test_filter')
c.get('/') # Flash more messages.
c.get('/test_filters')
c.get('/') # Flash more messages.
c.get('/test_filters_without_returning_categories')
def test_request_processing(self):
app = flask.Flask(__name__)
evts = []