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])))}
  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 not 'headers' 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                 '{"username": "joapi", "email": "joapi@example.com"}'
  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()