Support to specify the valid external network name
[functest.git] / functest / opnfv_tests / openstack / snaps / snaps_utils.py
1 # Copyright (c) 2015 All rights reserved
2 # This program and the accompanying materials
3 # are made available under the terms of the Apache License, Version 2.0
4 # which accompanies this distribution, and is available at
5 #
6 # http://www.apache.org/licenses/LICENSE-2.0
7
8 from functest.utils.constants import CONST
9
10 from snaps.openstack.utils import neutron_utils, nova_utils
11
12
13 def get_ext_net_name(os_creds):
14     """
15     Returns the configured external network name or
16     the first retrieved external network name
17     :param: os_creds: an instance of snaps OSCreds object
18     :return:
19     """
20     neutron = neutron_utils.neutron_client(os_creds)
21     ext_nets = neutron_utils.get_external_networks(neutron)
22     if (hasattr(CONST, 'EXTERNAL_NETWORK')):
23         extnet_config = CONST.__getattribute__('EXTERNAL_NETWORK')
24         for ext_net in ext_nets:
25             if ext_net.name == extnet_config:
26                 return extnet_config
27     return ext_nets[0].name if ext_nets else ""
28
29
30 def get_active_compute_cnt(os_creds):
31     """
32     Returns the number of active compute servers
33     :param: os_creds: an instance of snaps OSCreds object
34     :return: the number of active compute servers
35     """
36     nova = nova_utils.nova_client(os_creds)
37     computes = nova_utils.get_availability_zone_hosts(nova, zone_name='nova')
38     return len(computes)