diff options
author | Vishvananda Ishaya <vishvananda@gmail.com> | 2011-08-31 12:51:16 -0700 |
---|---|---|
committer | Vishvananda Ishaya <vishvananda@gmail.com> | 2011-08-31 12:51:16 -0700 |
commit | c9758dd4832c167562baefad5dcc88f2a1a19b73 (patch) | |
tree | 1cf2d6f3bcf15868a9920687e53fbb98df277542 /nova/tests/api/openstack/fakes.py | |
parent | abd6b240b5247a2981e86c1d3161306fb2b4c4c5 (diff) | |
download | nova-c9758dd4832c167562baefad5dcc88f2a1a19b73.tar.gz |
fix keypairs stubs
Diffstat (limited to 'nova/tests/api/openstack/fakes.py')
-rw-r--r-- | nova/tests/api/openstack/fakes.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/nova/tests/api/openstack/fakes.py b/nova/tests/api/openstack/fakes.py index a095dd90ad..a4a90e6da7 100644 --- a/nova/tests/api/openstack/fakes.py +++ b/nova/tests/api/openstack/fakes.py @@ -107,11 +107,18 @@ def stub_out_key_pair_funcs(stubs, have_key_pair=True): def key_pair(context, user_id): return [dict(name='key', public_key='public_key')] + def one_key_pair(context, user_id, name): + if name == 'key': + return dict(name='key', public_key='public_key') + else: + raise exc.KeypairNotFound(user_id=user_id, name=name) + def no_key_pair(context, user_id): return [] if have_key_pair: stubs.Set(nova.db, 'key_pair_get_all_by_user', key_pair) + stubs.Set(nova.db, 'key_pair_get', one_key_pair) else: stubs.Set(nova.db, 'key_pair_get_all_by_user', no_key_pair) |