Mercurial > kallithea
changeset 4697:5225ed43bbb3
tests: let checkSessionFlash search through all messages
author | Mads Kiilerich <madski@unity3d.com> |
---|---|
date | Mon, 15 Dec 2014 13:47:36 +0100 |
parents | fbb992c719aa |
children | 0b14eb8c690b |
files | kallithea/tests/__init__.py |
diffstat | 1 files changed, 5 insertions(+), 3 deletions(-) [+] |
line wrap: on
line diff
--- a/kallithea/tests/__init__.py Mon Dec 15 13:47:36 2014 +0100 +++ b/kallithea/tests/__init__.py Mon Dec 15 13:47:36 2014 +0100 @@ -208,7 +208,9 @@ def checkSessionFlash(self, response, msg): self.assertTrue('flash' in response.session, msg='Response session have no flash key' % response.session) - if not msg in response.session['flash'][0][1]: - msg = u'msg `%s` not found in session flash: got `%s` instead' % ( - msg, response.session['flash'][0][1]) + if not any(msg in m for level, m in response.session['flash']): + for level, m in response.session['flash']: + msg = u'msg `%s` not found in session flash: got `%s` instead' % (msg, m) + self.fail(safe_str(msg)) + msg = u'msg `%s` not found in empty session flash' % (msg) self.fail(safe_str(msg))