diff options
Diffstat (limited to 'tests/test_shell.py')
| -rw-r--r-- | tests/test_shell.py | 387 |
1 files changed, 27 insertions, 360 deletions
diff --git a/tests/test_shell.py b/tests/test_shell.py index 0d8b1075..ce7d7663 100644 --- a/tests/test_shell.py +++ b/tests/test_shell.py @@ -1,372 +1,39 @@ + import os import mock import httplib2 -from nose.tools import assert_raises, assert_equal -from novaclient.shell import OpenStackShell, CommandError -from fakeserver import FakeServer -from utils import assert_in - - -# Patch os.environ to avoid required auth info. -def setup(): - global _old_env - fake_env = { - 'NOVA_USERNAME': 'username', - 'NOVA_API_KEY': 'password', - 'NOVA_PROJECT_ID': 'project_id' - } - _old_env, os.environ = os.environ, fake_env.copy() - - # Make a fake shell object, a helping wrapper to call it, and a quick way - # of asserting that certain API calls were made. - global shell, _shell, assert_called, assert_called_anytime - _shell = OpenStackShell() - _shell._api_class = FakeServer - assert_called = lambda m, u, b=None: _shell.cs.assert_called(m, u, b) - assert_called_anytime = lambda m, u, b=None: \ - _shell.cs.assert_called_anytime(m, u, b) - shell = lambda cmd: _shell.main(cmd.split()) - - -def teardown(): - global _old_env - os.environ = _old_env - - -def test_backup_schedule(): - shell('backup-schedule 1234') - assert_called('GET', '/servers/1234/backup_schedule') - - shell('backup-schedule sample-server --weekly monday') - assert_called( - 'POST', '/servers/1234/backup_schedule', - {'backupSchedule': {'enabled': True, 'daily': 'DISABLED', - 'weekly': 'MONDAY'}} - ) - - shell('backup-schedule sample-server ' - '--weekly disabled --daily h_0000_0200') - assert_called( - 'POST', '/servers/1234/backup_schedule', - {'backupSchedule': {'enabled': True, 'daily': 'H_0000_0200', - 'weekly': 'DISABLED'}} - ) - - shell('backup-schedule sample-server --disable') - assert_called( - 'POST', '/servers/1234/backup_schedule', - {'backupSchedule': {'enabled': False, 'daily': 'DISABLED', - 'weekly': 'DISABLED'}} - ) - -def test_backup_schedule_delete(): - shell('backup-schedule-delete 1234') - assert_called('DELETE', '/servers/1234/backup_schedule') +from novaclient.shell import OpenStackComputeShell +from novaclient import exceptions +from tests import utils -def test_boot(): - shell('boot --image 1 some-server') - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'min_count': 1, 'max_count': 1}} - ) +class ShellTest(utils.TestCase): - shell('boot --image 1 --meta foo=bar --meta spam=eggs some-server ') - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'min_count': 1, 'max_count': 1, - 'metadata': {'foo': 'bar', 'spam': 'eggs'}}} - ) - - -def test_boot_files(): - testfile = os.path.join(os.path.dirname(__file__), 'testfile.txt') - expected_file_data = open(testfile).read().encode('base64') - - shell('boot some-server --image 1 --file /tmp/foo=%s --file /tmp/bar=%s' % - (testfile, testfile)) - - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'min_count': 1, 'max_count': 1, - 'personality': [ - {'path': '/tmp/bar', 'contents': expected_file_data}, - {'path': '/tmp/foo', 'contents': expected_file_data} - ]} + # Patch os.environ to avoid required auth info. + def setUp(self): + global _old_env + fake_env = { + 'NOVA_USERNAME': 'username', + 'NOVA_API_KEY': 'password', + 'NOVA_PROJECT_ID': 'project_id', } - ) - - -def test_boot_invalid_file(): - invalid_file = os.path.join(os.path.dirname(__file__), 'asdfasdfasdfasdf') - assert_raises(CommandError, shell, 'boot some-server --image 1 ' - '--file /foo=%s' % invalid_file) - - -def test_boot_key_auto(): - mock_exists = mock.Mock(return_value=True) - mock_open = mock.Mock() - mock_open.return_value = mock.Mock() - mock_open.return_value.read = mock.Mock(return_value='SSHKEY') - - @mock.patch('os.path.exists', mock_exists) - @mock.patch('__builtin__.open', mock_open) - def test_shell_call(): - shell('boot some-server --image 1 --key') - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'min_count': 1, 'max_count': 1, - 'personality': [{ - 'path': '/root/.ssh/authorized_keys2', - 'contents': ('SSHKEY').encode('base64')}, - ]} - } - ) - - test_shell_call() - - -def test_boot_key_auto_no_keys(): - mock_exists = mock.Mock(return_value=False) - - @mock.patch('os.path.exists', mock_exists) - def test_shell_call(): - assert_raises(CommandError, shell, 'boot some-server --image 1 --key') - - test_shell_call() - - -def test_boot_key_file(): - testfile = os.path.join(os.path.dirname(__file__), 'testfile.txt') - expected_file_data = open(testfile).read().encode('base64') - shell('boot some-server --image 1 --key %s' % testfile) - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'min_count': 1, 'max_count': 1, - 'personality': [ - {'path': '/root/.ssh/authorized_keys2', 'contents': - expected_file_data}, - ]} - } - ) - - -def test_boot_invalid_keyfile(): - invalid_file = os.path.join(os.path.dirname(__file__), 'asdfasdfasdfasdf') - assert_raises(CommandError, shell, 'boot some-server ' - '--image 1 --key %s' % invalid_file) - - -def test_boot_ipgroup(): - shell('boot --image 1 --ipgroup 1 some-server') - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'sharedIpGroupId': 1, 'min_count': 1, 'max_count': 1}} - ) - - -def test_boot_ipgroup_name(): - shell('boot --image 1 --ipgroup group1 some-server') - assert_called( - 'POST', '/servers', - {'server': {'flavorId': 1, 'name': 'some-server', 'imageId': '1', - 'sharedIpGroupId': 1, 'min_count': 1, 'max_count': 1}} - ) - - -def test_flavor_list(): - shell('flavor-list') - assert_called_anytime('GET', '/flavors/detail') - - -def test_image_list(): - shell('image-list') - assert_called('GET', '/images/detail') - - -def test_snapshot_create(): - shell('image-create sample-server mysnapshot') - assert_called( - 'POST', '/images', - {'image': {'name': 'mysnapshot', 'serverId': 1234}} - ) - - -def test_image_delete(): - shell('image-delete 1') - assert_called('DELETE', '/images/1') - - -def test_ip_share(): - shell('ip-share sample-server 1 1.2.3.4') - assert_called( - 'PUT', '/servers/1234/ips/public/1.2.3.4', - {'shareIp': {'sharedIpGroupId': 1, 'configureServer': True}} - ) - - -def test_ip_unshare(): - shell('ip-unshare sample-server 1.2.3.4') - assert_called('DELETE', '/servers/1234/ips/public/1.2.3.4') - - -def test_ipgroup_list(): - shell('ipgroup-list') - assert_in(('GET', '/shared_ip_groups/detail', None), - _shell.cs.client.callstack) - assert_called('GET', '/servers/5678') - - -def test_ipgroup_show(): - shell('ipgroup-show 1') - assert_called('GET', '/shared_ip_groups/1') - shell('ipgroup-show group2') - # does a search, not a direct GET - assert_called('GET', '/shared_ip_groups/detail') - - -def test_ipgroup_create(): - shell('ipgroup-create a-group') - assert_called( - 'POST', '/shared_ip_groups', - {'sharedIpGroup': {'name': 'a-group'}} - ) - shell('ipgroup-create a-group sample-server') - assert_called( - 'POST', '/shared_ip_groups', - {'sharedIpGroup': {'name': 'a-group', 'server': 1234}} - ) - - -def test_ipgroup_delete(): - shell('ipgroup-delete group1') - assert_called('DELETE', '/shared_ip_groups/1') - - -def test_list(): - shell('list') - assert_called('GET', '/servers/detail') - - -def test_reboot(): - shell('reboot sample-server') - assert_called('POST', '/servers/1234/action', {'reboot': {'type': 'SOFT'}}) - shell('reboot sample-server --hard') - assert_called('POST', '/servers/1234/action', {'reboot': {'type': 'HARD'}}) - - -def test_rebuild(): - shell('rebuild sample-server 1') - assert_called('POST', '/servers/1234/action', {'rebuild': {'imageId': 1}}) - - -def test_rename(): - shell('rename sample-server newname') - assert_called('PUT', '/servers/1234', {'server': {'name': 'newname'}}) - - -def test_resize(): - shell('resize sample-server 1') - assert_called('POST', '/servers/1234/action', {'resize': {'flavorId': 1}}) - - -def test_resize_confirm(): - shell('resize-confirm sample-server') - assert_called('POST', '/servers/1234/action', {'confirmResize': None}) - - -def test_resize_revert(): - shell('resize-revert sample-server') - assert_called('POST', '/servers/1234/action', {'revertResize': None}) - - -def test_backup(): - shell('backup sample-server mybackup daily 1') - assert_called( - 'POST', '/servers/1234/action', - {'createBackup': {'name': 'mybackup', 'backup_type': 'daily', - 'rotation': 1}} - ) - - -@mock.patch('getpass.getpass', mock.Mock(return_value='p')) -def test_root_password(): - shell('root-password sample-server') - assert_called('PUT', '/servers/1234', {'server': {'adminPass': 'p'}}) - - -def test_show(): - shell('show 1234') - # XXX need a way to test multiple calls - # assert_called('GET', '/servers/1234') - assert_called('GET', '/images/2') - - -def test_delete(): - shell('delete 1234') - assert_called('DELETE', '/servers/1234') - shell('delete sample-server') - assert_called('DELETE', '/servers/1234') - - -def test_zone(): - shell('zone 1') - assert_called('GET', '/zones/1') - - shell('zone 1 --api_url=http://zzz --zone_username=frank --password=xxx') - assert_called( - 'PUT', '/zones/1', - {'zone': {'api_url': 'http://zzz', 'username': 'frank', - 'password': 'xxx'}} - ) - - -def test_zone_add(): - shell('zone-add http://zzz frank xxx 0.0 1.0') - assert_called( - 'POST', '/zones', - {'zone': {'api_url': 'http://zzz', 'username': 'frank', - 'password': 'xxx', - 'weight_offset': '0.0', 'weight_scale': '1.0'}} - ) - - -def test_zone_delete(): - shell('zone-delete 1') - assert_called('DELETE', '/zones/1') - - -def test_zone_list(): - shell('zone-list') - assert_in(('GET', '/zones/detail', None), - _shell.cs.client.callstack) - - -def test_help(): - @mock.patch.object(_shell.parser, 'print_help') - def test_help(m): - shell('help') - m.assert_called() - - @mock.patch.object(_shell.subcommands['delete'], 'print_help') - def test_help_delete(m): - shell('help delete') - m.assert_called() + _old_env, os.environ = os.environ, fake_env.copy() - test_help() - test_help_delete() + # Make a fake shell object, a helping wrapper to call it, and a quick way + # of asserting that certain API calls were made. + global shell, _shell, assert_called, assert_called_anytime + _shell = OpenStackComputeShell() + shell = lambda cmd: _shell.main(cmd.split()) - assert_raises(CommandError, shell, 'help foofoo') + def tearDown(self): + global _old_env + os.environ = _old_env + def test_help_unknown_command(self): + self.assertRaises(exceptions.CommandError, shell, 'help foofoo') -def test_debug(): - httplib2.debuglevel = 0 - shell('--debug list') - assert httplib2.debuglevel == 1 + def test_debug(self): + httplib2.debuglevel = 0 + shell('--debug help') + assert httplib2.debuglevel == 1 |