github.com/jiasir/deis@v1.12.2/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']), 3)
    25  
    26      def test_non_super_user_cannot_list(self):
    27          url = '/v1/users/'
    28  
    29          user = User.objects.get(username='autotest2')
    30          token = Token.objects.get(user=user)
    31  
    32          response = self.client.get(url,
    33                                     HTTP_AUTHORIZATION='token {}'.format(token))
    34          self.assertEqual(response.status_code, 403)