def test_get_message(self):
"""Test get_message_objects() can retrieve a single message."""
# Test one message can be retrieved.
message = get_message_objects('TestMessageA')
self.assertEqual(message, self.TestMessageA)
# Ensure exception is raised on non-string input.
with self.assertRaises(TypeError):
get_message_objects(False)
with self.assertRaises(TypeError):
get_message_objects(['TestMessageA', False])
def test_get_message(self):
"""Test get_message_objects() can retrieve a single message."""
# Test one message can be retrieved.
message = get_message_objects('TestMessageA')
self.assertEqual(message, self.TestMessageA)
# Ensure exception is raised on non-string input.
with self.assertRaises(TypeError):
get_message_objects(False)
with self.assertRaises(TypeError):
get_message_objects(['TestMessageA', False])
def test_get_duplicates(self):
"""Test get_message_objects() can detect duplicate messages."""
# Create a new message legitimately.
TestMessageE = message_factory('TestMessageE', ('E',), 4)
# Bypass the name checking mechanism in _RegisterMeta().
TestMessageE.__name__ = 'TestMessageA'
# Ensure an exception is raised if multiple messages with the same name
# exist.
with self.assertRaises(NameError):
get_message_objects('TestMessageA')
def test_get_duplicates(self):
"""Test get_message_objects() can detect duplicate messages."""
# Create a new message legitimately.
TestMessageE = message_factory('TestMessageE', ('E', ), 4)
# Bypass the name checking mechanism in _RegisterMeta().
TestMessageE.__name__ = 'TestMessageA'
# Ensure an exception is raised if multiple messages with the same name
# exist.
with self.assertRaises(NameError):
get_message_objects('TestMessageA')
def test_get_messages(self):
"""Test get_message_objects() can retrieve multiple messages."""
# Message objects to remove.
message_names = ['TestMessageA', 'TestMessageB', 'TestMessageD']
# Test multiple messages can be retrieved.
messages = get_message_objects(message_names)
self.assertEqual(
messages,
[self.TestMessageA, self.TestMessageB, self.TestMessageD])
# Ensure an exception is raised if a message that does not exists is
# requested.
with self.assertRaises(NameError):
get_message_objects('TestMessageDoesNotExist')
# Ensure errors with list inputs are propagated.
message_names.append('TestMessageDoesNotExist')
with self.assertRaises(Exception):
get_message_objects(message_names)
def test_get_messages(self):
"""Test get_message_objects() can retrieve multiple messages."""
# Message objects to remove.
message_names = ['TestMessageA',
'TestMessageB',
'TestMessageD']
# Test multiple messages can be retrieved.
messages = get_message_objects(message_names)
self.assertEqual(messages,
[self.TestMessageA,
self.TestMessageB,
self.TestMessageD])
# Ensure an exception is raised if a message that does not exists is
# requested.
with self.assertRaises(NameError):
get_message_objects('TestMessageDoesNotExist')
# Ensure errors with list inputs are propagated.
message_names.append('TestMessageDoesNotExist')
with self.assertRaises(Exception):
get_message_objects(message_names)
