def configure_resources(*args):
"""Create/discover resources for management of load balancer instances."""
if not reactive.is_flag_set('leadership.is_leader'):
return ch_core.hookenv.action_fail('action must be run on the leader '
'unit.')
if not reactive.all_flags_set(
'identity-service.available', 'neutron-api.available',
'sdn-subordinate.available', 'amqp.available'):
return ch_core.hookenv.action_fail('all required relations not '
'available, please defer action'
'until deployment is complete.')
identity_service = reactive.endpoint_from_flag(
'identity-service.available')
try:
(network, secgrp) = api_crud.get_mgmt_network(
identity_service,
create=reactive.is_flag_set('config.default.create-mgmt-network'),
)
except api_crud.APIUnavailable as e:
ch_core.hookenv.action_fail(
'Neutron API not available yet, deferring '
'network creation/discovery. ("{}")'.format(e))
return
if network and secgrp:
leadership.leader_set({
'amp-boot-network-list': network['id'],
'amp-secgroup-list': secgrp['id']
})
if reactive.is_flag_set('config.default.custom-amp-flavor-id'):
# NOTE(fnordahl): custom flavor provided through configuration is
# handled in the charm class configuration property.
try:
flavor = api_crud.get_nova_flavor(identity_service)
except api_crud.APIUnavailable as e:
ch_core.hookenv.action_fail('Nova API not available yet, '
'deferring flavor '
'creation. ("{}")'.format(e))
return
else:
leadership.leader_set({'amp-flavor-id': flavor.id})
amp_key_name = ch_core.hookenv.config('amp-ssh-key-name')
if amp_key_name:
identity_service = reactive.endpoint_from_flag(
'identity-service.available')
api_crud.create_nova_keypair(identity_service, amp_key_name)
# Set qutotas to unlimited
try:
api_crud.set_service_quotas_unlimited(identity_service)
except api_crud.APIUnavailable as e:
ch_core.hookenv.action_fail(
'Unbable to set quotas to unlimited: {}'.format(e))
# execute port setup for leader, the followers will execute theirs on
# `leader-settings-changed` hook
with charm.provide_charm_instance() as octavia_charm:
api_crud.setup_hm_port(identity_service, octavia_charm)
octavia_charm.render_all_configs()
octavia_charm._assess_status()
def test_get_nova_flavor(self):
self.patch_object(api_crud, 'get_nova_client')
self.patch_object(api_crud, 'nova_client')
self.patch_object(api_crud, 'session_from_identity_service')
self.patch_object(api_crud, 'keystone_exceptions')
nova = mock.MagicMock()
self.get_nova_client.return_value = nova
flavor = mock.MagicMock()
flavor.id = 'fake-id'
flavor.name = 'charm-octavia'
nova.flavors.list.return_value = [flavor]
self.keystone_exceptions.catalog.EndpointNotFound = Exception
self.keystone_exceptions.connection.ConnectFailure = Exception
self.nova_client.exceptions.ClientException = Exception
nova.flavors.list.side_effect = Exception
identity_service = mock.MagicMock()
with self.assertRaises(api_crud.APIUnavailable):
api_crud.get_nova_flavor(identity_service)
nova.flavors.list.side_effect = None
api_crud.get_nova_flavor(identity_service)
nova.flavors.list.assert_called_with(is_public=False)
self.assertFalse(nova.flavors.create.called)
nova.flavors.list.return_value = []
nova.flavors.create.return_value = flavor
api_crud.get_nova_flavor(identity_service)
nova.flavors.create.assert_called_with('charm-octavia',
1024,
1,
8,
is_public=False)