def test_task_id_create_in_put_deploy_one_server_pool_success(self, *args):
"""Test success of id task generate for server pool put deploy success."""
mock_get_user = args[1]
mock_get_pool = args[2]
mock_update_real_pool = args[3]
user = Usuario(id=1, nome='test')
divisaodc = DivisaoDc(id=1, nome='test')
amblog = AmbienteLogico(id=1, nome='test')
gl3 = GrupoL3(id=1, nome='test')
ambiente = Ambiente(id=1,
grupo_l3=gl3,
divisao_dc=divisaodc,
ambiente_logico=amblog)
pool = ServerPool(id=1, identifier='test', environment=ambiente)
mock_update_real_pool.return_value = pool
mock_get_pool.return_value = pool
mock_get_user.return_value = user
redeploy({'id': pool.id}, user.id)
mock_update_real_pool.assert_called_with([{'id': pool.id}], user)
def test_task_id_create_in_delete_deploy_one_server_pool_success(
self, *args):
"""Test success of id task generate for server pool delete deploy success."""
mock_get_user = args[1]
mock_get_pool = args[2]
mock_delete_real_pool = args[3]
user = Usuario(id=1, nome='test')
divisaodc = DivisaoDc(id=1, nome='test')
amblog = AmbienteLogico(id=1, nome='test')
gl3 = GrupoL3(id=1, nome='test')
ambiente = Ambiente(id=1,
grupo_l3=gl3,
divisao_dc=divisaodc,
ambiente_logico=amblog)
optionpool = OptionPool(id=1, type='test', name='test')
healthcheck = Healthcheck(id=1)
pool = ServerPool(id=1,
identifier='test',
environment=ambiente,
servicedownaction=optionpool,
healthcheck=healthcheck)
pool_serializer = serializers.PoolV3Serializer(pool)
mock_delete_real_pool.return_value = pool
mock_get_pool.return_value = pool
mock_get_user.return_value = user
undeploy(pool.id, user.id)
mock_delete_real_pool.assert_called_with([pool_serializer.data], user)
def ambiente_spn_lf(user, rack, environment_list):
vlans, redes, ipv6 = dic_lf_spn(user, rack.numero)
divisaoDC = ['BE', 'FE', 'BORDA', 'BORDACACHOS']
vrfNames = ['BEVrf', 'FEVrf', 'BordaVrf', 'BordaCachosVrf']
spines = ['01', '02', '03', '04']
grupol3 = GrupoL3()
grupol3.nome = rack.nome
grupol3.save()
ambientes = dict()
ambientes['L3'] = rack.nome
ranges = dict()
hosts = dict()
hosts['TIPO'] = 'Ponto a ponto'
ipv6['TIPO'] = 'Ponto a ponto'
for divisaodc, vrf in zip(divisaoDC, vrfNames):
ambientes['DC'] = divisaodc
for i in spines:
ambientes['LOG'] = 'SPINE' + i + 'LEAF'
rede = 'SPINE' + i[1] + 'ipv4'
rede_ipv6 = 'SPINE' + i[1] + 'ipv6'
# cadastro dos ambientes
vlan_name = 'VLAN' + divisaodc + 'LEAF'
ranges['MAX'] = vlans.get(vlan_name)[rack.numero][int(i[1]) -
1] + 119
ranges['MIN'] = vlans.get(vlan_name)[rack.numero][int(i[1]) - 1]
env = criar_ambiente(user, ambientes, ranges, None, None, vrf)
environment_list.append(env)
log.debug(environment_list)
vlan = dict()
vlan['VLAN_NUM'] = vlans.get(vlan_name)[rack.numero][int(i[1]) - 1]
vlan['VLAN_NAME'] = 'VLAN_' + 'SPN' + i[1] + 'LF' + '_' + divisaodc
vlan = criar_vlan(user, vlan, ambientes)
criar_rede(user, hosts['TIPO'], redes.get(rede + '_net'), vlan)
criar_rede_ipv6(user, ipv6['TIPO'], ipv6.get(rede_ipv6 + '_net'),
vlan)
# configuracao do ambiente
hosts['REDE'] = redes.get(rede)
hosts['PREFIX'] = '31'
ambientes['LOG'] = 'SPINE' + i + 'LEAF'
hosts['VERSION'] = 'ipv4'
config_ambiente(user, hosts, ambientes)
ipv6['REDE'] = ipv6.get(rede_ipv6)
ipv6['PREFIX'] = '127'
ipv6['VERSION'] = 'ipv6'
config_ambiente(user, ipv6, ambientes)
return environment_list
def create_l3_environment(env):
"""Create environment."""
try:
env_obj = GrupoL3()
env_obj.create_v3(env)
except EnvironmentErrorV3, e:
raise ValidationAPIException(str(e))
def remover_ambiente(user, lista_amb, rack):
for amb in lista_amb:
amb.remove(user, amb.id)
grupo_l3 = GrupoL3()
try:
grupo_l3 = grupo_l3.get_by_name(rack.nome)
grupo_l3.delete()
except:
pass
def handle_post(self, request, user, *args, **kwargs):
"""Treat requests POST to add Group l3.
URL: groupl3/
"""
try:
self.log.info('Add Group l3')
# User permission
if not has_perm(user, AdminPermission.ENVIRONMENT_MANAGEMENT, AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(3, u'There is no value to the networkapi tag of XML request.')
group_l3_map = networkapi_map.get('group_l3')
if group_l3_map is None:
return self.response_error(3, u'There is no value to the group_l3 tag of XML request.')
# Get XML data
name = group_l3_map.get('name')
try:
GrupoL3.get_by_name(name)
raise GrupoL3NameDuplicatedError(
None, u'Já existe um grupo l3 com o valor name %s.' % name)
except GroupL3NotFoundError:
pass
l3_group = GrupoL3()
# set variables
l3_group.nome = name
try:
# save Group l3
l3_group.save()
except Exception, e:
self.log.error(u'Failed to save the Group l3.')
raise AmbienteError(e, u'Failed to save the Group l3.')
l3_group_map = dict()
l3_group_map['group_l3'] = model_to_dict(
l3_group, exclude=['nome'])
return self.response(dumps_networkapi(l3_group_map))
def criar_ambiente(user,
ambientes,
ranges,
acl_path=None,
filter=None,
vrf=None):
# ambiente cloud
environment = Ambiente()
environment.grupo_l3 = GrupoL3()
environment.ambiente_logico = AmbienteLogico()
environment.divisao_dc = DivisaoDc()
environment.grupo_l3.id = environment.grupo_l3.get_by_name(
ambientes.get('L3')).id
environment.ambiente_logico.id = environment.ambiente_logico.get_by_name(
ambientes.get('LOG')).id
environment.divisao_dc.id = environment.divisao_dc.get_by_name(
ambientes.get('DC')).id
environment.acl_path = acl_path
environment.ipv4_template = None
environment.ipv6_template = None
if vrf is not None:
environment.vrf = vrf
environment.max_num_vlan_1 = ranges.get('MAX')
environment.min_num_vlan_1 = ranges.get('MIN')
environment.max_num_vlan_2 = ranges.get('MAX')
environment.min_num_vlan_2 = ranges.get('MIN')
environment.link = ' '
if filter is not None:
try:
filter_obj = Filter.objects.get(name__iexact=filter)
environment.filter = filter_obj
except ObjectDoesNotExist:
pass
environment.create(user)
return environment
def create_environment(self):
return Ambiente(divisao_dc=DivisaoDc(nome='division_1'), ambiente_logico=AmbienteLogico(nome='env'), grupo_l3=GrupoL3(nome='l3'))
def handle_post(self, request, user, *args, **kwargs):
"""Trata requisições POST para inserir novo Ambiente.
URL: ambiente/ or ambiente/ipconfig/
"""
try:
if not has_perm(user, AdminPermission.ENVIRONMENT_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
return self.not_authorized()
xml_map, attrs_map = loads(request.raw_post_data)
self.log.debug('XML_MAP: %s', xml_map)
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(
3,
u'Não existe valor para a tag networkapi do XML de requisição.'
)
environment_map = networkapi_map.get('ambiente')
if environment_map is None:
return self.response_error(
3,
u'Não existe valor para a tag ambiente do XML de requisição.'
)
link = environment_map.get('link')
if not is_valid_string_maxsize(link, 200, False):
self.log.error(u'Parameter link is invalid. Value: %s', link)
raise InvalidValueError(None, 'link', link)
l3_group_id = environment_map.get('id_grupo_l3')
if not is_valid_int_greater_zero_param(l3_group_id):
self.log.error(
u'The l3_group_id parameter is not a valid value: %s.',
l3_group_id)
raise InvalidValueError(None, 'l3_group_id', l3_group_id)
else:
l3_group_id = int(l3_group_id)
logic_environment_id = environment_map.get('id_ambiente_logico')
if not is_valid_int_greater_zero_param(logic_environment_id):
self.log.error(
u'The logic_environment_id parameter is not a valid value: %s.',
logic_environment_id)
raise InvalidValueError(None, 'logic_environment_id',
logic_environment_id)
else:
logic_environment_id = int(logic_environment_id)
dc_division_id = environment_map.get('id_divisao')
if not is_valid_int_greater_zero_param(dc_division_id):
self.log.error(
u'The dc_division_id parameter is not a valid value: %s.',
dc_division_id)
raise InvalidValueError(None, 'dc_division_id', dc_division_id)
else:
dc_division_id = int(dc_division_id)
filter_id = environment_map.get('id_filter')
if filter_id is not None:
if not is_valid_int_greater_zero_param(filter_id):
self.log.error(
u'Parameter filter_id is invalid. Value: %s.',
filter_id)
raise InvalidValueError(None, 'filter_id', filter_id)
acl_path = environment_map.get('acl_path')
if not is_valid_string_maxsize(acl_path, 250, False):
self.log.error(u'Parameter acl_path is invalid. Value: %s',
acl_path)
raise InvalidValueError(None, 'acl_path', acl_path)
ipv4_template = environment_map.get('ipv4_template')
if not is_valid_string_maxsize(ipv4_template, 250, False):
self.log.error(
u'Parameter ipv4_template is invalid. Value: %s',
ipv4_template)
raise InvalidValueError(None, 'ipv4_template', ipv4_template)
ipv6_template = environment_map.get('ipv6_template')
if not is_valid_string_maxsize(ipv6_template, 250, False):
self.log.error(
u'Parameter ipv6_template is invalid. Value: %s',
ipv6_template)
raise InvalidValueError(None, 'ipv6_template', ipv6_template)
max_num_vlan_1 = environment_map.get('max_num_vlan_1')
min_num_vlan_1 = environment_map.get('min_num_vlan_1')
max_num_vlan_2 = environment_map.get('max_num_vlan_2')
min_num_vlan_2 = environment_map.get('min_num_vlan_2')
# validate max_num_vlan_1 and min_num_vlan_1
if (max_num_vlan_1 is not None and min_num_vlan_1 is None) or (
min_num_vlan_1 is not None and max_num_vlan_1 is None):
self.log.error(
u'Parameters min_num_vlan_1, max_num_vlan_1 is invalid. Values: %s, %s',
(min_num_vlan_1, max_num_vlan_1))
raise InvalidValueError(None, 'min_num_vlan_1, max_num_vlan_1',
min_num_vlan_1 + ',' + max_num_vlan_1)
if max_num_vlan_1 is not None and min_num_vlan_1 is not None:
max_num_vlan_1 = int(max_num_vlan_1)
min_num_vlan_1 = int(min_num_vlan_1)
if max_num_vlan_1 < 1 or min_num_vlan_1 < 1:
self.log.error(
u'Parameters min_num_vlan_1, max_num_vlan_1 is invalid. Values: %s, %s',
(min_num_vlan_1, max_num_vlan_1))
raise InvalidValueError(
None, 'min_num_vlan_1, max_num_vlan_1',
min_num_vlan_1 + ',' + max_num_vlan_1)
if max_num_vlan_1 <= min_num_vlan_1:
self.log.error(
u'Parameters min_num_vlan_1, max_num_vlan_1 is invalid. Values: %s, %s',
(min_num_vlan_1, max_num_vlan_1))
raise InvalidValueError(
None, 'min_num_vlan_1, max_num_vlan_1',
min_num_vlan_1 + ',' + max_num_vlan_1)
else:
max_num_vlan_1 = max_num_vlan_2
min_num_vlan_1 = min_num_vlan_2
# validate max_num_vlan_1 and min_num_vlan_1
# validate max_num_vlan_2 and min_num_vlan_2
if (max_num_vlan_2 is not None and min_num_vlan_2 is None) or (
min_num_vlan_2 is not None and max_num_vlan_2 is None):
self.log.error(
u'Parameters min_num_vlan_2, max_num_vlan_2 is invalid. Values: %s, %s',
(min_num_vlan_2, max_num_vlan_2))
raise InvalidValueError(None, 'min_num_vlan_2, max_num_vlan_2',
min_num_vlan_2 + ',' + max_num_vlan_1)
if max_num_vlan_2 is not None and min_num_vlan_2 is not None:
max_num_vlan_2 = int(max_num_vlan_2)
min_num_vlan_2 = int(min_num_vlan_2)
max_num_vlan_1 = int(max_num_vlan_1)
min_num_vlan_1 = int(min_num_vlan_1)
if max_num_vlan_2 < 1 or min_num_vlan_2 < 1:
self.log.error(
u'Parameters min_num_vlan_2, max_num_vlan_2 is invalid. Values: %s, %s',
(min_num_vlan_2, max_num_vlan_2))
raise InvalidValueError(
None, 'min_num_vlan_2, max_num_vlan_2',
min_num_vlan_2 + ',' + max_num_vlan_1)
if max_num_vlan_2 <= min_num_vlan_2:
self.log.error(
u'Parameters min_num_vlan_2, max_num_vlan_2 is invalid. Values: %s, %s',
(min_num_vlan_2, max_num_vlan_2))
raise InvalidValueError(
None, 'min_num_vlan_2, max_num_vlan_2',
min_num_vlan_2 + ',' + max_num_vlan_1)
else:
max_num_vlan_2 = max_num_vlan_1
min_num_vlan_2 = min_num_vlan_1
# validate max_num_vlan_2 and min_num_vlan_2
vrf = environment_map.get('vrf')
if not is_valid_string_maxsize(vrf, 100, False):
self.log.error(u'Parameter vrf is invalid. Value: %s', vrf)
raise InvalidValueError(None, 'link', vrf)
environment = Ambiente()
environment.grupo_l3 = GrupoL3()
environment.ambiente_logico = AmbienteLogico()
environment.divisao_dc = DivisaoDc()
environment.grupo_l3.id = l3_group_id
environment.ambiente_logico.id = logic_environment_id
environment.divisao_dc.id = dc_division_id
environment.acl_path = fix_acl_path(acl_path)
environment.ipv4_template = ipv4_template
environment.ipv6_template = ipv6_template
environment.max_num_vlan_1 = max_num_vlan_1
environment.min_num_vlan_1 = min_num_vlan_1
environment.max_num_vlan_2 = max_num_vlan_2
environment.min_num_vlan_2 = min_num_vlan_2
environment.vrf = vrf
if filter_id is not None:
environment.filter = Filter()
environment.filter.id = filter_id
environment.link = link
environment.create(user)
# IP Config
ip_config = kwargs.get('ip_config')
# If ip config is set
if ip_config:
# Add this to environment
id_ip_config = environment_map.get('id_ip_config')
# Valid ip config
if not is_valid_int_greater_zero_param(id_ip_config):
raise InvalidValueError(None, 'id_ip_config', id_ip_config)
# Ip config must exists
ip_conf = IPConfig().get_by_pk(id_ip_config)
# Makes the relationship
config = ConfigEnvironment()
config.environment = environment
config.ip_config = ip_conf
config.save()
environment_map = dict()
environment_map['id'] = environment.id
return self.response(
dumps_networkapi({'ambiente': environment_map}))
except GrupoError:
return self.response_error(1)
except XMLError, x:
self.log.error(u'Erro ao ler o XML da requisicao.')
return self.response_error(3, x)
