mediagoblin/tests/test_http_callback.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 import json
  18
  19 import pytest
  20 from urlparse import urlparse, parse_qs
  21
  22 from mediagoblin import mg_globals
  23 from mediagoblin.tools import processing
  24 from mediagoblin.tests.tools import fixture_add_user
  25 from mediagoblin.tests.test_submission import GOOD_PNG
  26 from mediagoblin.tests import test_oauth as oauth
  27
  28
  29 class TestHTTPCallback(object):
  30     @pytest.fixture(autouse=True)
  31     def setup(self, test_app):
  32         self.test_app = test_app
  33
  34         self.db = mg_globals.database
  35
  36         self.user_password = u'secret'
  37         self.user = fixture_add_user(u'call_back', self.user_password)
  38
  39         self.login()
  40
  41     def login(self):
  42         self.test_app.post('/auth/login/', {
  43             'username': self.user.username,
  44             'password': self.user_password})
  45
  46     def get_access_token(self, client_id, client_secret, code):
  47         response = self.test_app.get('/oauth/access_token', {
  48                 'code': code,
  49                 'client_id': client_id,
  50                 'client_secret': client_secret})
  51
  52         response_data = json.loads(response.body)
  53
  54         return response_data['access_token']
  55
  56     def test_callback(self):
  57         ''' Test processing HTTP callback '''
  58         self.oauth = oauth.TestOAuth()
  59         self.oauth.setup(self.test_app)
  60
  61         redirect, client_id = self.oauth.test_4_authorize_confidential_client()
  62
  63         code = parse_qs(urlparse(redirect.location).query)['code'][0]
  64
  65         client = self.db.OAuthClient.query.filter(
  66                 self.db.OAuthClient.identifier == unicode(client_id)).first()
  67
  68         client_secret = client.secret
  69
  70         access_token = self.get_access_token(client_id, client_secret, code)
  71
  72         callback_url = 'https://foo.example?secrettestmediagoblinparam'
  73
  74         self.test_app.post('/api/submit?client_id={0}&access_token={1}\
  75 &client_secret={2}'.format(
  76                     client_id,
  77                     access_token,
  78                     client_secret), {
  79             'title': 'Test',
  80             'callback_url': callback_url},
  81             upload_files=[('file', GOOD_PNG)])
  82
  83         assert processing.TESTS_CALLBACKS[callback_url]['state'] == u'processed'