Merge "Keypair API test cleanup"
This commit is contained in:
Jenkins
Gerrit Code Review
commit
c25300009a
@ -8338,124 +8338,6 @@ class ComputePolicyTestCase(BaseTestCase):
|
|||||||
availability_zone='1:1')
|
availability_zone='1:1')
|
||||||
|
|
||||||
|
|
||||||
class KeypairAPITestCase(BaseTestCase):
|
|
||||||
def setUp(self):
|
|
||||||
super(KeypairAPITestCase, self).setUp()
|
|
||||||
self.keypair_api = compute_api.KeypairAPI()
|
|
||||||
self.ctxt = context.RequestContext('fake', 'fake')
|
|
||||||
self._keypair_db_call_stubs()
|
|
||||||
self.existing_key_name = 'fake existing key name'
|
|
||||||
self.pub_key = ('ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDLnVkqJu9WVf'
|
|
||||||
'/5StU3JCrBR2r1s1j8K1tux+5XeSvdqaM8lMFNorzbY5iyoBbR'
|
|
||||||
'S56gy1jmm43QsMPJsrpfUZKcJpRENSe3OxIIwWXRoiapZe78u/'
|
|
||||||
'a9xKwj0avFYMcws9Rk9iAB7W4K1nEJbyCPl5lRBoyqeHBqrnnu'
|
|
||||||
'XWEgGxJCK0Ah6wcOzwlEiVjdf4kxzXrwPHyi7Ea1qvnNXTziF8'
|
|
||||||
'yYmUlH4C8UXfpTQckwSwpDyxZUc63P8q+vPbs3Q2kw+/7vvkCK'
|
|
||||||
'HJAXVI+oCiyMMfffoTq16M1xfV58JstgtTqAXG+ZFpicGajREU'
|
|
||||||
'E/E3hO5MGgcHmyzIrWHKpe1n3oEGuz')
|
|
||||||
self.fingerprint = '4e:48:c6:a0:4a:f9:dd:b5:4c:85:54:5a:af:43:47:5a'
|
|
||||||
|
|
||||||
def _keypair_db_call_stubs(self):
|
|
||||||
|
|
||||||
def db_key_pair_get_all_by_user(self, user_id):
|
|
||||||
return []
|
|
||||||
|
|
||||||
def db_key_pair_create(self, keypair):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def db_key_pair_destroy(context, user_id, name):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def db_key_pair_get(context, user_id, name):
|
|
||||||
if name == self.existing_key_name:
|
|
||||||
return {'name': self.existing_key_name,
|
|
||||||
'public_key': self.pub_key,
|
|
||||||
'fingerprint': self.fingerprint}
|
|
||||||
else:
|
|
||||||
raise exception.KeypairNotFound(user_id=user_id, name=name)
|
|
||||||
|
|
||||||
self.stubs.Set(db, "key_pair_get_all_by_user",
|
|
||||||
db_key_pair_get_all_by_user)
|
|
||||||
self.stubs.Set(db, "key_pair_create",
|
|
||||||
db_key_pair_create)
|
|
||||||
self.stubs.Set(db, "key_pair_destroy",
|
|
||||||
db_key_pair_destroy)
|
|
||||||
self.stubs.Set(db, "key_pair_get",
|
|
||||||
db_key_pair_get)
|
|
||||||
|
|
||||||
def test_create_keypair(self):
|
|
||||||
keypair = self.keypair_api.create_key_pair(self.ctxt,
|
|
||||||
self.ctxt.user_id, 'foo')
|
|
||||||
self.assertEqual('foo', keypair['name'])
|
|
||||||
|
|
||||||
def test_create_keypair_name_too_long(self):
|
|
||||||
self.assertRaises(exception.InvalidKeypair,
|
|
||||||
self.keypair_api.create_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, 'x' * 256)
|
|
||||||
|
|
||||||
def test_create_keypair_invalid_chars(self):
|
|
||||||
self.assertRaises(exception.InvalidKeypair,
|
|
||||||
self.keypair_api.create_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, '* BAD CHARACTERS! *')
|
|
||||||
|
|
||||||
def test_create_keypair_already_exists(self):
|
|
||||||
def db_key_pair_create_duplicate(context, keypair):
|
|
||||||
raise exception.KeyPairExists(key_name=keypair.get('name', ''))
|
|
||||||
self.stubs.Set(db, "key_pair_create", db_key_pair_create_duplicate)
|
|
||||||
self.assertRaises(exception.KeyPairExists,
|
|
||||||
self.keypair_api.create_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id,
|
|
||||||
self.existing_key_name)
|
|
||||||
|
|
||||||
def test_create_keypair_quota_limit(self):
|
|
||||||
def fake_quotas_count(self, context, resource, *args, **kwargs):
|
|
||||||
return CONF.quota_key_pairs
|
|
||||||
self.stubs.Set(QUOTAS, "count", fake_quotas_count)
|
|
||||||
self.assertRaises(exception.KeypairLimitExceeded,
|
|
||||||
self.keypair_api.create_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, 'foo')
|
|
||||||
|
|
||||||
def test_import_keypair(self):
|
|
||||||
keypair = self.keypair_api.import_key_pair(self.ctxt,
|
|
||||||
self.ctxt.user_id,
|
|
||||||
'foo',
|
|
||||||
self.pub_key)
|
|
||||||
self.assertEqual('foo', keypair['name'])
|
|
||||||
self.assertEqual(self.fingerprint, keypair['fingerprint'])
|
|
||||||
self.assertEqual(self.pub_key, keypair['public_key'])
|
|
||||||
|
|
||||||
def test_import_keypair_bad_public_key(self):
|
|
||||||
self.assertRaises(exception.InvalidKeypair,
|
|
||||||
self.keypair_api.import_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, 'foo', 'bad key data')
|
|
||||||
|
|
||||||
def test_import_keypair_name_too_long(self):
|
|
||||||
self.assertRaises(exception.InvalidKeypair,
|
|
||||||
self.keypair_api.import_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, 'x' * 256,
|
|
||||||
self.pub_key)
|
|
||||||
|
|
||||||
def test_import_keypair_invalid_chars(self):
|
|
||||||
self.assertRaises(exception.InvalidKeypair,
|
|
||||||
self.keypair_api.import_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id,
|
|
||||||
'* BAD CHARACTERS! *', self.pub_key)
|
|
||||||
|
|
||||||
def test_import_keypair_quota_limit(self):
|
|
||||||
def fake_quotas_count(self, context, resource, *args, **kwargs):
|
|
||||||
return CONF.quota_key_pairs
|
|
||||||
self.stubs.Set(QUOTAS, "count", fake_quotas_count)
|
|
||||||
self.assertRaises(exception.KeypairLimitExceeded,
|
|
||||||
self.keypair_api.import_key_pair,
|
|
||||||
self.ctxt, self.ctxt.user_id, 'foo', self.pub_key)
|
|
||||||
|
|
||||||
def test_get_keypair(self):
|
|
||||||
keypair = self.keypair_api.get_key_pair(self.ctxt,
|
|
||||||
self.ctxt.user_id,
|
|
||||||
self.existing_key_name)
|
|
||||||
self.assertEqual(self.existing_key_name, keypair['name'])
|
|
||||||
|
|
||||||
|
|
||||||
class DisabledInstanceTypesTestCase(BaseTestCase):
|
class DisabledInstanceTypesTestCase(BaseTestCase):
|
||||||
"""
|
"""
|
||||||
Some instance-types are marked 'disabled' which means that they will not
|
Some instance-types are marked 'disabled' which means that they will not
|
||||||
|
|||||||
154
nova/tests/compute/test_keypairs.py
Normal file
154
nova/tests/compute/test_keypairs.py
Normal file
@ -0,0 +1,154 @@
|
|||||||
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
||||||
|
|
||||||
|
# Copyright 2013 OpenStack Foundation
|
||||||
|
# All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||||||
|
# not use this file except in compliance with the License. You may obtain
|
||||||
|
# a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
# License for the specific language governing permissions and limitations
|
||||||
|
# under the License.
|
||||||
|
"""Tests for keypair API."""
|
||||||
|
|
||||||
|
from oslo.config import cfg
|
||||||
|
|
||||||
|
from nova.compute import api as compute_api
|
||||||
|
from nova import context
|
||||||
|
from nova import db
|
||||||
|
from nova import exception
|
||||||
|
from nova import quota
|
||||||
|
from nova.tests.compute import test_compute
|
||||||
|
|
||||||
|
|
||||||
|
CONF = cfg.CONF
|
||||||
|
QUOTAS = quota.QUOTAS
|
||||||
|
|
||||||
|
|
||||||
|
class KeypairAPITestCase(test_compute.BaseTestCase):
|
||||||
|
def setUp(self):
|
||||||
|
super(KeypairAPITestCase, self).setUp()
|
||||||
|
self.keypair_api = compute_api.KeypairAPI()
|
||||||
|
self.ctxt = context.RequestContext('fake', 'fake')
|
||||||
|
self._keypair_db_call_stubs()
|
||||||
|
self.existing_key_name = 'fake existing key name'
|
||||||
|
self.pub_key = ('ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDLnVkqJu9WVf'
|
||||||
|
'/5StU3JCrBR2r1s1j8K1tux+5XeSvdqaM8lMFNorzbY5iyoBbR'
|
||||||
|
'S56gy1jmm43QsMPJsrpfUZKcJpRENSe3OxIIwWXRoiapZe78u/'
|
||||||
|
'a9xKwj0avFYMcws9Rk9iAB7W4K1nEJbyCPl5lRBoyqeHBqrnnu'
|
||||||
|
'XWEgGxJCK0Ah6wcOzwlEiVjdf4kxzXrwPHyi7Ea1qvnNXTziF8'
|
||||||
|
'yYmUlH4C8UXfpTQckwSwpDyxZUc63P8q+vPbs3Q2kw+/7vvkCK'
|
||||||
|
'HJAXVI+oCiyMMfffoTq16M1xfV58JstgtTqAXG+ZFpicGajREU'
|
||||||
|
'E/E3hO5MGgcHmyzIrWHKpe1n3oEGuz')
|
||||||
|
self.fingerprint = '4e:48:c6:a0:4a:f9:dd:b5:4c:85:54:5a:af:43:47:5a'
|
||||||
|
|
||||||
|
def _keypair_db_call_stubs(self):
|
||||||
|
|
||||||
|
def db_key_pair_get_all_by_user(self, user_id):
|
||||||
|
return []
|
||||||
|
|
||||||
|
def db_key_pair_create(self, keypair):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def db_key_pair_destroy(context, user_id, name):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def db_key_pair_get(context, user_id, name):
|
||||||
|
if name == self.existing_key_name:
|
||||||
|
return {'name': self.existing_key_name,
|
||||||
|
'public_key': self.pub_key,
|
||||||
|
'fingerprint': self.fingerprint}
|
||||||
|
else:
|
||||||
|
raise exception.KeypairNotFound(user_id=user_id, name=name)
|
||||||
|
|
||||||
|
self.stubs.Set(db, "key_pair_get_all_by_user",
|
||||||
|
db_key_pair_get_all_by_user)
|
||||||
|
self.stubs.Set(db, "key_pair_create",
|
||||||
|
db_key_pair_create)
|
||||||
|
self.stubs.Set(db, "key_pair_destroy",
|
||||||
|
db_key_pair_destroy)
|
||||||
|
self.stubs.Set(db, "key_pair_get",
|
||||||
|
db_key_pair_get)
|
||||||
|
|
||||||
|
|
||||||
|
class CreateImportSharedTestMixIn(object):
|
||||||
|
"""Tests shared between create and import_key.
|
||||||
|
|
||||||
|
Mix-in pattern is used here so that these `test_*` methods aren't picked
|
||||||
|
up by the test runner unless they are part of a 'concrete' test case.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def assertKeyNameRaises(self, exc_class, name):
|
||||||
|
func = getattr(self.keypair_api, self.func_name)
|
||||||
|
|
||||||
|
args = []
|
||||||
|
if self.func_name == 'import_key_pair':
|
||||||
|
args.append(self.pub_key)
|
||||||
|
|
||||||
|
self.assertRaises(exc_class, func, self.ctxt, self.ctxt.user_id,
|
||||||
|
name, *args)
|
||||||
|
|
||||||
|
def test_name_too_short(self):
|
||||||
|
self.assertKeyNameRaises(exception.InvalidKeypair, '')
|
||||||
|
|
||||||
|
def test_name_too_long(self):
|
||||||
|
self.assertKeyNameRaises(exception.InvalidKeypair, 'x' * 256)
|
||||||
|
|
||||||
|
def test_invalid_chars(self):
|
||||||
|
self.assertKeyNameRaises(exception.InvalidKeypair,
|
||||||
|
'* BAD CHARACTERS! *')
|
||||||
|
|
||||||
|
def test_already_exists(self):
|
||||||
|
def db_key_pair_create_duplicate(context, keypair):
|
||||||
|
raise exception.KeyPairExists(key_name=keypair.get('name', ''))
|
||||||
|
|
||||||
|
self.stubs.Set(db, "key_pair_create", db_key_pair_create_duplicate)
|
||||||
|
self.assertKeyNameRaises(exception.KeyPairExists,
|
||||||
|
self.existing_key_name)
|
||||||
|
|
||||||
|
def test_quota_limit(self):
|
||||||
|
def fake_quotas_count(self, context, resource, *args, **kwargs):
|
||||||
|
return CONF.quota_key_pairs
|
||||||
|
|
||||||
|
self.stubs.Set(QUOTAS, "count", fake_quotas_count)
|
||||||
|
self.assertKeyNameRaises(exception.KeypairLimitExceeded, 'foo')
|
||||||
|
|
||||||
|
|
||||||
|
class CreateKeypairTestCase(KeypairAPITestCase, CreateImportSharedTestMixIn):
|
||||||
|
func_name = 'create_key_pair'
|
||||||
|
|
||||||
|
def test_success(self):
|
||||||
|
keypair = self.keypair_api.create_key_pair(self.ctxt,
|
||||||
|
self.ctxt.user_id, 'foo')
|
||||||
|
self.assertEqual('foo', keypair['name'])
|
||||||
|
|
||||||
|
|
||||||
|
class ImportKeypairTestCase(KeypairAPITestCase, CreateImportSharedTestMixIn):
|
||||||
|
func_name = 'import_key_pair'
|
||||||
|
|
||||||
|
def test_success(self):
|
||||||
|
keypair = self.keypair_api.import_key_pair(self.ctxt,
|
||||||
|
self.ctxt.user_id,
|
||||||
|
'foo',
|
||||||
|
self.pub_key)
|
||||||
|
self.assertEqual('foo', keypair['name'])
|
||||||
|
self.assertEqual(self.fingerprint, keypair['fingerprint'])
|
||||||
|
self.assertEqual(self.pub_key, keypair['public_key'])
|
||||||
|
|
||||||
|
def test_bad_key_data(self):
|
||||||
|
self.assertRaises(exception.InvalidKeypair,
|
||||||
|
self.keypair_api.import_key_pair,
|
||||||
|
self.ctxt, self.ctxt.user_id, 'foo', 'bad key data')
|
||||||
|
|
||||||
|
|
||||||
|
class GetKeypairTestCase(KeypairAPITestCase):
|
||||||
|
def test_success(self):
|
||||||
|
keypair = self.keypair_api.get_key_pair(self.ctxt,
|
||||||
|
self.ctxt.user_id,
|
||||||
|
self.existing_key_name)
|
||||||
|
self.assertEqual(self.existing_key_name, keypair['name'])
|
||||||
Loading…
x
Reference in New Issue
Block a user