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