Commit | Line | Data |
---|---|---|
9f094bea | 1 | # GNU MediaGoblin -- federated, autonomous media hosting |
cf29e8a8 | 2 | # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS. |
9f094bea CFD |
3 | # |
4 | # This program is free software: you can redistribute it and/or modify | |
5 | # it under the terms of the GNU Affero General Public License as published by | |
6 | # the Free Software Foundation, either version 3 of the License, or | |
7 | # (at your option) any later version. | |
8 | # | |
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU Affero General Public License for more details. | |
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | ||
17 | from mediagoblin.messages import fetch_messages, add_message | |
18 | from mediagoblin.tests.tools import setup_fresh_app | |
ae3bc7fa | 19 | from mediagoblin.tools import template |
9f094bea CFD |
20 | |
21 | ||
22 | @setup_fresh_app | |
23 | def test_messages(test_app): | |
24 | """ | |
25 | Added messages should show up in the request.session, | |
26 | fetched messages should be the same as the added ones, | |
27 | and fetching should clear the message list. | |
28 | """ | |
29 | # Aquire a request object | |
30 | test_app.get('/') | |
ae3bc7fa | 31 | context = template.TEMPLATE_TEST_CONTEXT['mediagoblin/root.html'] |
9f094bea CFD |
32 | request = context['request'] |
33 | ||
34 | # The message queue should be empty | |
35 | assert request.session.get('messages', []) == [] | |
36 | ||
37 | # Adding a message should modify the session accordingly | |
38 | add_message(request, 'herp_derp', 'First!') | |
39 | test_msg_queue = [{'text': 'First!', 'level': 'herp_derp'}] | |
40 | assert request.session['messages'] == test_msg_queue | |
41 | ||
42 | # fetch_messages should return and empty the queue | |
43 | assert fetch_messages(request) == test_msg_queue | |
44 | assert request.session.get('messages') == [] |