rally/tests/test_osclients.py
Hugh Saunders 1046a985b7 Move rally.test to tests.test
This will separate totally tests/code base directories

Change-Id: I2089f13d9f7e864ee7bb03ebfd937049c7acb0a5
trello: https://trello.com/c/2pzdYz6Z
2014-01-06 16:24:00 +00:00

88 lines
4.0 KiB
Python

# Copyright 2013: Mirantis Inc.
# 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.
import mock
from rally import osclients
from tests import fakes
from tests import test
class OSClientsTestCase(test.TestCase):
def _get_auth_params(self):
args = ['user', 'pass', 'tenant', 'http://auth_url']
keys = ['username', 'password', 'tenant_name', 'auth_url']
return (args, dict(zip(keys, args)))
def setUp(self):
super(OSClientsTestCase, self).setUp()
self.args, self.kwargs = self._get_auth_params()
self.clients = osclients.Clients(*self.args)
def test_init(self):
self.assertEqual(self.kwargs, self.clients.kw)
def test_get_keystone_client(self):
with mock.patch('rally.osclients.keystone') as mock_keystone:
fake_keystone = fakes.FakeKeystoneClient()
mock_keystone.Client = mock.MagicMock(return_value=fake_keystone)
self.assertTrue("keystone" not in self.clients.cache)
client = self.clients.get_keystone_client()
self.assertEqual(client, fake_keystone)
endpoint = {"endpoint": "http://auth_url:35357"}
kwargs = dict(self.kwargs.items() + endpoint.items())
mock_keystone.Client.assert_called_once_with(**kwargs)
self.assertEqual(self.clients.cache["keystone"], fake_keystone)
def test_get_nova_client(self):
with mock.patch('rally.osclients.nova') as mock_nova:
fake_nova = fakes.FakeNovaClient()
mock_nova.Client = mock.MagicMock(return_value=fake_nova)
self.assertTrue("nova" not in self.clients.cache)
client = self.clients.get_nova_client()
self.assertEqual(client, fake_nova)
mock_nova.Client.assert_called_once_with('2', *self.args[:3],
auth_url=self.args[-1],
service_type='compute')
self.assertEqual(self.clients.cache["nova"], fake_nova)
def test_get_glance_client(self):
with mock.patch('rally.osclients.glance') as mock_glance:
fake_glance = fakes.FakeGlanceClient()
mock_glance.Client = mock.MagicMock(return_value=fake_glance)
kc = fakes.FakeKeystoneClient()
self.clients.get_keystone_client = mock.MagicMock(return_value=kc)
self.assertTrue("glance" not in self.clients.cache)
client = self.clients.get_glance_client()
self.assertEqual(client, fake_glance)
endpoint = kc.service_catalog.get_endpoints()['image'][0]
kw = {'endpoint': endpoint['publicURL'], 'token': kc.auth_token}
mock_glance.Client.assert_called_once_with('1', **kw)
self.assertEqual(self.clients.cache["glance"], fake_glance)
def test_get_cinder_client(self):
with mock.patch('rally.osclients.cinder') as mock_cinder:
fake_cinder = fakes.FakeCinderClient()
mock_cinder.Client = mock.MagicMock(return_value=fake_cinder)
self.assertTrue("cinder" not in self.clients.cache)
client = self.clients.get_cinder_client()
self.assertEqual(client, fake_cinder)
mock_cinder.Client.assert_called_once_with('1', *self.args[:3],
auth_url=self.args[-1],
service_type='volume')
self.assertEqual(self.clients.cache["cinder"], fake_cinder)