Fix another tests.
[mediagoblin:mediagoblin.git] / mediagoblin / tests / test_api.py
1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
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
18 import logging
19 import base64
20
21 import pytest
22
23 from mediagoblin import mg_globals
24 from mediagoblin.tools import template, pluginapi
25 from mediagoblin.tests.tools import fixture_add_user
26 from .resources import GOOD_JPG, GOOD_PNG, EVIL_FILE, EVIL_JPG, EVIL_PNG, \
27     BIG_BLUE
28
29
30 _log = logging.getLogger(__name__)
31
32
33 class TestAPI(object):
34     def setup(self):
35         self.db = mg_globals.database
36
37         self.user_password = u'4cc355_70k3N'
38         self.user = fixture_add_user(u'joapi', self.user_password,
39             privileges=[u'active',u'uploader'])
40
41     def login(self, test_app):
42         test_app.post(
43             '/auth/login/', {
44                 'username': self.user.username,
45                 'password': self.user_password})
46
47     def get_context(self, template_name):
48         return template.TEMPLATE_TEST_CONTEXT[template_name]
49
50     def http_auth_headers(self):
51         return {'Authorization': ('Basic {0}'.format(
52                 base64.b64encode((':'.join([
53                     self.user.username,
54                     self.user_password])).encode('ascii')).decode()))}
55
56     def do_post(self, data, test_app, **kwargs):
57         url = kwargs.pop('url', '/api/submit')
58         do_follow = kwargs.pop('do_follow', False)
59
60         if 'headers' not in kwargs.keys():
61             kwargs['headers'] = self.http_auth_headers()
62
63         response = test_app.post(url, data, **kwargs)
64
65         if do_follow:
66             response.follow()
67
68         return response
69
70     def upload_data(self, filename):
71         return {'upload_files': [('file', filename)]}
72
73     def test_1_test_test_view(self, test_app):
74         self.login(test_app)
75
76         response = test_app.get(
77             '/api/test',
78             headers=self.http_auth_headers())
79
80         assert response.body == \
81                 b'{"email": "joapi@example.com", "username": "joapi"}'
82
83     def test_2_test_submission(self, test_app):
84         self.login(test_app)
85
86         response = self.do_post(
87             {'title': 'Great JPG!'},
88             test_app,
89             **self.upload_data(GOOD_JPG))
90
91         assert response.status_int == 200
92
93         assert self.db.MediaEntry.query.filter_by(title=u'Great JPG!').first()