cleanup code

Author: Massimiliano

oauth2_provider/tests/test_decorators.py

@@ -36,14 +36,6 @@ def setUp(self):
             application=self.application
         )
 
-        self.another_access_token = AccessToken.objects.create(
-            user=self.user,
-            scope='can_touch_this',
-            expires=timezone.now() + timedelta(seconds=300),
-            token='secret-access-token-key2',
-            application=self.application
-        )
-
         oauth2_settings._SCOPES = ['read', 'write']
 
     def test_access_denied(self):
@@ -72,15 +64,17 @@ def scoped_view(request, *args, **kwargs):
         self.assertEqual(response, "protected contents")
 
         # now with scopes
+        self.access_token.scope = 'can_touch_this'
+        self.access_token.save()
         auth_headers = {
-            'HTTP_AUTHORIZATION': 'Bearer ' + self.another_access_token.token,
+            'HTTP_AUTHORIZATION': 'Bearer ' + self.access_token.token,
         }
         request = self.request_factory.get("/fake-resource", **auth_headers)
         response = scoped_view(request)
         self.assertEqual(response, "moar protected contents")
 
     def test_rw_protected(self):
-        self.access_token.scope = 'read'
+        self.access_token.scope = 'write'
         self.access_token.save()
         auth_headers = {
             'HTTP_AUTHORIZATION': 'Bearer ' + self.access_token.token,
@@ -92,12 +86,8 @@ def scoped_view(request, *args, **kwargs):
 
         request = self.request_factory.post("/fake-resource", **auth_headers)
         response = scoped_view(request)
-        self.assertEqual(response.status_code, 403)
-
-        @rw_protected_resource()
-        def scoped_view(request, *args, **kwargs):
-            return 'other protected contents'
+        self.assertEqual(response, "other protected contents")
 
         request = self.request_factory.get("/fake-resource", **auth_headers)
         response = scoped_view(request)
-        self.assertEqual(response, "other protected contents")
+        self.assertEqual(response.status_code, 403)