From 34f6df1fe19147704075b4fae82a2253efd57949 Mon Sep 17 00:00:00 2001 From: Marcin Kuzminski Date: Thu, 16 Nov 2017 14:19:52 +0100 Subject: [PATCH] tests: improve assert_flash_message call when called with no_ and empty flash is present. --- rhodecode/tests/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rhodecode/tests/__init__.py b/rhodecode/tests/__init__.py index cdb3db9d..e1158370 100644 --- a/rhodecode/tests/__init__.py +++ b/rhodecode/tests/__init__.py @@ -208,7 +208,11 @@ def assert_session_flash(response, msg=None, category=None, no_=None): messages = flash.pop_messages(session=session) msg = _eval_if_lazy(msg) - assert messages, 'unable to find message `%s` in empty flash list' % msg + if no_: + error_msg = 'unable to detect no_ message `%s` in empty flash list' % no_ + else: + error_msg = 'unable to find message `%s` in empty flash list' % msg + assert messages, error_msg message = messages[0] message_text = _eval_if_lazy(message.message) or ''