def fetch_stream_messages(**kwargs):
log.info('Fetch Message Stream kwargs: {}'.format(kwargs))
return dict(
message=kwargs['message'],
channel=get_channel_by_id(channel_id=kwargs['channel']),
published_by_name=get_user_by_id(user_id=kwargs['published_by']).first_name,
created_on=datetime.now(),
)
def test_get_channel_by_id_for_value_error(self):
with self.assertRaises(ValueError) as err_wrong_param:
channel.get_channel_by_id(channel_id='wrong-id')
self.channel.deleted_on = '2016-04-04'
session.flush()
with self.assertRaises(ValueError) as err_deleted_param:
channel.get_channel_by_id(channel_id=self.channel_id)
with self.assertRaises(ValueError) as err_empty_string_param:
channel.get_channel_by_id(channel_id='')
self.assertEqual('INVALID-CHANNEL-ID', err_wrong_param.exception.message)
self.assertEqual('INVALID-CHANNEL-ID', err_deleted_param.exception.message)
self.assertEqual('REQ-CHANNEL-ID', err_empty_string_param.exception.message)
def test_get_user_by_id_user_for_key_error(self):
with self.assertRaises(KeyError) as err_empty_param:
channel.get_channel_by_id(channel_wrong_id=self.channel_id)
self.assertEqual('REQ-CHANNEL-ID', err_empty_param.exception.message)
def user_channel_response_transformation(user_channel_mapping):
channel_obj = get_channel_by_id(channel_id=user_channel_mapping.channel_id)
return channel_response_transformation(channel_obj)