def test_availability_zone_juju(self, mock_get): def environ_get_side_effect(key): return { 'JUJU_AVAILABILITY_ZONE': 'az1', }[key] mock_get.side_effect = environ_get_side_effect self.config.side_effect = self.test_config.get self.test_config.set('customize-failure-domain', True) az_context = context.NovaComputeAvailabilityZoneContext() self.assertEqual( {'default_availability_zone': 'az1'}, az_context())
def test_availability_zone_no_juju_with_env(self, mock_get, mock_config): def environ_get_side_effect(key): return { 'JUJU_AVAILABILITY_ZONE': 'az1', }[key] mock_get.side_effect = environ_get_side_effect def config_side_effect(key): return { 'customize-failure-domain': False, 'default-availability-zone': 'nova', }[key] mock_config.side_effect = config_side_effect az_context = context.NovaComputeAvailabilityZoneContext() self.assertEqual({'default_availability_zone': 'nova'}, az_context())