def reset_ea_for_zone(ib_zone):
if not ib_zone or not ib_zone.extattrs:
return
utils.reset_required_eas(ib_zone)
eas = ib_zone.extattrs
ea_dict = eas.to_dict()
map(lambda ea: ea_dict.pop(ea, None), const.ZONE_EA_LIST)
ib_zone.extattrs = ib_objects.EA.from_dict(ea_dict)
def reset_ea_for_network(ib_network):
if not ib_network or not ib_network.extattrs:
return
utils.reset_required_eas(ib_network)
eas = ib_network.extattrs
ea_dict = eas.to_dict()
map(lambda ea: ea_dict.pop(ea, None), const.NETWORK_EA_LIST)
ib_network.extattrs = ib_objects.EA.from_dict(ea_dict)
def reset_ea_for_range(ib_range):
if not ib_range or not ib_range.extattrs:
return
utils.reset_required_eas(ib_range)
eas = ib_range.extattrs
ea_dict = eas.to_dict()
list(map(lambda ea: ea_dict.pop(ea, None), const.RANGE_EA_LIST))
ib_range.extattrs = ib_objects.EA.from_dict(ea_dict)
def reset_ea_for_network(ib_network):
if not ib_network or not ib_network.extattrs:
return
utils.reset_required_eas(ib_network)
eas = ib_network.extattrs
ea_dict = eas.to_dict()
map(lambda ea: ea_dict.pop(ea, None), const.NETWORK_EA_LIST)
ib_network.extattrs = ib_objects.EA.from_dict(ea_dict)
def reset_ea_for_zone(ib_zone):
if not ib_zone or not ib_zone.extattrs:
return
utils.reset_required_eas(ib_zone)
eas = ib_zone.extattrs
ea_dict = eas.to_dict()
map(lambda ea: ea_dict.pop(ea, None), const.ZONE_EA_LIST)
ib_zone.extattrs = ib_objects.EA.from_dict(ea_dict)
def test_reset_required_eas(self):
network_ea = {
'CMP Type': {
'value': 'OpenStack'
},
'Cloud API Owned': {
'value': 'True'
},
'Tenant ID': {
'value': 'test-id'
},
'Tenant Name': {
'value': 'tenant-name'
},
'Account': {
'value': 'admin'
},
'Network View ID': {
'value': 'default'
},
'Is External': {
'value': 'False'
},
'Is Shared': {
'value': 'True'
},
'Network ID': {
'value': 'True'
},
'Network Name': {
'value': 'True'
},
'Subnet ID': {
'value': 'True'
}
}
ib_network_ea = ib_objects.EA.from_dict(network_ea)
ib_network_mock = mock.Mock(extattrs=ib_network_ea)
utils.reset_required_eas(ib_network_mock)
for ea in const.REQUIRED_EA_LIST:
ea_value = ib_network_mock.extattrs.get(ea)
if ea == const.EA_CLOUD_API_OWNED:
assert ea_value == 'False'
else:
assert ea_value == const.EA_RESET_VALUE
def test_reset_required_eas(self):
network_ea = {'CMP Type': {'value': 'OpenStack'},
'Cloud API Owned': {'value': 'True'},
'Tenant ID': {'value': 'test-id'},
'Tenant Name': {'value': 'tenant-name'},
'Account': {'value': 'admin'},
'Network View ID': {'value': 'default'},
'Is External': {'value': 'False'},
'Is Shared': {'value': 'True'},
'Network ID': {'value': 'True'},
'Network Name': {'value': 'True'},
'Subnet ID': {'value': 'True'}}
ib_network_ea = ib_objects.EA.from_dict(network_ea)
ib_network_mock = mock.Mock(extattrs=ib_network_ea)
utils.reset_required_eas(ib_network_mock)
for ea in const.REQUIRED_EA_LIST:
ea_value = ib_network_mock.extattrs.get(ea)
if ea == const.EA_CLOUD_API_OWNED:
assert ea_value == 'False'
else:
assert ea_value == const.EA_RESET_VALUE
