github.com/spg/deis@v1.7.3/controller/api/tests/test_users.py (about)

     1  
     2  from __future__ import unicode_literals
     3  
     4  from django.contrib.auth.models import User
     5  from django.test import TestCase
     6  from rest_framework.authtoken.models import Token
     7  
     8  
     9  class TestUsers(TestCase):
    10      """ Tests users endpoint"""
    11  
    12      fixtures = ['tests.json']
    13  
    14      def test_super_user_can_list(self):
    15          url = '/v1/users/'
    16  
    17          user = User.objects.get(username='autotest')
    18          token = Token.objects.get(user=user)
    19  
    20          response = self.client.get(url,
    21                                     HTTP_AUTHORIZATION='token {}'.format(token))
    22  
    23          self.assertEqual(response.status_code, 200)
    24          self.assertEqual(len(response.data['results']), 2)
    25          self.assertEqual(response.data['results'][0]['username'], 'autotest')
    26          self.assertEqual(response.data['results'][1]['username'], 'autotest2')
    27  
    28      def test_non_super_user_cannot_list(self):
    29          url = '/v1/users/'
    30  
    31          user = User.objects.get(username='autotest2')
    32          token = Token.objects.get(user=user)
    33  
    34          response = self.client.get(url,
    35                                     HTTP_AUTHORIZATION='token {}'.format(token))
    36          self.assertEqual(response.status_code, 403)