1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
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.
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.
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/>.
17 from mediagoblin
.messages
import fetch_messages
, add_message
18 from mediagoblin
.tests
.tools
import setup_fresh_app
19 from mediagoblin
.tools
import template
23 def test_messages(test_app
):
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.
29 # Aquire a request object
31 context
= template
.TEMPLATE_TEST_CONTEXT
['mediagoblin/root.html']
32 request
= context
['request']
34 # The message queue should be empty
35 assert request
.session
.get('messages', []) == []
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
42 # fetch_messages should return and empty the queue
43 assert fetch_messages(request
) == test_msg_queue
44 assert request
.session
.get('messages') == []