def test_from_dict_invalid_option(self):
"""Check if an exception is raised when an invalid option is given"""
opts = {'delay': 1, 'max_retries': 3, 'custom_param': 'myparam'}
with self.assertRaisesRegex(
ValueError, "unknown 'custom_param' task config parameter"):
_ = SchedulingTaskConfig.from_dict(opts)
def test_from_dict(self):
"""Check if an object is created when its properties are given from a dict"""
opts = {'delay': 1, 'max_retries': 3}
scheduling_cfg = SchedulingTaskConfig.from_dict(opts)
self.assertIsInstance(scheduling_cfg, SchedulingTaskConfig)
self.assertEqual(scheduling_cfg.delay, 1)
self.assertEqual(scheduling_cfg.max_retries, 3)
def test_from_dict_queue_none(self):
"""Check if an object is created from a dict when queue is None"""
# Test None options
opts = {'queue': None}
scheduling_cfg = SchedulingTaskConfig.from_dict(opts)
self.assertIsInstance(scheduling_cfg, SchedulingTaskConfig)
self.assertEqual(scheduling_cfg.queue, None)