Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use NewCheck method of v1 Package

Best K6 code snippet using v1.NewCheck

schema.go

Source:schema.go

1package config2import (3	"bytes"4	"encoding/json"5	"errors"6	"fmt"7	"io"8	"strings"9	"text/template"10	"github.com/qri-io/jsonschema"11	"github.com/thoas/go-funk"12	corev1 "k8s.io/api/core/v1"13	"k8s.io/apimachinery/pkg/api/resource"14	k8sYaml "k8s.io/apimachinery/pkg/util/yaml"15)16// TargetKind represents the part of the config to be validated17type TargetKind string18const (19	// TargetController points to the controller's spec20	TargetController TargetKind = "Controller"21	// TargetContainer points to the container spec22	TargetContainer TargetKind = "Container"23	// TargetPod points to the pod spec24	TargetPod TargetKind = "Pod"25)26// HandledTargets is a list of target names that are explicitly handled27var HandledTargets = []TargetKind{28	TargetController,29	TargetContainer,30	TargetPod,31}32// SchemaCheck is a Polaris check that runs using JSON Schema33type SchemaCheck struct {34	ID                      string                            `yaml:"id" json:"id"`35	Category                string                            `yaml:"category" json:"category"`36	SuccessMessage          string                            `yaml:"successMessage" json:"successMessage"`37	FailureMessage          string                            `yaml:"failureMessage" json:"failureMessage"`38	Controllers             includeExcludeList                `yaml:"controllers" json:"controllers"`39	Containers              includeExcludeList                `yaml:"containers" json:"containers"`40	Target                  TargetKind                        `yaml:"target" json:"target"`41	SchemaTarget            TargetKind                        `yaml:"schemaTarget" json:"schemaTarget"`42	Schema                  map[string]interface{}            `yaml:"schema" json:"schema"`43	SchemaString            string                            `yaml:"schemaString" json:"schemaString"`44	Validator               jsonschema.RootSchema             `yaml:"-" json:"-"`45	AdditionalSchemas       map[string]map[string]interface{} `yaml:"additionalSchemas" json:"additionalSchemas"`46	AdditionalSchemaStrings map[string]string                 `yaml:"additionalSchemaStrings" json:"additionalSchemaStrings"`47	AdditionalValidators    map[string]jsonschema.RootSchema  `yaml:"-" json:"-"`48}49type resourceMinimum string50type resourceMaximum string51func unmarshalYAMLOrJSON(raw []byte, dest interface{}) error {52	reader := bytes.NewReader(raw)53	d := k8sYaml.NewYAMLOrJSONDecoder(reader, 4096)54	for {55		if err := d.Decode(dest); err != nil {56			if err == io.EOF {57				break58			}59			return fmt.Errorf("Decoding schema check failed: %v", err)60		}61	}62	return nil63}64// ParseCheck parses a check from a byte array65func ParseCheck(id string, rawBytes []byte) (SchemaCheck, error) {66	check := SchemaCheck{}67	err := unmarshalYAMLOrJSON(rawBytes, &check)68	if err != nil {69		return check, err70	}71	check.Initialize(id)72	return check, nil73}74func init() {75	jsonschema.RegisterValidator("resourceMinimum", newResourceMinimum)76	jsonschema.RegisterValidator("resourceMaximum", newResourceMaximum)77}78type includeExcludeList struct {79	Include []string `yaml:"include"`80	Exclude []string `yaml:"exclude"`81}82func newResourceMinimum() jsonschema.Validator {83	return new(resourceMinimum)84}85func newResourceMaximum() jsonschema.Validator {86	return new(resourceMaximum)87}88// Validate checks that a specified quanitity is not less than the minimum89func (min resourceMinimum) Validate(path string, data interface{}, errs *[]jsonschema.ValError) {90	err := validateRange(path, string(min), data, true)91	if err != nil {92		*errs = append(*errs, *err...)93	}94}95// Validate checks that a specified quanitity is not greater than the maximum96func (max resourceMaximum) Validate(path string, data interface{}, errs *[]jsonschema.ValError) {97	err := validateRange(path, string(max), data, false)98	if err != nil {99		*errs = append(*errs, *err...)100	}101}102func parseQuantity(i interface{}) (resource.Quantity, *[]jsonschema.ValError) {103	resStr, ok := i.(string)104	if !ok {105		return resource.Quantity{}, &[]jsonschema.ValError{106			{Message: fmt.Sprintf("Resource quantity %v is not a string", i)},107		}108	}109	q, err := resource.ParseQuantity(resStr)110	if err != nil {111		return resource.Quantity{}, &[]jsonschema.ValError{112			{Message: fmt.Sprintf("Could not parse resource quantity: %s", resStr)},113		}114	}115	return q, nil116}117func validateRange(path string, limit interface{}, data interface{}, isMinimum bool) *[]jsonschema.ValError {118	limitQuantity, err := parseQuantity(limit)119	if err != nil {120		return err121	}122	actualQuantity, err := parseQuantity(data)123	if err != nil {124		return err125	}126	cmp := limitQuantity.Cmp(actualQuantity)127	if isMinimum {128		if cmp == 1 {129			return &[]jsonschema.ValError{130				{Message: fmt.Sprintf("%s quantity %v is > %v", path, actualQuantity, limitQuantity)},131			}132		}133	} else {134		if cmp == -1 {135			return &[]jsonschema.ValError{136				{Message: fmt.Sprintf("%s quantity %v is < %v", path, actualQuantity, limitQuantity)},137			}138		}139	}140	return nil141}142// Initialize sets up the schema143func (check *SchemaCheck) Initialize(id string) error {144	check.ID = id145	if check.SchemaString == "" {146		jsonBytes, err := json.Marshal(check.Schema)147		if err != nil {148			return err149		}150		check.SchemaString = string(jsonBytes)151	}152	if check.AdditionalSchemaStrings == nil {153		check.AdditionalSchemaStrings = make(map[string]string)154	}155	for kind, schema := range check.AdditionalSchemas {156		jsonBytes, err := json.Marshal(schema)157		if err != nil {158			return err159		}160		check.AdditionalSchemaStrings[kind] = string(jsonBytes)161	}162	check.Schema = map[string]interface{}{}163	check.AdditionalSchemas = map[string]map[string]interface{}{}164	return nil165}166// TemplateForResource fills out a check's templated fields given a particular resource167func (check SchemaCheck) TemplateForResource(res interface{}) (*SchemaCheck, error) {168	newCheck := check // Make a copy of the check, since we're going to modify the schema169	templateStrings := map[string]string{170		"": newCheck.SchemaString,171	}172	for kind, schema := range newCheck.AdditionalSchemaStrings {173		templateStrings[kind] = schema174	}175	newCheck.SchemaString = ""176	newCheck.AdditionalSchemaStrings = map[string]string{}177	for kind, tmplString := range templateStrings {178		tmpl := template.New(newCheck.ID)179		tmpl, err := tmpl.Parse(tmplString)180		if err != nil {181			return nil, err182		}183		w := bytes.Buffer{}184		err = tmpl.Execute(&w, res)185		if err != nil {186			return nil, err187		}188		if kind == "" {189			newCheck.SchemaString = w.String()190		} else {191			newCheck.AdditionalSchemaStrings[kind] = w.String()192		}193	}194	newCheck.AdditionalValidators = map[string]jsonschema.RootSchema{}195	for kind, schemaStr := range newCheck.AdditionalSchemaStrings {196		val := jsonschema.RootSchema{}197		err := unmarshalYAMLOrJSON([]byte(schemaStr), &val)198		if err != nil {199			return nil, err200		}201		newCheck.AdditionalValidators[kind] = val202	}203	err := unmarshalYAMLOrJSON([]byte(newCheck.SchemaString), &newCheck.Validator)204	if err != nil {205		return nil, err206	}207	return &newCheck, err208}209// CheckPod checks a pod spec against the schema210func (check SchemaCheck) CheckPod(pod *corev1.PodSpec) (bool, []jsonschema.ValError, error) {211	return check.CheckObject(pod)212}213// CheckController checks a controler's spec against the schema214func (check SchemaCheck) CheckController(bytes []byte) (bool, []jsonschema.ValError, error) {215	errs, err := check.Validator.ValidateBytes(bytes)216	return len(errs) == 0, errs, err217}218// CheckContainer checks a container spec against the schema219func (check SchemaCheck) CheckContainer(container *corev1.Container) (bool, []jsonschema.ValError, error) {220	return check.CheckObject(container)221}222// CheckObject checks arbitrary data against the schema223func (check SchemaCheck) CheckObject(obj interface{}) (bool, []jsonschema.ValError, error) {224	bytes, err := json.Marshal(obj)225	if err != nil {226		return false, nil, err227	}228	errs, err := check.Validator.ValidateBytes(bytes)229	return len(errs) == 0, errs, err230}231// CheckAdditionalObjects looks for an object that passes the specified additional schema232func (check SchemaCheck) CheckAdditionalObjects(groupkind string, objects []interface{}) (bool, error) {233	val, ok := check.AdditionalValidators[groupkind]234	if !ok {235		return false, errors.New("No validator found for " + groupkind)236	}237	for _, obj := range objects {238		bytes, err := json.Marshal(obj)239		if err != nil {240			return false, err241		}242		errs, err := val.ValidateBytes(bytes)243		if err != nil {244			return false, err245		}246		if len(errs) == 0 {247			return true, nil248		}249	}250	return false, nil251}252// IsActionable decides if this check applies to a particular target253func (check SchemaCheck) IsActionable(target TargetKind, kind string, isInit bool) bool {254	if funk.Contains(HandledTargets, target) {255		if check.Target != target {256			return false257		}258	} else if string(check.Target) != kind && !strings.HasSuffix(string(check.Target), "/"+kind) {259		return false260	}261	isIncluded := len(check.Controllers.Include) == 0262	for _, inclusion := range check.Controllers.Include {263		if inclusion == kind {264			isIncluded = true265			break266		}267	}268	if !isIncluded {269		return false270	}271	for _, exclusion := range check.Controllers.Exclude {272		if exclusion == kind {273			return false274		}275	}276	if check.Target == TargetContainer {277		isIncluded := len(check.Containers.Include) == 0278		for _, inclusion := range check.Containers.Include {279			if (inclusion == "initContainer" && isInit) || (inclusion == "container" && !isInit) {280				isIncluded = true281				break282			}283		}284		if !isIncluded {285			return false286		}287		for _, exclusion := range check.Containers.Exclude {288			if (exclusion == "initContainer" && isInit) || (exclusion == "container" && !isInit) {289				return false290			}291		}292	}293	return true294}...

report.go

Source:report.go

1/*2 * Copyright 2021 Red Hat3 *4 * Licensed under the Apache License, Version 2.0 (the "License");5 * you may not use this file except in compliance with the License.6 * You may obtain a copy of the License at7 *8 *     http://www.apache.org/licenses/LICENSE-2.09 *10 * Unless required by applicable law or agreed to in writing, software11 * distributed under the License is distributed on an "AS IS" BASIS,12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.13 * See the License for the specific language governing permissions and14 * limitations under the License.15 */16package chartverifier17import (18	"fmt"19	"github.com/redhat-certification/chart-verifier/pkg/chartverifier/checks"20	helmchart "helm.sh/helm/v3/pkg/chart"21)22var ReportApiVersion = "v1"23var ReportKind = "verify-report"24type OutcomeType string25const (26	FailOutcomeType    OutcomeType = "FAIL"27	PassOutcomeType    OutcomeType = "PASS"28	UnknownOutcomeType OutcomeType = "UNKNOWN"29)30type Report struct {31	Apiversion string         `json:"apiversion" yaml:"apiversion"`32	Kind       string         `json:"kind" yaml:"kind"`33	Metadata   ReportMetadata `json:"metadata" yaml:"metadata"`34	Results    []*CheckReport `json:"results" yaml:"results"`35}36type ReportMetadata struct {37	ToolMetadata ToolMetadata        `json:"tool" yaml:"tool"`38	ChartData    *helmchart.Metadata `json:"chart" yaml:"chart"`39	Overrides    string              `json:"chart-overrides" yaml:"chart-overrides"`40}41type ToolMetadata struct {42	Version                    string  `json:"verifier-version" yaml:"verifier-version"`43	Profile                    Profile `json:"profile" yaml:"profile"`44	ChartUri                   string  `json:"chart-uri" yaml:"chart-uri"`45	Digests                    Digests `json:"digests" yaml:"digests"`46	LastCertifiedTimestamp     string  `json:"lastCertifiedTimestamp,omitempty" yaml:"lastCertifiedTimestamp,omitempty"`47	CertifiedOpenShiftVersions string  `json:"certifiedOpenShiftVersions,omitempty" yaml:"certifiedOpenShiftVersions,omitempty"`48	TestedOpenShiftVersion     string  `json:"testedOpenShiftVersion,omitempty" yaml:"testedOpenShiftVersion,omitempty"`49	SupportedOpenShiftVersions string  `json:"supportedOpenShiftVersions,omitempty" yaml:"supportedOpenShiftVersions,omitempty"`50	ProviderDelivery           bool    `json:"providerControlledDelivery" yaml:"providerControlledDelivery"`51}52type Digests struct {53	Chart   string `json:"chart" yaml:"chart"`54	Package string `json:"package,omitempty" yaml:"package,omitempty"`55}56type Profile struct {57	VendorType string `json:"vendorType" yaml:"VendorType"`58	Version    string `json:"version" yaml:"version"`59}60type CheckReport struct {61	Check   checks.CheckName `json:"check" yaml:"check"`62	Type    checks.CheckType `json:"type" yaml:"type"`63	Outcome OutcomeType      `json:"outcome" yaml:"outcome"`64	Reason  string           `json:"reason" yaml:"reason"`65}66func newReport() Report {67	report := Report{Apiversion: ReportApiVersion, Kind: ReportKind}68	report.Metadata = ReportMetadata{}69	report.Metadata.ToolMetadata = ToolMetadata{}70	return report71}72func (c *Report) AddCheck(check checks.Check) *CheckReport {73	newCheck := CheckReport{}74	newCheck.Check = checks.CheckName(fmt.Sprintf("%s/%s", check.CheckId.Version, check.CheckId.Name))75	newCheck.Type = check.Type76	newCheck.Outcome = UnknownOutcomeType77	c.Results = append(c.Results, &newCheck)78	return &newCheck79}80func (cr *CheckReport) SetResult(outcome bool, reason string) {81	if outcome {82		cr.Outcome = PassOutcomeType83	} else {84		cr.Outcome = FailOutcomeType85	}86	cr.Reason = reason87}...

pulse_health.go

Source:pulse_health.go

...70		amem += m.AllocatableMEM71		tcpu += m.TotalCPU72		tmem += m.TotalMEM73	}74	c1 := health.NewCheck("cpu")75	c1.Set(health.S1, ccpu)76	c1.Set(health.S2, acpu)77	c1.Set(health.S3, tcpu)78	c2 := health.NewCheck("mem")79	c2.Set(health.S1, cmem)80	c2.Set(health.S2, amem)81	c2.Set(health.S3, tmem)82	return health.Checks{c1, c2}, nil83}84func (h *PulseHealth) check(ctx context.Context, ns, gvr string) (*health.Check, error) {85	meta, ok := Registry[gvr]86	if !ok {87		meta = ResourceMeta{88			DAO:      &dao.Table{},89			Renderer: &render.Generic{},90		}91	}92	if meta.DAO == nil {93		meta.DAO = &dao.Resource{}94	}95	meta.DAO.Init(h.factory, client.NewGVR(gvr))96	oo, err := meta.DAO.List(ctx, ns)97	if err != nil {98		return nil, err99	}100	c := health.NewCheck(gvr)101	if meta.Renderer.IsGeneric() {102		table, ok := oo[0].(*metav1beta1.Table)103		if !ok {104			return nil, fmt.Errorf("expecting a meta table but got %T", oo[0])105		}106		rows := make(render.Rows, len(table.Rows))107		re, _ := meta.Renderer.(Generic)108		re.SetTable(table)109		for i, row := range table.Rows {110			if err := re.Render(row, ns, &rows[i]); err != nil {111				return nil, err112			}113			if !render.Happy(ns, re.Header(ns), rows[i]) {114				c.Inc(health.S2)...

NewCheck

Using AI Code Generation

1import (2type Check struct {3}4func NewCheck(name string) *Check {5	return &Check{6	}7}8func main() {9	c := NewCheck("check")10	fmt.Println(c.Name)11}

NewCheck

Using AI Code Generation

1import (2func main() {3    fmt.Println("main")4    v1.NewCheck()5}6import (7func NewCheck() {8    fmt.Println("v1.NewCheck")9}10import (11func NewCheck() {12    fmt.Println("v2.NewCheck")13}14import (15func NewCheck() {16    fmt.Println("v2.NewCheck")17}18import (19func NewCheck() {20    fmt.Println("v3.NewCheck")21}22import (23func NewCheck() {24    fmt.Println("v4.NewCheck")25}26import (27func NewCheck() {28    fmt.Println("v5.NewCheck")29}30import (31func NewCheck() {32    fmt.Println("v6.NewCheck")33}34import (35func NewCheck() {36    fmt.Println("v7.NewCheck")37}38import (39func NewCheck() {40    fmt.Println("v8.NewCheck")41}

NewCheck

Using AI Code Generation

1func main() {2 v1.Check()3}4func main() {5 v1.Check()6}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.