def test_set_schedule(self, *args):
user_channel_vo = UserChannelFactory(domain_id=self.domain_id,
secret_id='secret-xyz',
is_scheduled=False,
schedule=None)
schedule = {
'day_of_week': ['MON', 'TUE'],
'start_hour': 1,
'end_hour': 23
}
params = {
'user_channel_id': user_channel_vo.user_channel_id,
'is_scheduled': True,
'schedule': schedule,
'domain_id': self.domain_id
}
self.transaction.method = 'set_schedule'
user_ch_svc = UserChannelService(transaction=self.transaction)
schedule_user_ch_vo = user_ch_svc.set_schedule(params.copy())
print_data(schedule_user_ch_vo.to_dict(), 'test_set_schedule')
UserChannelInfo(schedule_user_ch_vo)
self.assertIsInstance(schedule_user_ch_vo, UserChannel)
self.assertEqual(schedule_user_ch_vo.is_scheduled, True)
self.assertEqual(schedule['day_of_week'],
schedule_user_ch_vo.schedule.day_of_week)
self.assertEqual(schedule['start_hour'],
schedule_user_ch_vo.schedule.start_hour)
self.assertEqual(schedule['end_hour'],
schedule_user_ch_vo.schedule.end_hour)
def test_set_subscription(self, *args):
user_channel_vo = UserChannelFactory(domain_id=self.domain_id,
secret_id='secret-xyz',
is_subscribe=False,
subscriptions=[])
subscriptions = ['a', 'b', 'c']
params = {
'user_channel_id': user_channel_vo.user_channel_id,
'is_subscribe': True,
'subscriptions': subscriptions,
'domain_id': self.domain_id
}
self.transaction.method = 'set_schedule'
user_ch_svc = UserChannelService(transaction=self.transaction)
subscription_user_ch_vo = user_ch_svc.set_schedule(params.copy())
print_data(subscription_user_ch_vo.to_dict(), 'test_set_subscription')
UserChannelInfo(subscription_user_ch_vo)
self.assertIsInstance(subscription_user_ch_vo, UserChannel)
self.assertEqual(subscription_user_ch_vo.is_subscribe, True)
self.assertEqual(subscriptions, subscription_user_ch_vo.subscriptions)