Merge "Added capabilities/options to 'openstack flavor list' command to match capabilities already provided by 'nova flavor-list':"

1 files changed, 274 insertions, 0 deletions

diff --git a/openstackclient/tests/compute/v2/test_flavor.py b/openstackclient/tests/compute/v2/test_flavor.py
new file mode 100644
index 00000000..8f33ccfe
--- /dev/null
+++ b/openstackclient/tests/compute/v2/test_flavor.py
@@ -0,0 +1,274 @@
+#   Copyright 2015 Symantec Corporation
+#
+#   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 copy
+
+from openstackclient.compute.v2 import flavor
+from openstackclient.tests.compute.v2 import fakes as compute_fakes
+from openstackclient.tests import fakes
+
+
+class FakeFlavorResource(fakes.FakeResource):
+
+    def get_keys(self):
+        return {'property': 'value'}
+
+
+class TestFlavor(compute_fakes.TestComputev2):
+
+    def setUp(self):
+        super(TestFlavor, self).setUp()
+
+        # Get a shortcut to the FlavorManager Mock
+        self.flavors_mock = self.app.client_manager.compute.flavors
+        self.flavors_mock.reset_mock()
+
+
+class TestFlavorList(TestFlavor):
+
+    def setUp(self):
+        super(TestFlavorList, self).setUp()
+
+        self.flavors_mock.list.return_value = [
+            FakeFlavorResource(
+                None,
+                copy.deepcopy(compute_fakes.FLAVOR),
+                loaded=True,
+            ),
+        ]
+
+        # Get the command object to test
+        self.cmd = flavor.ListFlavor(self.app, None)
+
+    def test_flavor_list_no_options(self):
+        arglist = []
+        verifylist = [
+            ('public', True),
+            ('all', False),
+            ('long', False),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'is_public': True
+        }
+
+        self.flavors_mock.list.assert_called_with(
+            **kwargs
+        )
+
+        collist = (
+            'ID',
+            'Name',
+            'RAM',
+            'Disk',
+            'Ephemeral',
+            'VCPUs',
+            'Is Public',
+        )
+        self.assertEqual(collist, columns)
+        datalist = ((
+            compute_fakes.flavor_id,
+            compute_fakes.flavor_name,
+            compute_fakes.flavor_ram,
+            '',
+            '',
+            compute_fakes.flavor_vcpus,
+            ''
+        ), )
+        self.assertEqual(datalist, tuple(data))
+
+    def test_flavor_list_all_flavors(self):
+        arglist = [
+            '--all',
+        ]
+        verifylist = [
+            ('all', True),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'is_public': None
+        }
+
+        self.flavors_mock.list.assert_called_with(
+            **kwargs
+        )
+
+        collist = (
+            'ID',
+            'Name',
+            'RAM',
+            'Disk',
+            'Ephemeral',
+            'VCPUs',
+            'Is Public',
+        )
+        self.assertEqual(collist, columns)
+        datalist = ((
+            compute_fakes.flavor_id,
+            compute_fakes.flavor_name,
+            compute_fakes.flavor_ram,
+            '',
+            '',
+            compute_fakes.flavor_vcpus,
+            ''
+        ), )
+        self.assertEqual(datalist, tuple(data))
+
+    def test_flavor_list_private_flavors(self):
+        arglist = [
+            '--private',
+            ]
+        verifylist = [
+            ('public', False),
+            ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'is_public': False
+        }
+
+        self.flavors_mock.list.assert_called_with(
+            **kwargs
+        )
+
+        collist = (
+            'ID',
+            'Name',
+            'RAM',
+            'Disk',
+            'Ephemeral',
+            'VCPUs',
+            'Is Public',
+        )
+        self.assertEqual(collist, columns)
+        datalist = ((
+            compute_fakes.flavor_id,
+            compute_fakes.flavor_name,
+            compute_fakes.flavor_ram,
+            '',
+            '',
+            compute_fakes.flavor_vcpus,
+            ''
+        ), )
+        self.assertEqual(datalist, tuple(data))
+
+    def test_flavor_list_public_flavors(self):
+        arglist = [
+            '--public',
+            ]
+        verifylist = [
+            ('public', True),
+            ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'is_public': True
+        }
+
+        self.flavors_mock.list.assert_called_with(
+            **kwargs
+        )
+
+        collist = (
+            'ID',
+            'Name',
+            'RAM',
+            'Disk',
+            'Ephemeral',
+            'VCPUs',
+            'Is Public',
+        )
+        self.assertEqual(collist, columns)
+        datalist = ((
+            compute_fakes.flavor_id,
+            compute_fakes.flavor_name,
+            compute_fakes.flavor_ram,
+            '',
+            '',
+            compute_fakes.flavor_vcpus,
+            ''
+        ), )
+        self.assertEqual(datalist, tuple(data))
+
+    def test_flavor_list_long(self):
+        arglist = [
+            '--long',
+        ]
+        verifylist = [
+            ('long', True),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'is_public': True
+        }
+
+        self.flavors_mock.list.assert_called_with(
+            **kwargs
+        )
+
+        collist = (
+            'ID',
+            'Name',
+            'RAM',
+            'Disk',
+            'Ephemeral',
+            'VCPUs',
+            'Is Public',
+            'Swap',
+            'RXTX Factor',
+            'Properties'
+        )
+        self.assertEqual(collist, columns)
+        datalist = ((
+            compute_fakes.flavor_id,
+            compute_fakes.flavor_name,
+            compute_fakes.flavor_ram,
+            '',
+            '',
+            compute_fakes.flavor_vcpus,
+            '',
+            '',
+            '',
+            'property=\'value\''
+        ), )
+        self.assertEqual(datalist, tuple(data))