def Trigger(tc):
netagent_cfg_api.DeleteEndpoints()
netagent_cfg_api.DeleteNetworks()
netagent_cfg_api.DeleteSgPolicies()
netagent_cfg_api.AddNetworks()
netagent_cfg_api.AddEndpoints()
netagent_cfg_api.AddSgPolicies()
return api.types.status.SUCCESS
def Setup(tc):
tc.workload_pairs = api.GetRemoteWorkloadPairs()
agent_api.DeleteSgPolicies()
if getattr(tc.args, 'vmotion_enable', False):
wloads = []
for pair in tc.workload_pairs:
wloads.append(pair[1])
vmotion_utils.PrepareWorkloadVMotion(tc, wloads)
return api.types.status.SUCCESS
def configurationChangeEvent(tc):
if tc.cancel:
api.Logger.info("Canceling configurationChangeEvent...")
sys.exit(0)
api.Logger.info("Running configurationChangeEvent...")
for proto in ["tcp", "udp"]:
policies = utils.GetTargetJsons(proto)
for policy_json in policies:
# Delete allow-all policy
agent_api.DeleteSgPolicies()
api.Logger.info("Pushing Security policy: %s " % (policy_json))
newObjects = agent_api.AddOneConfig(policy_json)
ret = agent_api.PushConfigObjects(newObjects)
if ret != api.types.status.SUCCESS:
api.Logger.error("Failed to push policies for %s" %
policy_json)
if agent_api.DeleteConfigObjects(newObjects):
api.Logger.error("Failed to delete config object for %s" %
policy_json)
if agent_api.RemoveConfigObjects(newObjects):
api.Logger.error("Failed to remove config object for %s" %
policy_json)
# Restore allow-all policy
agent_api.PushConfigObjects(
agent_api.QueryConfigs(kind='NetworkSecurityPolicy'))
if tc.cancel:
return api.types.status.SUCCESS
for proto in ['tcp', 'udp', 'icmp', 'mixed', 'scale']:
mirrorPolicies = GetTargetJsons('mirror', proto)
flowmonPolicies = GetTargetJsons('flowmon', proto)
for mp_json, fp_json in zip(mirrorPolicies, flowmonPolicies):
mpObjs = agent_api.AddOneConfig(mp_json)
fpObjs = agent_api.AddOneConfig(fp_json)
ret = agent_api.PushConfigObjects(mpObjs + fpObjs)
if ret != api.types.status.SUCCESS:
api.Logger.error("Failed to push the telemetry objects")
ret = agent_api.DeleteConfigObjects(fpObjs + mpObjs)
if ret != api.types.status.SUCCESS:
api.Logger.error("Failed to delete the telemetry objects")
ret = agent_api.RemoveConfigObjects(mpObjs + fpObjs)
if ret != api.types.status.SUCCESS:
api.Logger.error("Failed to remove the telemetry objects")
if tc.cancel:
return api.types.status.SUCCESS
return api.types.status.SUCCESS
def Setup(tc):
setupWorkloadDict(tc, tc.iterators.kind)
if getattr(tc.args, 'vmotion_enable', False):
wllist = []
for w1 in tc.workload_dict.keys():
wllist.append(w1)
vmotion_utils.PrepareWorkloadVMotion(tc, wllist)
agent_api.DeleteSgPolicies()
tc.scale = tc.iterators.scale
result = test.execute()
api.Logger.info("ip rule table module sanity check - %s"%
("SUCCESS" if result else "FAIL"))
if not result:
return api.types.status.FAILURE
if not copyTestCommand(tc):
return api.types.status.FAILURE
if not addPktFltrRuleOnEp(tc, enable=True):
return api.types.status.FAILURE
return api.types.status.SUCCESS
def Setup(tc):
tc.workload_pairs = api.GetRemoteWorkloadPairs()
agent_api.DeleteSgPolicies()
return api.types.status.SUCCESS