89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2015 Objectif Libre
|
|
#
|
|
# 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.
|
|
#
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from oslo_utils import uuidutils
|
|
|
|
from cloudkitty.fetcher import keystone
|
|
from cloudkitty import tests
|
|
|
|
|
|
class FakeRole(object):
|
|
def __init__(self, name, uuid=None):
|
|
self.id = uuid if uuid else uuidutils.generate_uuid()
|
|
self.name = name
|
|
|
|
|
|
class FakeTenant(object):
|
|
def __init__(self, id):
|
|
self.id = id
|
|
|
|
|
|
class FakeKeystoneClient(object):
|
|
|
|
user_id = 'd89e3fee-2b92-4387-b564-63901d62e591'
|
|
|
|
def __init__(self, **kwargs):
|
|
pass
|
|
|
|
class FakeTenants(object):
|
|
@classmethod
|
|
def list(cls):
|
|
return [FakeTenant('f266f30b11f246b589fd266f85eeec39'),
|
|
FakeTenant('4dfb25b0947c4f5481daf7b948c14187')]
|
|
|
|
class FakeRoles(object):
|
|
roles_mapping = {
|
|
'd89e3fee-2b92-4387-b564-63901d62e591': {
|
|
'f266f30b11f246b589fd266f85eeec39': [FakeRole('rating'),
|
|
FakeRole('admin')],
|
|
'4dfb25b0947c4f5481daf7b948c14187': [FakeRole('admin')]}}
|
|
|
|
@classmethod
|
|
def roles_for_user(cls, user_id, tenant, **kwargs):
|
|
return cls.roles_mapping[user_id][tenant.id]
|
|
|
|
roles = FakeRoles()
|
|
tenants = FakeTenants()
|
|
|
|
|
|
def Client(**kwargs):
|
|
return FakeKeystoneClient(**kwargs)
|
|
|
|
|
|
class KeystoneFetcherTest(tests.TestCase):
|
|
def setUp(self):
|
|
super(KeystoneFetcherTest, self).setUp()
|
|
self.conf.set_override('backend', 'keystone', 'tenant_fetcher')
|
|
self.conf.import_group('fetcher_keystone',
|
|
'cloudkitty.fetcher.keystone')
|
|
|
|
@unittest.SkipTest
|
|
def test_fetcher_keystone_filter_list(self):
|
|
kclient = 'keystoneclient.client.Client'
|
|
with mock.patch(kclient) as kclientmock:
|
|
kclientmock.return_value = Client()
|
|
fetcher = keystone.KeystoneFetcher()
|
|
kclientmock.assert_called_once_with(
|
|
auth_url='http://127.0.0.1:5000/v2.0',
|
|
username='cloudkitty',
|
|
password='cloudkitty',
|
|
tenant_name='cloudkitty',
|
|
region_name='RegionOne')
|
|
tenants = fetcher.get_tenants()
|
|
self.assertEqual(['f266f30b11f246b589fd266f85eeec39'], tenants)
|