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))