def test_get_choices_with_kwargs(self):
"""Testing ConditionChoices.get_choices with kwargs"""
class MyChoice1(BaseConditionChoice):
choice_id = 'my-choice-1'
class MyChoice2(BaseConditionChoice):
choice_id = 'my-choice-2'
choices = ConditionChoices([MyChoice1, MyChoice2])
choices = list(choices.get_choices(choice_kwargs={'abc': 123}))
self.assertEqual(len(choices), 2)
self.assertEqual(choices[0].extra_state, {'abc': 123})
self.assertEqual(choices[1].extra_state, {'abc': 123})
def test_get_choices(self):
"""Testing ConditionChoices.get_choices"""
class MyChoice1(BaseConditionChoice):
choice_id = 'my-choice-1'
class MyChoice2(BaseConditionChoice):
choice_id = 'my-choice-2'
choices = ConditionChoices([MyChoice1, MyChoice2])
choices = list(choices.get_choices())
self.assertEqual(len(choices), 2)
self.assertEqual(choices[0].__class__, MyChoice1)
self.assertEqual(choices[1].__class__, MyChoice2)