def RunSubTest(self, short_name, version, **kwargs): policy = copy.deepcopy(GOOD_POLICY) os_type = agent_policy.OpsAgentPolicy.Assignment.OsType( agent_policy.OpsAgentPolicy.Assignment.OsType.OsShortName( short_name), version) policy.assignment.os_types = [os_type] return validator.ValidateOpsAgentsPolicy(policy)
def Run(self, args): """See base class.""" release_track = self.ReleaseTrack() client = osconfig_api_utils.GetClientInstance( release_track, api_version_override='v1beta') messages = osconfig_api_utils.GetClientMessages( release_track, api_version_override='v1beta') ops_agents_policy = agent_policy.CreateOpsAgentPolicy( args.description, args.agent_rules, args.group_labels, args.os_types, args.zones, args.instances) validator.ValidateOpsAgentsPolicy(ops_agents_policy) guest_policy = to_guest_policy.ConvertOpsAgentPolicyToGuestPolicy( messages, ops_agents_policy) project = properties.VALUES.core.project.GetOrFail() parent_path = osconfig_command_utils.GetProjectUriPath(project) request = messages.OsconfigProjectsGuestPoliciesCreateRequest( guestPolicy=guest_policy, guestPolicyId=args.POLICY_ID, parent=parent_path, ) service = client.projects_guestPolicies complete_guest_policy = service.Create(request) ops_agents_policy = to_ops_agents.ConvertGuestPolicyToOpsAgentPolicy( complete_guest_policy) return ops_agents_policy
def RunSubTest(self, version, enable_autoupgrade, **kwargs): agent_rule = copy.deepcopy(GOOD_METRICS_AGENT_RULE) agent_rule.version = version agent_rule.enable_autoupgrade = enable_autoupgrade policy = copy.deepcopy(GOOD_POLICY) policy.agent_rules = [agent_rule] return validator.ValidateOpsAgentsPolicy(policy)
def RunSubTest(self, agent_type, version, **kwargs): agent_rule = copy.deepcopy(GOOD_LOGGING_AGENT_RULE) agent_rule.version = version agent_rule.type = agent_type agent_rule.enable_autoupgrade = False policy = copy.deepcopy(GOOD_POLICY) policy.agent_rules = [agent_rule] return validator.ValidateOpsAgentsPolicy(policy)
def Run(self, args): """See base class.""" release_track = self.ReleaseTrack() client = osconfig_api_utils.GetClientInstance( release_track, api_version_override='v1beta') messages = osconfig_api_utils.GetClientMessages( release_track, api_version_override='v1beta') project = properties.VALUES.core.project.GetOrFail() request = messages.OsconfigProjectsGuestPoliciesGetRequest( name=osconfig_command_utils.GetGuestPolicyUriPath( 'projects', project, args.POLICY_ID)) service = client.projects_guestPolicies current_guest_policy = service.Get(request) current_ops_agents_policy = guest_policy_to_ops_agents_policy_converter.ConvertGuestPolicyToOpsAgentPolicy( current_guest_policy) updated_ops_agents_policy = ops_agents_policy.UpdateOpsAgentsPolicy( current_ops_agents_policy, args.description, args.etag, args.agent_rules, args.os_types, [] if args.clear_group_labels else args.group_labels, [] if args.clear_zones else args.zones, [] if args.clear_instances else args.instances) ops_agents_policy_validator.ValidateOpsAgentsPolicy( updated_ops_agents_policy) updated_os_config_policy = ops_agents_policy_to_guest_policy_converter.ConvertOpsAgentPolicyToGuestPolicy( messages, updated_ops_agents_policy, current_guest_policy.recipes) request = messages.OsconfigProjectsGuestPoliciesPatchRequest( guestPolicy=updated_os_config_policy, name=osconfig_command_utils.GetGuestPolicyUriPath( 'projects', project, args.POLICY_ID), updateMask=None, ) complete_os_config_policy = service.Patch(request) complete_ops_agent_policy = guest_policy_to_ops_agents_policy_converter.ConvertGuestPolicyToOpsAgentPolicy( complete_os_config_policy) return complete_ops_agent_policy
def RunSubTest(self, agent_rules, **kwargs): policy = copy.deepcopy(GOOD_POLICY) policy.agent_rules = agent_rules return validator.ValidateOpsAgentsPolicy(policy)
def RunSubTest(self, os_types, **kwargs): policy = copy.deepcopy(GOOD_POLICY) policy.assignment.os_types = os_types return validator.ValidateOpsAgentsPolicy(policy)