def _get_application(self, client_id, request): """ Load application instance for given client_id and store it in request as 'client' attribute """ assert hasattr(request, 'client'), "'client' attribute missing from 'request'" Application = get_application_model() try: request.client = request.client or Application.objects.get(client_id=client_id) return request.client except Application.DoesNotExist: return None
from django.contrib.auth import get_user_model from django.core.urlresolvers import reverse from django.utils import timezone from oauthlib.oauth2 import (InvalidClientIdError, MissingClientIdError, InvalidRedirectURIError, MismatchingRedirectURIError) from rest_framework import status from oauth_api.models import get_application_model, AuthorizationCode, RefreshToken from oauth_api.settings import oauth_api_settings from oauth_api.tests.utils import TestCaseUtils from oauth_api.tests.views import RESPONSE_DATA Application = get_application_model() User = get_user_model() class BaseTest(TestCaseUtils): def setUp(self): self.test_user = User.objects.create_user('test_user', '*****@*****.**', '1234') self.dev_user = User.objects.create_user('dev_user', '*****@*****.**', '1234') self.application = Application( name='Test Application', redirect_uris='http://localhost http://example.com', user=self.dev_user, client_type=Application.CLIENT_CONFIDENTIAL, authorization_grant_type=Application.GRANT_AUTHORIZATION_CODE, ) self.application.save()
from django.contrib.auth import get_user_model from django.urls import reverse from django.utils import timezone from rest_framework import status from rest_framework.test import APIRequestFactory from oauth_api.models import get_application_model, AccessToken, RefreshToken from oauth_api.tests.utils import TestCaseUtils Application = get_application_model() User = get_user_model() class BaseTest(TestCaseUtils): @classmethod def setUpTestData(cls): cls.test_user = User.objects.create_user('test_user', '*****@*****.**', '1234') cls.dev_user = User.objects.create_user('dev_user', '*****@*****.**', '1234') cls.application = Application( name='Test Application', redirect_uris='http://localhost http://example.com', user=cls.dev_user, client_type=Application.CLIENT_CONFIDENTIAL, authorization_grant_type=Application.GRANT_AUTHORIZATION_CODE, )