diff options
author | Jef van Schendel <jefvanschendel@gmail.com> | 2011-06-29 13:53:17 +0200 |
---|---|---|
committer | Jef van Schendel <jefvanschendel@gmail.com> | 2011-06-29 13:53:17 +0200 |
commit | d022d9f93f9486d547e3954788c9d881bd62fd2c (patch) | |
tree | eca68d5b4e866f7f5ed5b5e3d8364c8605f5a3e1 /mediagoblin/tests/test_messages.py | |
parent | a4e4d775485f51eb4a2c3c30c624bd364b0c6504 (diff) | |
parent | b0d835fcd0f46025eee6c839e9bf26f513543898 (diff) | |
download | mediagoblin-d022d9f93f9486d547e3954788c9d881bd62fd2c.tar.lz mediagoblin-d022d9f93f9486d547e3954788c9d881bd62fd2c.tar.xz mediagoblin-d022d9f93f9486d547e3954788c9d881bd62fd2c.zip |
Merge remote-tracking branch 'gitorious/master'
Diffstat (limited to 'mediagoblin/tests/test_messages.py')
-rw-r--r-- | mediagoblin/tests/test_messages.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/mediagoblin/tests/test_messages.py b/mediagoblin/tests/test_messages.py new file mode 100644 index 00000000..4cd9381a --- /dev/null +++ b/mediagoblin/tests/test_messages.py @@ -0,0 +1,44 @@ +# GNU MediaGoblin -- federated, autonomous media hosting +# Copyright (C) 2011 Free Software Foundation, Inc +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +from mediagoblin.messages import fetch_messages, add_message +from mediagoblin.tests.tools import setup_fresh_app +from mediagoblin import util + + +@setup_fresh_app +def test_messages(test_app): + """ + Added messages should show up in the request.session, + fetched messages should be the same as the added ones, + and fetching should clear the message list. + """ + # Aquire a request object + test_app.get('/') + context = util.TEMPLATE_TEST_CONTEXT['mediagoblin/root.html'] + request = context['request'] + + # The message queue should be empty + assert request.session.get('messages', []) == [] + + # Adding a message should modify the session accordingly + add_message(request, 'herp_derp', 'First!') + test_msg_queue = [{'text': 'First!', 'level': 'herp_derp'}] + assert request.session['messages'] == test_msg_queue + + # fetch_messages should return and empty the queue + assert fetch_messages(request) == test_msg_queue + assert request.session.get('messages') == [] |