def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger, inputs=self.inputs)
self.neutron_handle = self.orch.get_network_handler()
return self.neutron_handle
def __init__(self, inputs=None, logger=None, project_name=None,
username=None, password=None, domain_name=None, ini_file=None):
self.inputs = inputs or ContrailTestInit(ini_file,
stack_tenant=project_name)
self.project_name = project_name or self.inputs.project_name
self.domain_name = domain_name or self.inputs.domain_name
self.username = username or self.inputs.stack_user
self.password = password or self.inputs.stack_password
self.logger = logger or self.inputs.logger
self.nova_h = None
self.quantum_h = None
self.api_server_inspects = custom_dict(self.get_api_inspect_handle,
'api_inspect:'+self.project_name+':'+self.username)
self.dnsagent_inspect = custom_dict(self.get_dns_agent_inspect_handle,
'dns_inspect')
self.agent_inspect = custom_dict(self.get_vrouter_agent_inspect_handle,
'agent_inspect')
self.ops_inspects = custom_dict(self.get_opserver_inspect_handle,
'ops_inspect')
self.cn_inspect = custom_dict(self.get_control_node_inspect_handle,
'cn_inspect')
self.ds_inspect = custom_dict(self.get_discovery_service_inspect_handle,
'ds_inspect')
# ToDo: msenthil/sandipd rest of init needs to be better handled
self.vnc_lib = self.get_vnc_lib_h()
self.auth = self.get_auth_h()
if self.inputs.orchestrator == 'openstack':
self.project_id = self.get_project_id()
if self.inputs.verify_thru_gui():
self.ui_login = UILogin(self, self.inputs, project_name, username, password)
self.browser = self.ui_login.browser
self.browser_openstack = self.ui_login.browser_openstack
self.orch = OpenstackOrchestrator(username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
inputs=self.inputs,
vnclib=self.vnc_lib,
logger=self.logger,
auth_server_ip=self.inputs.auth_ip)
self.nova_h = self.orch.get_compute_handler()
self.quantum_h = self.orch.get_network_handler()
else: # vcenter
self.orch = VcenterOrchestrator(user=self.username,
pwd=self.password,
host=self.inputs.auth_ip,
port=self.inputs.auth_port,
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=self.logger)
def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger)
self.neutron_handle = self.orch.get_network_handler()
def get_orch_h(self):
if self.connections:
return self.connections.orch
else:
if self.orchestrator == 'openstack':
return OpenstackOrchestrator(vnclib=self.vnc_api_h,
logger=self.logger,
auth_h=self.auth_client,
inputs=self.inputs)
elif self.orchestrator == 'vcenter':
vcenter_dc = self.inputs.vcenter_dc if self.inputs else \
os.getenv('VCENTER_DC', None)
return VcenterOrchestrator(user=self.username,
pwd=self.password,
host=self.inputs.vcenter_server,
port=self.inputs.vcenter_port,
dc_name=vcenter_dc,
vnc=self.vnc_api_h,
inputs=self.inputs,
logger=self.logger)
return
def get_orch_h(self):
if not getattr(self, 'orch', None):
if self.inputs.orchestrator == 'openstack':
self.orch = OpenstackOrchestrator(
username=self.username,
password=self.password,
project_name=self.project_name,
inputs=self.inputs,
logger=self.logger)
self.nova_h = self.orch.get_compute_h()
self.quantum_h = self.orch.get_network_h()
else:
self.get_vnc_lib_h()
self.orch = VcenterOrchestrator(user=self.username,
pwd=self.password,
host=self.inputs.auth_ip,
port=self.inputs.auth_port,
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=self.logger)
return self.orch
def __init__(self,
inputs=None,
logger=None,
project_name=None,
username=None,
password=None,
domain_name=None,
input_file=None,
domain_obj=None,
scope='domain'):
self.inputs = inputs or ContrailTestInit(input_file,
stack_tenant=project_name)
self.project_name = project_name or self.inputs.project_name
self.domain_name = domain_name or self.inputs.domain_name
self.orch_domain_name = domain_name or self.inputs.domain_name
if self.orch_domain_name == 'Default':
self.domain_name = 'default-domain'
self.scope = scope
self.username = username or self.inputs.stack_user
self.password = password or self.inputs.stack_password
self.logger = logger or self.inputs.logger
self.nova_h = None
self.quantum_h = None
self.vnc_lib_fixture = None
self.api_server_inspects = custom_dict(
self.get_api_inspect_handle,
'api_inspect:' + self.project_name + ':' + self.username)
self.dnsagent_inspect = custom_dict(self.get_dns_agent_inspect_handle,
'dns_inspect')
self.agent_inspect = custom_dict(self.get_vrouter_agent_inspect_handle,
'agent_inspect')
self.ops_inspects = custom_dict(
self.get_opserver_inspect_handle,
'ops_inspect:' + self.project_name + ':' + self.username)
self.cn_inspect = custom_dict(self.get_control_node_inspect_handle,
'cn_inspect')
self.k8s_client = self.get_k8s_api_client_handle()
# ToDo: msenthil/sandipd rest of init needs to be better handled
self.domain_id = None
if self.inputs.domain_isolation:
#get admin auth to list domains and get domain_id
auth = self.get_auth_h(username=self.inputs.admin_username,
password=self.inputs.admin_password,
project_name=self.inputs.admin_tenant,
domain_name=self.inputs.admin_domain)
self.domain_id = auth.get_domain_id(self.domain_name)
self.auth = self.get_auth_h()
self.vnc_lib = self.get_vnc_lib_h()
self.project_id = self.get_project_id()
if self.inputs.orchestrator == 'openstack':
if self.inputs.verify_thru_gui():
self.ui_login = UILogin(self, self.inputs, project_name,
username, password)
self.browser = self.ui_login.browser
self.browser_openstack = self.ui_login.browser_openstack
self.orch = OpenstackOrchestrator(inputs=self.inputs,
vnclib=self.vnc_lib,
logger=self.logger,
auth_h=self.auth)
self.nova_h = self.orch.get_compute_handler()
self.quantum_h = self.orch.get_network_handler()
self.glance_h = self.orch.get_image_handler()
elif self.inputs.orchestrator == 'vcenter' and self.inputs.slave_orchestrator == 'vro':
self.orch = VroWorkflows(user=self.inputs.vcenter_username,
pwd=self.inputs.vcenter_password,
host=self.inputs.vcenter_server,
port=self.inputs.vcenter_port,
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=self.logger)
elif self.inputs.orchestrator == 'vcenter': # vcenter
self.orch = VcenterOrchestrator(user=self.inputs.vcenter_username,
pwd=self.inputs.vcenter_password,
host=self.inputs.vcenter_server,
port=self.inputs.vcenter_port,
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=self.logger)
elif self.inputs.orchestrator == 'kubernetes':
self.orch = None
if self.inputs.vcenter_gw_setup: # vcenter_gateway
self.slave_orch = VcenterGatewayOrch(
user=self.inputs.vcenter_username,
pwd=self.inputs.vcenter_password,
host=self.inputs.vcenter_server,
port=int(self.inputs.vcenter_port),
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=self.logger)
self._kube_manager_inspect = None
class VncLibFixture(fixtures.Fixture):
''' Wrapper for VncApi
:param domain : default is default-domain
:param project_name : default is admin
:param cfgm_ip : default is 127.0.0.1
:param api_port : default is 8082
:param connections : ContrailConnections object. default is None
:param username : default is admin
:param password : default is contrail123
:param auth_server_ip : default is 127.0.0.1
:param logger : logger object
'''
def __init__(self, *args, **kwargs):
self.username = kwargs.get('username', 'admin')
self.password = kwargs.get('password', 'contrail123')
self.project_name = kwargs.get('project_name', 'admin')
self.domain = kwargs.get('domain', 'default-domain')
self.api_server_port = kwargs.get('api_server_port', '8082')
self.cfgm_ip = kwargs.get('cfgm_ip', '127.0.0.1')
self.auth_server_ip = kwargs.get('auth_server_ip', None)
self.logger = kwargs.get('logger', logging.getLogger(__name__))
self.connections = kwargs.get('connections', None)
self.orchestrator = kwargs.get('orchestrator', 'openstack')
self.vnc_api_h = None
self.auth_client_h = None
self.inputs = kwargs.get('inputs', None)
self.neutron_handle = None
self.auth_url = os.getenv('OS_AUTH_URL')
if self.auth_server_ip:
self.auth_url = 'http://' + self.auth_server_ip + ':5000/v2.0'
# end __init__
def setUp(self):
super(VncLibFixture, self).setUp()
if self.connections:
self.logger = self.connections.logger
self.project_name = self.connections.project_name
self.inputs = self.connections.inputs
self.neutron_handle = self.connections.quantum_h
self.vnc_api_h = self.connections.vnc_lib
self.username = self.connections.username
self.password = self.connections.password
self.cfgm_ip = self.inputs.cfgm_ip
self.auth_server_ip = self.inputs.auth_ip
self.project_id = self.connections.project_id
self.auth_url = 'http://' + self.inputs.auth_ip + ':5000/v2.0'
else:
self.vnc_api_h = VncApi(
username=self.username,
password=self.password,
tenant_name=self.project_name,
api_server_host=self.cfgm_ip,
api_server_port=self.api_server_port,
auth_host=self.auth_server_ip)
if self.orchestrator == 'openstack':
self.auth_client = OpenstackAuth(
self.username,
self.password,
self.project_name,
auth_url=self.auth_url,
logger=self.logger)
self.project_id = self.auth_client.get_project_id()
elif self.orchestrator == 'vcenter':
self.auth_client = VcenterAuth(self.username,
self.password,
self.project_name,
self.inputs
)
self.project_id = self.auth_client.get_project_id()
# end setUp
def cleanUp(self):
super(VncLibFixture, self).cleanUp()
def get_handle(self):
return self.vnc_api_h
# end get_handle
def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger)
self.neutron_handle = self.orch.get_network_handler()
# end get_neutron_handle
def get_forwarding_mode(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties()
if vni_obj_properties:
fw_mode = vni_obj_properties.get_forwarding_mode()
else:
fw_mode = None
return fw_mode
# end get_forwarding_mode
def get_vn_subnet_dhcp_flag(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
subnets = vni_obj.network_ipam_refs[0]['attr']
ipam = subnets.get_ipam_subnets()
enable_dhcp = ipam[0].get_enable_dhcp()
return enable_dhcp
# get_vn_subnet_dhcp_flag
def set_rpf_mode(self, vn_fq_name, mode):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = self.vnc_api_h.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties() or VirtualNetworkType()
vni_obj_properties.set_rpf(mode)
vni_obj.set_virtual_network_properties(vni_obj_properties)
vnc_lib.virtual_network_update(vni_obj)
# end set_rpf_mode
def id_to_fq_name(self, id):
return self.vnc_api_h.id_to_fq_name(id)
def set_vxlan_mode(self, vxlan_mode='automatic'):
''' one of automatic or configured
'''
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
vrouter_config = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
vrouter_config.set_vxlan_network_identifier_mode(vxlan_mode)
self.vnc_api_h.global_vrouter_config_update(vrouter_config)
def get_vxlan_mode(self):
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
vrouter_config = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
return vrouter_config.get_vxlan_network_identifier_mode()
# end
def get_global_asn(self, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
return gsc_obj.get_autonomous_system()
# end get_global_asn
def set_global_asn(self, asn, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
gsc_obj.set_autonomous_system(int(asn))
self.vnc_api_h.global_system_config_update(gsc_obj)
class VncLibFixture(fixtures.Fixture):
''' Wrapper for VncApi
:param domain : default is default-domain
:param project_name : default is admin
:param cfgm_ip : default is 127.0.0.1
:param api_port : default is 8082
:param connections : ContrailConnections object. default is None
:param username : default is admin
:param password : default is contrail123
:param auth_server_ip : default is 127.0.0.1
:param project_id : defualt is None
:param logger : logger object
'''
def __init__(self, *args, **kwargs):
self.username = os.getenv('OS_USERNAME') or \
kwargs.get('username', 'admin')
self.password = os.getenv('OS_PASSWORD') or \
kwargs.get('password', 'contrail123')
self.project_name = kwargs.get('project_name', 'admin')
self.domain = kwargs.get('domain', 'default-domain')
self.api_server_port = kwargs.get('api_server_port', '8082')
self.cfgm_ip = kwargs.get('cfgm_ip', '127.0.0.1')
self.logger = kwargs.get('logger', None)
self.connections = kwargs.get('connections', None)
self.orchestrator = kwargs.get('orchestrator', 'openstack')
self.vnc_api_h = None
self.inputs = self.connections.inputs if self.connections \
else kwargs.get('inputs', None)
self.neutron_handle = None
self.auth_server_ip = self.inputs.auth_ip if self.inputs else \
kwargs.get('auth_server_ip', '127.0.0.1')
self.auth_url = self.inputs.auth_url if self.inputs else \
os.getenv('OS_AUTH_URL') or \
'http://%s:5000/v2.0'%self.auth_server_ip
self.project_id = kwargs.get('project_id', None)
# end __init__
def setUp(self):
super(VncLibFixture, self).setUp()
if self.connections:
self.logger = self.connections.logger
self.project_name = self.connections.project_name
self.inputs = self.connections.inputs
self.neutron_handle = self.connections.quantum_h
self.vnc_api_h = self.connections.vnc_lib
self.username = self.connections.username
self.password = self.connections.password
self.cfgm_ip = self.inputs.cfgm_ip
self.auth_server_ip = self.inputs.auth_ip
self.project_id = self.connections.project_id
self.auth_url = 'http://' + self.inputs.auth_ip + ':5000/v2.0'
else:
self.logger = self.logger or contrail_logging.getLogger(__name__)
self.vnc_api_h = VncApi(username=self.username,
password=self.password,
tenant_name=self.project_name,
api_server_host=self.cfgm_ip,
api_server_port=self.api_server_port,
auth_host=self.auth_server_ip)
if not self.project_id:
if self.orchestrator == 'openstack':
self.auth_client = OpenstackAuth(self.username,
self.password,
self.project_name,
auth_url=self.auth_url,
logger=self.logger)
self.project_id = self.auth_client.get_project_id()
elif self.orchestrator == 'vcenter':
self.auth_client = VcenterAuth(self.username,
self.password,
self.project_name,
self.inputs)
self.project_id = self.auth_client.get_project_id()
self.vnc_h = ContrailVncApi(self.vnc_api_h, self.logger)
# end setUp
def cleanUp(self):
super(VncLibFixture, self).cleanUp()
def get_handle(self):
return self.vnc_api_h
# end get_handle
def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(
username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger,
inputs=self.inputs)
self.neutron_handle = self.orch.get_network_handler()
return self.neutron_handle
# end get_neutron_handle
def get_project_obj(self):
if self.connections:
project_id = self.connections.project_id
elif self.project_id:
project_id = self.project_id
else:
project_id = self.vnc_api_h.project_read(
fq_name_str='default-domain:default-project').uuid
parent_obj = self.vnc_api_h.project_read(id=project_id)
return parent_obj
# end get_parent_obj
def get_forwarding_mode(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties()
if vni_obj_properties:
fw_mode = vni_obj_properties.get_forwarding_mode()
else:
fw_mode = None
return fw_mode
# end get_forwarding_mode
def get_vn_subnet_dhcp_flag(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
subnets = vni_obj.network_ipam_refs[0]['attr']
ipam = subnets.get_ipam_subnets()
enable_dhcp = ipam[0].get_enable_dhcp()
return enable_dhcp
# get_vn_subnet_dhcp_flag
def set_rpf_mode(self, vn_fq_name, mode):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = self.vnc_api_h.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties(
) or VirtualNetworkType()
vni_obj_properties.set_rpf(mode)
vni_obj.set_virtual_network_properties(vni_obj_properties)
vnc_lib.virtual_network_update(vni_obj)
# end set_rpf_mode
def id_to_fq_name(self, id):
return self.vnc_api_h.id_to_fq_name(id)
def set_vxlan_mode(self, vxlan_mode='automatic'):
''' one of automatic or configured
'''
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
vrouter_config = self.vnc_api_h.global_vrouter_config_read(
fq_name=fq_name)
vrouter_config.set_vxlan_network_identifier_mode(vxlan_mode)
self.vnc_api_h.global_vrouter_config_update(vrouter_config)
def get_vxlan_mode(self):
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
vrouter_config = self.vnc_api_h.global_vrouter_config_read(
fq_name=fq_name)
return vrouter_config.get_vxlan_network_identifier_mode()
# end
def get_global_asn(self, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
return gsc_obj.get_autonomous_system()
# end get_global_asn
def set_global_asn(self, asn, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
gsc_obj.set_autonomous_system(int(asn))
self.vnc_api_h.global_system_config_update(gsc_obj)
# end set_global_asn
def get_global_forwarding_mode(self):
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
gsc_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
return gsc_obj.get_forwarding_mode()
# end get_global_forwarding_mode
def get_active_forwarding_mode(self, vn_fq_name):
''' Returns l2 or l3 or l2_l3
Returns Vn's forwarding mode if set.
If VN forwarding mode is not set, returns global forwarding mode
If global forwarding mode too is not set, returns 'l2_l3' since this is the default.
"'''
if type(vn_fq_name).__name__ == 'str':
vn_fq_name = vn_fq_name.split(':')
gl_fw_mode = self.get_global_forwarding_mode()
vn_fw_mode = self.get_forwarding_mode(vn_fq_name)
if vn_fw_mode:
return vn_fw_mode
elif gl_fw_mode:
return gl_fw_mode
else:
return 'l2_l3'
#end get_active_forwarding_mode
def set_global_forwarding_mode(self, forwarding_mode):
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
gsc_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
gsc_obj.set_forwarding_mode(forwarding_mode)
self.vnc_api_h.global_vrouter_config_update(gsc_obj)
#end set_global_forwarding_mode
def get_flow_export_rate(self):
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
gv_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
rate = gv_obj.get_flow_export_rate()
if not rate:
# If not set, return 100 , since this is default
return 100
else:
return rate
# end get_flow_export_rate
def set_flow_export_rate(self, value):
'''
Set flow export rate in default global vrouter config
value : Value of flow export rate to be set
'''
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
gv_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
gv_obj.set_flow_export_rate(int(value))
self.vnc_api_h.global_vrouter_config_update(gv_obj)
self.logger.info('Setting flow export rate: %s' % (value))
return True
def __init__(self,
inputs,
logger,
project_name=None,
username=None,
password=None):
self.inputs = inputs
project_name = project_name or self.inputs.project_name
username = username or self.inputs.stack_user
password = password or self.inputs.stack_password
self.username = username
self.password = password
self.project_name = project_name
self.vnc_lib_fixture = VncLibFixture(
username=username,
password=password,
domain=self.inputs.domain_name,
project=project_name,
inputs=self.inputs,
cfgm_ip=self.inputs.cfgm_ip,
api_port=self.inputs.api_server_port)
self.vnc_lib_fixture.setUp()
self.vnc_lib = self.vnc_lib_fixture.get_handle()
self.nova_h = None
self.quantum_h = None
if self.inputs.orchestrator == 'openstack':
self.auth = OpenstackAuth(username, password, project_name,
self.inputs, logger)
self.project_id = self.auth.get_project_id(self.inputs.domain_name,
project_name)
if self.inputs.verify_thru_gui():
self.ui_login = UILogin(self, self.inputs, project_name,
username, password)
self.browser = self.ui_login.browser
self.browser_openstack = self.ui_login.browser_openstack
self.orch = OpenstackOrchestrator(username=username,
password=password,
project_id=self.project_id,
project_name=project_name,
inputs=inputs,
vnclib=self.vnc_lib,
logger=logger)
self.nova_h = self.orch.nova_h
self.quantum_h = self.orch.quantum_h
else: # vcenter
self.auth = VcenterAuth(username, password, project_name,
self.inputs)
self.orch = VcenterOrchestrator(user=username,
pwd=password,
host=self.inputs.auth_ip,
port=self.inputs.auth_port,
dc_name=self.inputs.vcenter_dc,
vnc=self.vnc_lib,
inputs=self.inputs,
logger=logger)
self.api_server_inspects = {}
self.dnsagent_inspect = {}
self.cn_inspect = {}
self.agent_inspect = {}
self.ops_inspects = {}
self.ds_inspect = {}
self.update_inspect_handles()
class VncLibFixture(fixtures.Fixture):
''' Wrapper for VncApi
:param domain : default is default-domain
:param project_name : default is admin
:param cfgm_ip : default is 127.0.0.1
:param api_port : default is 8082
:param connections : ContrailConnections object. default is None
:param username : default is admin
:param password : default is contrail123
:param auth_server_ip : default is 127.0.0.1
:param logger : logger object
'''
def __init__(self, *args, **kwargs):
self.username = os.getenv('OS_USERNAME') or \
kwargs.get('username', 'admin')
self.password = os.getenv('OS_PASSWORD') or \
kwargs.get('password', 'contrail123')
self.project_name = kwargs.get('project_name', 'admin')
self.domain = kwargs.get('domain', 'default-domain')
self.api_server_port = kwargs.get('api_server_port', '8082')
self.cfgm_ip = kwargs.get('cfgm_ip', '127.0.0.1')
self.auth_server_ip = kwargs.get('auth_server_ip', '127.0.0.1')
self.logger = kwargs.get('logger', logging.getLogger(__name__))
self.connections = kwargs.get('connections', None)
self.orchestrator = kwargs.get('orchestrator', 'openstack')
self.vnc_api_h = None
self.auth_client_h = None
self.inputs = kwargs.get('inputs', None)
self.neutron_handle = None
self.auth_url = os.getenv('OS_AUTH_URL')
if self.auth_server_ip:
self.auth_url = 'http://' + self.auth_server_ip + ':5000/v2.0'
# end __init__
def setUp(self):
super(VncLibFixture, self).setUp()
if self.connections:
self.logger = self.connections.logger
self.project_name = self.connections.project_name
self.inputs = self.connections.inputs
self.neutron_handle = self.connections.quantum_h
self.vnc_api_h = self.connections.vnc_lib
self.username = self.connections.username
self.password = self.connections.password
self.cfgm_ip = self.inputs.cfgm_ip
self.auth_server_ip = self.inputs.auth_ip
self.project_id = self.connections.project_id
self.auth_url = 'http://' + self.inputs.auth_ip + ':5000/v2.0'
else:
self.vnc_api_h = VncApi(username=self.username,
password=self.password,
tenant_name=self.project_name,
api_server_host=self.cfgm_ip,
api_server_port=self.api_server_port,
auth_host=self.auth_server_ip)
if self.orchestrator == 'openstack':
self.auth_client = OpenstackAuth(self.username,
self.password,
self.project_name,
auth_url=self.auth_url,
logger=self.logger)
self.project_id = self.auth_client.get_project_id()
elif self.orchestrator == 'vcenter':
self.auth_client = VcenterAuth(self.username, self.password,
self.project_name, self.inputs)
self.project_id = self.auth_client.get_project_id()
# end setUp
def cleanUp(self):
super(VncLibFixture, self).cleanUp()
def get_handle(self):
return self.vnc_api_h
# end get_handle
def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(
username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger,
inputs=None)
self.neutron_handle = self.orch.get_network_handler()
return self.neutron_handle
# end get_neutron_handle
def get_forwarding_mode(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties()
if vni_obj_properties:
fw_mode = vni_obj_properties.get_forwarding_mode()
else:
fw_mode = None
return fw_mode
# end get_forwarding_mode
def get_vn_subnet_dhcp_flag(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
subnets = vni_obj.network_ipam_refs[0]['attr']
ipam = subnets.get_ipam_subnets()
enable_dhcp = ipam[0].get_enable_dhcp()
return enable_dhcp
# get_vn_subnet_dhcp_flag
def set_rpf_mode(self, vn_fq_name, mode):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = self.vnc_api_h.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0]
and vni_record['fq_name'][1] == vn_fq_name.split(":")[1]
and vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties(
) or VirtualNetworkType()
vni_obj_properties.set_rpf(mode)
vni_obj.set_virtual_network_properties(vni_obj_properties)
vnc_lib.virtual_network_update(vni_obj)
# end set_rpf_mode
def id_to_fq_name(self, id):
return self.vnc_api_h.id_to_fq_name(id)
def set_vxlan_mode(self, vxlan_mode='automatic'):
''' one of automatic or configured
'''
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
vrouter_config = self.vnc_api_h.global_vrouter_config_read(
fq_name=fq_name)
vrouter_config.set_vxlan_network_identifier_mode(vxlan_mode)
self.vnc_api_h.global_vrouter_config_update(vrouter_config)
def get_vxlan_mode(self):
fq_name = [
'default-global-system-config', 'default-global-vrouter-config'
]
vrouter_config = self.vnc_api_h.global_vrouter_config_read(
fq_name=fq_name)
return vrouter_config.get_vxlan_network_identifier_mode()
# end
def get_global_asn(self, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
return gsc_obj.get_autonomous_system()
# end get_global_asn
def set_global_asn(self, asn, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
gsc_obj.set_autonomous_system(int(asn))
self.vnc_api_h.global_system_config_update(gsc_obj)
class VncLibFixture(fixtures.Fixture):
''' Wrapper for VncApi
:param domain : default is default-domain
:param project_name : default is admin
:param cfgm_ip : default is 127.0.0.1
:param api_port : default is 8082
:param connections : ContrailConnections object. default is None
:param username : default is admin
:param password : default is contrail123
:param auth_server_ip : default is 127.0.0.1
:param project_id : defualt is None
:param logger : logger object
'''
def __init__(self, *args, **kwargs):
self.username = os.getenv('OS_USERNAME') or \
kwargs.get('username', 'admin')
self.password = os.getenv('OS_PASSWORD') or \
kwargs.get('password', 'contrail123')
self.project_name = kwargs.get('project_name', 'admin')
self.domain = kwargs.get('domain', 'default-domain')
self.api_server_port = kwargs.get('api_server_port', '8082')
self.logger = kwargs.get('logger', None)
self.connections = kwargs.get('connections', None)
self.orchestrator = kwargs.get('orchestrator', 'openstack')
self.vnc_api_h = None
self.inputs = self.connections.inputs if self.connections \
else kwargs.get('inputs', None)
self.neutron_handle = None
self.cfgm_ip = kwargs.get('cfgm_ip', self.inputs.cfgm_ip if self.inputs else '127.0.0.1')
self.auth_server_ip = self.inputs.auth_ip if self.inputs else \
kwargs.get('auth_server_ip', '127.0.0.1')
self.auth_url = self.inputs.auth_url if self.inputs else \
os.getenv('OS_AUTH_URL') or \
'http://%s:5000/v2.0'%self.auth_server_ip
self.project_id = kwargs.get('project_id', None)
# end __init__
def setUp(self):
super(VncLibFixture, self).setUp()
if self.connections:
self.logger = self.connections.logger
self.project_name = self.connections.project_name
self.inputs = self.connections.inputs
self.neutron_handle = self.connections.quantum_h
self.vnc_api_h = self.connections.vnc_lib
self.username = self.connections.username
self.password = self.connections.password
self.cfgm_ip = self.inputs.cfgm_ip
self.auth_server_ip = self.inputs.auth_ip
self.project_id = self.connections.project_id
self.auth_url = 'http://' + self.inputs.auth_ip + ':5000/v2.0'
else:
self.logger = self.logger or contrail_logging.getLogger(__name__)
self.vnc_api_h = VncApi(
username=self.username,
password=self.password,
tenant_name=self.project_name,
api_server_host=self.cfgm_ip,
api_server_port=self.api_server_port,
auth_host=self.auth_server_ip)
if not self.project_id:
if self.orchestrator == 'openstack':
self.auth_client = OpenstackAuth(
self.username,
self.password,
self.project_name,
auth_url=self.auth_url,
logger=self.logger)
self.project_id = self.auth_client.get_project_id()
elif self.orchestrator == 'vcenter':
self.auth_client = VcenterAuth(self.username,
self.password,
self.project_name,
self.inputs
)
self.project_id = self.auth_client.get_project_id()
self.vnc_h = ContrailVncApi(self.vnc_api_h, self.logger)
# end setUp
def cleanUp(self):
super(VncLibFixture, self).cleanUp()
def get_handle(self):
return self.vnc_api_h
# end get_handle
def get_neutron_handle(self):
if self.neutron_handle:
return self.neutron_handle
else:
self.orch = OpenstackOrchestrator(username=self.username,
password=self.password,
project_id=self.project_id,
project_name=self.project_name,
auth_server_ip=self.auth_server_ip,
vnclib=self.vnc_api_h,
logger=self.logger, inputs=self.inputs)
self.neutron_handle = self.orch.get_network_handler()
return self.neutron_handle
# end get_neutron_handle
def get_project_obj(self):
if self.connections:
project_id = self.connections.project_id
elif self.project_id:
project_id = self.project_id
else:
project_id = self.vnc_api_h.project_read(
fq_name_str='default-domain:default-project').uuid
parent_obj = self.vnc_api_h.project_read(id=project_id)
return parent_obj
# end get_parent_obj
def get_forwarding_mode(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties()
if vni_obj_properties:
fw_mode = vni_obj_properties.get_forwarding_mode()
else:
fw_mode = None
return fw_mode
# end get_forwarding_mode
def get_vn_subnet_dhcp_flag(self, vn_fq_name):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = vnc_lib.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
subnets = vni_obj.network_ipam_refs[0]['attr']
ipam = subnets.get_ipam_subnets()
enable_dhcp = ipam[0].get_enable_dhcp()
return enable_dhcp
# get_vn_subnet_dhcp_flag
def set_rpf_mode(self, vn_fq_name, mode):
vnc_lib = self.vnc_api_h
# Figure out VN
vni_list = self.vnc_api_h.virtual_networks_list(
parent_id=self.project_id)['virtual-networks']
for vni_record in vni_list:
if (vni_record['fq_name'][0] == vn_fq_name.split(":")[0] and
vni_record['fq_name'][1] == vn_fq_name.split(":")[1] and
vni_record['fq_name'][2] == vn_fq_name.split(":")[2]):
vni_obj = vnc_lib.virtual_network_read(id=vni_record['uuid'])
vni_obj_properties = vni_obj.get_virtual_network_properties() or VirtualNetworkType()
vni_obj_properties.set_rpf(mode)
vni_obj.set_virtual_network_properties(vni_obj_properties)
vnc_lib.virtual_network_update(vni_obj)
# end set_rpf_mode
def id_to_fq_name(self, id):
return self.vnc_api_h.id_to_fq_name(id)
def set_vxlan_mode(self, vxlan_mode='automatic'):
''' one of automatic or configured
'''
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
vrouter_config = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
vrouter_config.set_vxlan_network_identifier_mode(vxlan_mode)
self.vnc_api_h.global_vrouter_config_update(vrouter_config)
def get_vxlan_mode(self):
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
vrouter_config = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
return vrouter_config.get_vxlan_network_identifier_mode()
# end
def get_global_asn(self, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
return gsc_obj.get_autonomous_system()
# end get_global_asn
def set_global_asn(self, asn, gsc_id=None):
gsc_id = gsc_id or self.vnc_api_h.get_default_global_system_config_id()
gsc_obj = self.vnc_api_h.global_system_config_read(id=gsc_id)
gsc_obj.set_autonomous_system(int(asn))
self.vnc_api_h.global_system_config_update(gsc_obj)
# end set_global_asn
def get_global_forwarding_mode(self):
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
gsc_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
return gsc_obj.get_forwarding_mode()
# end get_global_forwarding_mode
def get_active_forwarding_mode(self,vn_fq_name):
''' Returns l2 or l3 or l2_l3
Returns Vn's forwarding mode if set.
If VN forwarding mode is not set, returns global forwarding mode
If global forwarding mode too is not set, returns 'l2_l3' since this is the default.
"'''
if type(vn_fq_name).__name__ == 'str':
vn_fq_name = vn_fq_name.split(':')
gl_fw_mode = self.get_global_forwarding_mode()
vn_fw_mode = self.get_forwarding_mode(vn_fq_name)
if vn_fw_mode:
return vn_fw_mode
elif gl_fw_mode:
return gl_fw_mode
else:
return 'l2_l3'
#end get_active_forwarding_mode
def set_global_forwarding_mode(self,forwarding_mode):
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
gsc_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
gsc_obj.set_forwarding_mode(forwarding_mode)
self.vnc_api_h.global_vrouter_config_update(gsc_obj)
#end set_global_forwarding_mode
def get_flow_export_rate(self):
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
gv_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
rate = gv_obj.get_flow_export_rate()
if not rate:
# If not set, return 100 , since this is default
return 100
else:
return rate
# end get_flow_export_rate
def set_flow_export_rate(self, value):
'''
Set flow export rate in default global vrouter config
value : Value of flow export rate to be set
'''
fq_name = [ 'default-global-system-config',
'default-global-vrouter-config']
gv_obj = self.vnc_api_h.global_vrouter_config_read(fq_name=fq_name)
gv_obj.set_flow_export_rate(int(value))
self.vnc_api_h.global_vrouter_config_update(gv_obj)
self.logger.info('Setting flow export rate: %s' % (value))
return True
