def test_update_except_exception(self, mock_nova, mock_exc):
mock_nova.novaclient().availability_zones.list.side_effect = Exception
request = {'request': mock.Mock()}
a_zone_choice_field = fields.AZoneChoiceField()
a_zone_choice_field.choices = []
expected_choices = [("", _("No availability zones available"))]
a_zone_choice_field.update(request)
self.assertEqual(expected_choices, a_zone_choice_field.choices)
mock_exc.handle.assert_called_once_with(request['request'], mock.ANY)
def test_update(self, mock_nova):
mock_nova.novaclient().availability_zones.list.return_value = [
mock.Mock(zoneName='foo_zone', zoneState='foo_state'),
mock.Mock(zoneName='bar_zone', zoneState='bar_state')
]
request = {'request': mock.Mock()}
a_zone_choice_field = fields.AZoneChoiceField()
a_zone_choice_field.choices = []
expected_choices = [("bar_zone", "bar_zone"), ("foo_zone", "foo_zone")]
a_zone_choice_field.update(request)
self.assertEqual(expected_choices, a_zone_choice_field.choices)
