f82edf8721
The tests were putting clean-up code in tearDown. Using tearDown causes problems with developing and testing because if there's an exception raised in setUp then tearDown is not called and this can affect other tests and make it hard to figure out what the problem was. Also, this should make it easier to switch to using fixtures since fixtures use cleanUp. Change-Id: Ifc22dfd9012402c895a09534f4b1d9bd7a2ec931
142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# 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 sys
|
|
|
|
import mock
|
|
import six
|
|
|
|
from keystoneclient import exceptions
|
|
from keystoneclient.tests import utils as test_utils
|
|
from keystoneclient import utils
|
|
|
|
|
|
class FakeResource(object):
|
|
pass
|
|
|
|
|
|
class FakeManager(object):
|
|
|
|
resource_class = FakeResource
|
|
|
|
resources = {
|
|
'1234': {'name': 'entity_one'},
|
|
'8e8ec658-c7b0-4243-bdf8-6f7f2952c0d0': {'name': 'entity_two'},
|
|
'\xe3\x82\xbdtest': {'name': u'\u30bdtest'},
|
|
'5678': {'name': '9876'}
|
|
}
|
|
|
|
def get(self, resource_id):
|
|
try:
|
|
return self.resources[str(resource_id)]
|
|
except KeyError:
|
|
raise exceptions.NotFound(resource_id)
|
|
|
|
def find(self, name=None):
|
|
if name == '9999':
|
|
# NOTE(morganfainberg): special case that raises NoUniqueMatch.
|
|
raise exceptions.NoUniqueMatch()
|
|
for resource_id, resource in self.resources.items():
|
|
if resource['name'] == str(name):
|
|
return resource
|
|
raise exceptions.NotFound(name)
|
|
|
|
|
|
class FindResourceTestCase(test_utils.TestCase):
|
|
|
|
def setUp(self):
|
|
super(FindResourceTestCase, self).setUp()
|
|
self.manager = FakeManager()
|
|
|
|
def test_find_none(self):
|
|
self.assertRaises(exceptions.CommandError,
|
|
utils.find_resource,
|
|
self.manager,
|
|
'asdf')
|
|
|
|
def test_find_by_integer_id(self):
|
|
output = utils.find_resource(self.manager, 1234)
|
|
self.assertEqual(output, self.manager.resources['1234'])
|
|
|
|
def test_find_by_str_id(self):
|
|
output = utils.find_resource(self.manager, '1234')
|
|
self.assertEqual(output, self.manager.resources['1234'])
|
|
|
|
def test_find_by_uuid(self):
|
|
uuid = '8e8ec658-c7b0-4243-bdf8-6f7f2952c0d0'
|
|
output = utils.find_resource(self.manager, uuid)
|
|
self.assertEqual(output, self.manager.resources[uuid])
|
|
|
|
def test_find_by_unicode(self):
|
|
name = '\xe3\x82\xbdtest'
|
|
output = utils.find_resource(self.manager, name)
|
|
self.assertEqual(output, self.manager.resources[name])
|
|
|
|
def test_find_by_str_name(self):
|
|
output = utils.find_resource(self.manager, 'entity_one')
|
|
self.assertEqual(output, self.manager.resources['1234'])
|
|
|
|
def test_find_by_int_name(self):
|
|
output = utils.find_resource(self.manager, 9876)
|
|
self.assertEqual(output, self.manager.resources['5678'])
|
|
|
|
def test_find_no_unique_match(self):
|
|
self.assertRaises(exceptions.CommandError,
|
|
utils.find_resource,
|
|
self.manager,
|
|
9999)
|
|
|
|
|
|
class FakeObject(object):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
|
|
class PrintTestCase(test_utils.TestCase):
|
|
def setUp(self):
|
|
super(PrintTestCase, self).setUp()
|
|
self.old_stdout = sys.stdout
|
|
self.stdout = six.moves.cStringIO()
|
|
self.addCleanup(setattr, self, 'stdout', None)
|
|
sys.stdout = self.stdout
|
|
self.addCleanup(setattr, sys, 'stdout', self.old_stdout)
|
|
|
|
def test_print_list_unicode(self):
|
|
name = u'\u540d\u5b57'
|
|
objs = [FakeObject(name)]
|
|
# NOTE(Jeffrey4l) If the text's encode is proper, this method will not
|
|
# raise UnicodeEncodeError exceptions
|
|
utils.print_list(objs, ['name'])
|
|
self.assertIn(name, self.stdout.getvalue().decode('utf8'))
|
|
|
|
@mock.patch('keystoneclient.openstack.common.strutils.safe_encode')
|
|
def test_print_list_unicode_without_encode(self, safe_encode_mock):
|
|
safe_encode_mock.side_effect = lambda x, *args, **kwargs: x
|
|
|
|
name = u'\u540d\u5b57'
|
|
objs = [FakeObject(name)]
|
|
self.assertRaises(UnicodeEncodeError, utils.print_list, objs, ['name'])
|
|
|
|
def test_print_dict_unicode(self):
|
|
name = u'\u540d\u5b57'
|
|
utils.print_dict({'name': name})
|
|
self.assertIn(name, self.stdout.getvalue().decode('utf8'))
|
|
|
|
@mock.patch('keystoneclient.openstack.common.strutils.safe_encode')
|
|
def test_print_dict_unicode_without_encode(self, safe_encode_mock):
|
|
safe_encode_mock.side_effect = lambda x, *args, **kwargs: x
|
|
|
|
name = u'\u540d\u5b57'
|
|
self.assertRaises(UnicodeEncodeError, utils.print_dict, {'name': name})
|