openstack/rally
Code Issues Proposed changes

Merge "Add unit tests for the show command"

Browse Source
This commit is contained in:
Jenkins 2014-03-04 20:41:44 +00:00 committed by Gerrit Code Review
commit 7c1507940f
1 changed files with 118 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
tests/cmd/commands/test_show.py Normal file
View File

@ -0,0 +1,118 @@
# Copyright 2014: The Rally team
# 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
import uuid
from rally.cmd.commands import show
from tests import fakes
from tests import test
class ShowCommandsTestCase(test.TestCase):
def setUp(self):
super(ShowCommandsTestCase, self).setUp()
self.show = show.ShowCommands()
self.fake_endpoint = {'username': 'fake_username',
'password': 'fake_password',
'tenant_name': 'fake_tenant_name',
'auth_url': 'http://fake.auth.url'}
self.fake_deploy_id = str(uuid.uuid4)
self.fake_glance_client = fakes.FakeClients().get_glance_client()
self.fake_nova_client = fakes.FakeClients().get_nova_client()
@mock.patch('rally.cmd.commands.show.osclients.Clients.get_glance_client')
@mock.patch('rally.cmd.commands.show.prettytable.PrettyTable.add_row')
@mock.patch('rally.cmd.commands.show.db.deployment_get')
def test_images(self, mock_deployment_get, mock_add_row, mock_get_glance):
self.fake_glance_client.images.create('image', None, None, None)
fake_image = self.fake_glance_client.images.cache.values()[0]
fake_image.size = 1
mock_get_glance.return_value = self.fake_glance_client
mock_deployment_get.return_value = {'endpoints': [self.fake_endpoint]}
self.show.images(self.fake_deploy_id)
mock_deployment_get.assert_called_once_with(self.fake_deploy_id)
mock_get_glance.assert_called_once_with()
mock_add_row.assert_called_once_with([fake_image.id,
fake_image.name,
fake_image.size])
@mock.patch('rally.cmd.commands.show.osclients.Clients.get_nova_client')
@mock.patch('rally.cmd.commands.show.prettytable.PrettyTable.add_row')
@mock.patch('rally.cmd.commands.show.db.deployment_get')
def test_flavors(self, mock_deployment_get, mock_add_row, mock_get_nova):
self.fake_nova_client.flavors.create()
fake_flavor = self.fake_nova_client.flavors.cache.values()[0]
fake_flavor.id, fake_flavor.name, fake_flavor.vcpus = 1, 'm1.fake', 1
fake_flavor.ram, fake_flavor.swap, fake_flavor.disk = 1024, 128, 10
mock_get_nova.return_value = self.fake_nova_client
mock_deployment_get.return_value = {'endpoints': [self.fake_endpoint]}
self.show.flavors(self.fake_deploy_id)
mock_deployment_get.assert_called_once_with(self.fake_deploy_id)
mock_get_nova.assert_called_once_with()
mock_add_row.assert_called_once_with([fake_flavor.id,
fake_flavor.name,
fake_flavor.vcpus,
fake_flavor.ram,
fake_flavor.swap,
fake_flavor.disk])
@mock.patch('rally.cmd.commands.show.osclients.Clients.get_nova_client')
@mock.patch('rally.cmd.commands.show.prettytable.PrettyTable.add_row')
@mock.patch('rally.cmd.commands.show.db.deployment_get')
def test_networks(self, mock_deployment_get, mock_add_row, mock_get_nova):
self.fake_nova_client.networks.create(1234)
fake_network = self.fake_nova_client.networks.cache.values()[0]
fake_network.label = 'fakenet'
fake_network.cidr = '10.0.0.0/24'
mock_get_nova.return_value = self.fake_nova_client
mock_deployment_get.return_value = {'endpoints': [self.fake_endpoint]}
self.show.networks(self.fake_deploy_id)
mock_deployment_get.assert_called_once_with(self.fake_deploy_id)
mock_get_nova.assert_called_once_with()
mock_add_row.assert_called_once_with([fake_network.id,
fake_network.label,
fake_network.cidr])
@mock.patch('rally.cmd.commands.show.osclients.Clients.get_nova_client')
@mock.patch('rally.cmd.commands.show.prettytable.PrettyTable.add_row')
@mock.patch('rally.cmd.commands.show.db.deployment_get')
def test_secgroups(self, mock_deployment_get, mock_add_row, mock_get_nova):
fake_secgroup = self.fake_nova_client.security_groups.cache.values()[0]
fake_secgroup.id = 0
mock_get_nova.return_value = self.fake_nova_client
mock_deployment_get.return_value = {'endpoints': [self.fake_endpoint]}
self.show.secgroups(self.fake_deploy_id)
mock_deployment_get.assert_called_once_with(self.fake_deploy_id)
mock_get_nova.assert_called_once_with()
mock_add_row.assert_called_once_with([fake_secgroup.id,
fake_secgroup.name,
fake_secgroup.description])
@mock.patch('rally.cmd.commands.show.osclients.Clients.get_nova_client')
@mock.patch('rally.cmd.commands.show.prettytable.PrettyTable.add_row')
@mock.patch('rally.cmd.commands.show.db.deployment_get')
def test_keypairs(self, mock_deployment_get, mock_add_row, mock_get_nova):
self.fake_nova_client.keypairs.create('keypair')
fake_keypair = self.fake_nova_client.keypairs.cache.values()[0]
fake_keypair.fingerprint = '84:87:58'
mock_get_nova.return_value = self.fake_nova_client
mock_deployment_get.return_value = {'endpoints': [self.fake_endpoint]}
self.show.keypairs(self.fake_deploy_id)
mock_deployment_get.assert_called_once_with(self.fake_deploy_id)
mock_get_nova.assert_called_once_with()
mock_add_row.assert_called_once_with([fake_keypair.name,
fake_keypair.fingerprint])