github.com/amrnt/deis@v1.3.1/controller/api/tests/test_key.py (about)

     1  # -*- coding: utf-8 -*-
     2  """
     3  Unit tests for the Deis api app.
     4  
     5  Run the tests with "./manage.py test api"
     6  """
     7  
     8  from __future__ import unicode_literals
     9  
    10  import json
    11  
    12  from django.contrib.auth.models import User
    13  from django.test import TestCase
    14  from rest_framework.authtoken.models import Token
    15  
    16  from api.models import Key
    17  from api.utils import fingerprint
    18  
    19  
    20  RSA_PUBKEY = (
    21      "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQCfQkkUUoxpvcNMkvv7jqnfodgs37M2eBO"
    22      "APgLK+KNBMaZaaKB4GF1QhTCMfFhoiTW3rqa0J75bHJcdkoobtTHlK8XUrFqsquWyg3XhsT"
    23      "Yr/3RQQXvO86e2sF7SVDJqVtpnbQGc5SgNrHCeHJmf5HTbXSIjCO/AJSvIjnituT/SIAMGe"
    24      "Bw0Nq/iSltwYAek1hiKO7wSmLcIQ8U4A00KEUtalaumf2aHOcfjgPfzlbZGP0S0cuBwSqLr"
    25      "8b5XGPmkASNdUiuJY4MJOce7bFU14B7oMAy2xacODUs1momUeYtGI9T7X2WMowJaO7tP3Gl"
    26      "sgBMP81VfYTfYChAyJpKp2yoP autotest@autotesting comment"
    27  )
    28  
    29  ECDSA_PUBKEY = (
    30      "ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAAB"
    31      "BBCGB0x9lmubbLJTF5NekCI0Cgjyip6jJh/t/qQQi1LAZisbREBJ8Wy+hwSn3tnbf/Imh9X"
    32      "+MQnrrza0jaQ3QUAQ= autotest@autotesting comment"
    33  )
    34  
    35  BAD_KEY = (
    36      "ssh-rsa foo_bar"
    37  )
    38  
    39  
    40  class KeyTest(TestCase):
    41  
    42      """Tests cloud provider credentials"""
    43  
    44      fixtures = ['tests.json']
    45  
    46      def setUp(self):
    47          self.user = User.objects.get(username='autotest')
    48          self.token = Token.objects.get(user=self.user).key
    49  
    50      def _check_key(self, pubkey):
    51          """
    52          Test that a user can add, remove and manage their SSH public keys
    53          """
    54          url = '/v1/keys'
    55          body = {'id': 'mykey@box.local', 'public': pubkey}
    56          response = self.client.post(url, json.dumps(body), content_type='application/json',
    57                                      HTTP_AUTHORIZATION='token {}'.format(self.token))
    58          self.assertEqual(response.status_code, 201)
    59          key_id = response.data['id']
    60          response = self.client.get(url, HTTP_AUTHORIZATION='token {}'.format(self.token))
    61          self.assertEqual(response.status_code, 200)
    62          self.assertEqual(len(response.data['results']), 1)
    63          url = '/v1/keys/{key_id}'.format(**locals())
    64          response = self.client.get(url, HTTP_AUTHORIZATION='token {}'.format(self.token))
    65          self.assertEqual(response.status_code, 200)
    66          self.assertEqual(body['id'], response.data['id'])
    67          self.assertEqual(body['public'], response.data['public'])
    68          response = self.client.delete(url, HTTP_AUTHORIZATION='token {}'.format(self.token))
    69          self.assertEqual(response.status_code, 204)
    70  
    71      def _check_bad_key(self, pubkey):
    72          """
    73          Test that a user cannot add invalid SSH public keys
    74          """
    75          url = '/v1/keys'
    76          body = {'id': 'mykey@box.local', 'public': pubkey}
    77          response = self.client.post(url, json.dumps(body), content_type='application/json',
    78                                      HTTP_AUTHORIZATION='token {}'.format(self.token))
    79          self.assertEqual(response.status_code, 400)
    80          return response
    81  
    82      def test_rsa_key(self):
    83          self._check_key(RSA_PUBKEY)
    84  
    85      def test_ecdsa_key(self):
    86          self._check_key(ECDSA_PUBKEY)
    87  
    88      def test_bad_key(self):
    89          response = self._check_bad_key(BAD_KEY)
    90          self.assertEqual(response.data, {'public': ['Incorrect padding']})
    91  
    92      def _check_duplicate_key(self, pubkey):
    93          """
    94          Test that a user cannot add a duplicate key
    95          """
    96          url = '/v1/keys'
    97          body = {'id': 'mykey@box.local', 'public': pubkey}
    98          response = self.client.post(url, json.dumps(body), content_type='application/json',
    99                                      HTTP_AUTHORIZATION='token {}'.format(self.token))
   100          self.assertEqual(response.status_code, 201)
   101          response = self.client.post(url, json.dumps(body), content_type='application/json',
   102                                      HTTP_AUTHORIZATION='token {}'.format(self.token))
   103          self.assertEqual(response.status_code, 400)
   104  
   105      def test_rsa_duplicate_key(self):
   106          self._check_duplicate_key(RSA_PUBKEY)
   107  
   108      def test_ecdsa_duplicate_key(self):
   109          self._check_duplicate_key(ECDSA_PUBKEY)
   110  
   111      def test_rsa_key_str(self):
   112          """Test the text representation of a key"""
   113          url = '/v1/keys'
   114          body = {'id': 'autotest', 'public':
   115                  'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDzqPAwHN70xsB0LXG//KzO'
   116                  'gcPikyhdN/KRc4x3j/RA0pmFj63Ywv0PJ2b1LcMSqfR8F11WBlrW8c9xFua0'
   117                  'ZAKzI+gEk5uqvOR78bs/SITOtKPomW4e/1d2xEkJqOmYH30u94+NZZYwEBqY'
   118                  'aRb34fhtrnJS70XeGF0RhXE5Qea5eh7DBbeLxPfSYd8rfHgzMSb/wmx3h2vm'
   119                  'HdQGho20pfJktNu7DxeVkTHn9REMUphf85su7slTgTlWKq++3fASE8PdmFGz'
   120                  'b6PkOR4c+LS5WWXd2oM6HyBQBxxiwXbA2lSgQxOdgDiM2FzT0GVSFMUklkUH'
   121                  'MdsaG6/HJDw9QckTS0vN autotest@deis.io'}
   122          response = self.client.post(url, json.dumps(body), content_type='application/json',
   123                                      HTTP_AUTHORIZATION='token {}'.format(self.token))
   124          self.assertEqual(response.status_code, 201)
   125          key = Key.objects.get(uuid=response.data['uuid'])
   126          self.assertEqual(str(key), 'ssh-rsa AAAAB3NzaC.../HJDw9QckTS0vN autotest@deis.io')
   127  
   128      def test_rsa_key_fingerprint(self):
   129          fp = fingerprint(RSA_PUBKEY)
   130          self.assertEquals(fp, '54:6d:da:1f:91:b5:2b:6f:a2:83:90:c4:f9:73:76:f5')