def stub_out_key_pair_funcs(stubs, have_key_pair=True, **kwargs):

    def key_pair(context, user_id):
        return [dict(test_keypair.fake_keypair, name='key', public_key='public_key', **kwargs)]

    def one_key_pair(context, user_id, name):
        if (name == 'key'):
            return dict(test_keypair.fake_keypair, name='key', public_key='public_key', **kwargs)
        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)
def stub_out_key_pair_funcs(stubs, have_key_pair=True, **kwargs):

    def key_pair(context, user_id):
        return [dict(test_keypair.fake_keypair, name='key', public_key='public_key', **kwargs)]

    def one_key_pair(context, user_id, name):
        if (name == 'key'):
            return dict(test_keypair.fake_keypair, name='key', public_key='public_key', **kwargs)
        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)