def test_dehydrate_no_choice_config(self): no_choice_name = random.choice( list(name for name in CONFIG_ITEMS.keys() if not hasattr(get_config_field(name), "choices"))) Config.objects.set_config(no_choice_name, "myvalue") user = factory.make_User() handler = ConfigHandler(user, {}, None) dehydrated = handler.dehydrate_configs([no_choice_name])[0] self.assertEqual(no_choice_name, dehydrated["name"]) self.assertEqual("myvalue", dehydrated["value"]) self.assertNotIn("choices", dehydrated)
def test_dehydrate_choice_config(self): choice_name, choices = random.choice( list((name, get_config_field(name).choices) for name in CONFIG_ITEMS.keys() if hasattr(get_config_field(name), "choices"))) choice_value = random.choice([value for value, _ in choices]) Config.objects.set_config(choice_name, choice_value) user = factory.make_User() handler = ConfigHandler(user, {}, None) dehydrated = handler.dehydrate_configs([choice_name])[0] self.assertEqual(choice_name, dehydrated["name"]) self.assertEqual(choice_value, dehydrated["value"]) self.assertEqual(choices, dehydrated["choices"])