From 98ce23d20473d24335a12a8ced5ea269ab24a011 Mon Sep 17 00:00:00 2001 From: Gabi Thume Date: Mon, 29 Apr 2013 20:35:22 -0700 Subject: Fixing ticket #146 - Make messaging system testable --- mediagoblin/messages.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'mediagoblin/messages.py') diff --git a/mediagoblin/messages.py b/mediagoblin/messages.py index 80d8ece7..d58f13d4 100644 --- a/mediagoblin/messages.py +++ b/mediagoblin/messages.py @@ -14,16 +14,24 @@ # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . +from mediagoblin.tools import common + DEBUG = 'debug' INFO = 'info' SUCCESS = 'success' WARNING = 'warning' ERROR = 'error' +ADD_MESSAGE_TEST = [] + def add_message(request, level, text): messages = request.session.setdefault('messages', []) messages.append({'level': level, 'text': text}) + + if common.TESTS_ENABLED: + ADD_MESSAGE_TEST.append(messages) + request.session.save() @@ -33,4 +41,10 @@ def fetch_messages(request, clear_from_session=True): # Save that we removed the messages from the session request.session['messages'] = [] request.session.save() + return messages + + +def clear_add_message(): + global ADD_MESSAGE_TEST + ADD_MESSAGE_TEST = [] -- cgit v1.2.3