From 202415829afcbcb48827e15db725050d19eb1b84 Mon Sep 17 00:00:00 2001 From: Rick Harris Date: Thu, 6 Jun 2013 21:48:21 +0000 Subject: More KeypairAPI cleanups * DRY up quota checking for new keys * DRY up formatting of get_key/get_all_keys results * Add tests for get all keypairs Change-Id: I14a97c0f3cb3aa9b827d14002c21076a41942023 --- nova/tests/compute/test_keypairs.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) (limited to 'nova/tests') diff --git a/nova/tests/compute/test_keypairs.py b/nova/tests/compute/test_keypairs.py index fcb21b3e6..f82d69ccb 100644 --- a/nova/tests/compute/test_keypairs.py +++ b/nova/tests/compute/test_keypairs.py @@ -49,10 +49,12 @@ class KeypairAPITestCase(test_compute.BaseTestCase): def _keypair_db_call_stubs(self): - def db_key_pair_get_all_by_user(self, user_id): - return [] + def db_key_pair_get_all_by_user(context, user_id): + return [{'name': self.existing_key_name, + 'public_key': self.pub_key, + 'fingerprint': self.fingerprint}] - def db_key_pair_create(self, keypair): + def db_key_pair_create(context, keypair): pass def db_key_pair_destroy(context, user_id, name): @@ -163,3 +165,10 @@ class GetKeypairTestCase(KeypairAPITestCase): self.ctxt.user_id, self.existing_key_name) self.assertEqual(self.existing_key_name, keypair['name']) + + +class GetKeypairsTestCase(KeypairAPITestCase): + def test_success(self): + keypairs = self.keypair_api.get_key_pairs(self.ctxt, self.ctxt.user_id) + self.assertEqual([self.existing_key_name], + [k['name'] for k in keypairs]) -- cgit