changeset 4881:afddaa53ab8a

tests: fix handling of failing flash assertion ... again
author Mads Kiilerich <madski@unity3d.com>
date Fri, 06 Mar 2015 16:21:24 +0100
parents 5d8fbada35fc
children a5475229ba28
files kallithea/tests/__init__.py
diffstat 1 files changed, 3 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/kallithea/tests/__init__.py	Thu Feb 26 17:32:34 2015 +0100
+++ b/kallithea/tests/__init__.py	Fri Mar 06 16:21:24 2015 +0100
@@ -214,12 +214,14 @@
         return User.get_by_username(self._logged_username)
 
     def checkSessionFlash(self, response, msg, skip=0):
+        if 'flash' not in response.session:
+            self.fail(safe_str(u'msg `%s` not found - session has no flash ' % msg))
         try:
             level, m = response.session['flash'][-1 - skip]
             if msg in m:
                 return
         except IndexError:
             pass
-        self.fail(safe_str(u'msg `%s` not found in session flash skipping %s: %s' %
+        self.fail(safe_str(u'msg `%s` not found in session flash (skipping %s): %s' %
                            (msg, skip,
                             ', '.join('`%s`' % m for level, m in response.session['flash']))))