Dougal Matthews 531be10588 Implement create Plan for the OpenStack client
Change-Id: If6e0ad80868f6c1c8af17b1445f7e231a21a754f
2015-05-07 11:00:08 +01:00

45 lines
1.4 KiB
Python

# 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 openstackclient.tests import utils
def _create_mock(**kwargs):
mock_plan = mock.Mock()
mock_plan.configure_mock(**kwargs)
mock_plan.to_dict.return_value = kwargs
return mock_plan
mock_roles = [
_create_mock(uuid="UUID1", name="Role 1 Name", version=1,
description="Mock Role 1"),
_create_mock(uuid="UUID2", name="Role 2 Name", version=2,
description="Mock Role 2"),
]
mock_plans = [
_create_mock(uuid="UUID1", name="Plan 1 Name", description="Plan 1",
roles=mock_roles),
_create_mock(uuid="UUID2", name="Plan 2 Name", description="Plan 2",
roles=[])
]
class TestManagement(utils.TestCommand):
def setUp(self):
super(TestManagement, self).setUp()
self.app.client_manager.management = mock.Mock()