e07d88d40045597ae121d639772d001569f38b23
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
import messages
18 from mediagoblin
.tools
import template
19 from mediagoblin
.tools
.testing
import _activate_testing
24 def test_messages(test_app
):
26 Added messages should show up in the request.session,
27 fetched messages should be the same as the added ones,
28 and fetching should clear the message list.
30 # Aquire a request object
32 context
= template
.TEMPLATE_TEST_CONTEXT
['mediagoblin/root.html']
33 request
= context
['request']
35 # The message queue should be empty
36 assert request
.session
.get('messages', []) == []
38 # First of all, we should clear the messages queue
39 messages
.clear_add_message()
40 # Adding a message should modify the session accordingly
41 messages
.add_message(request
, 'herp_derp', 'First!')
42 test_msg_queue
= [{'text': 'First!', 'level': 'herp_derp'}]
44 # Alternative tests to the following, test divided in two steps:
45 # assert request.session['messages'] == test_msg_queue
46 # 1. Tests if add_message worked
47 assert messages
.ADD_MESSAGE_TEST
[-1] == test_msg_queue
48 # 2. Tests if add_message updated session information
49 assert messages
.ADD_MESSAGE_TEST
[-1] == request
.session
['messages']
51 # fetch_messages should return and empty the queue
52 assert messages
.fetch_messages(request
) == test_msg_queue
53 assert request
.session
.get('messages') == []