fail.go

package chaos

import (
	"context"

	chaos "github.com/chaos-mesh/chaos-mesh/api/v1alpha1"
	apimetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	"github.com/spacemeshos/go-spacemesh/systest/testcontext"
)

func selectPods(pods []string) chaos.PodSelectorSpec {
	return chaos.PodSelectorSpec{
		GenericSelectorSpec: chaos.GenericSelectorSpec{
			ExpressionSelectors: chaos.LabelSelectorRequirements{
				apimetav1.LabelSelectorRequirement{
					Key:      "id",
					Operator: apimetav1.LabelSelectorOpIn,
					Values:   pods,
				},
			},
		},
	}
}

// Teardown is returned by every chaos action and executed
// by the caller once chaos needs to be stopped.
type Teardown func(context.Context) error

// Fail the list of pods and prevents them from respawning until teardown is called.
func Fail(cctx *testcontext.Context, name string, pods ...string) (Teardown, error) {
	fail := chaos.PodChaos{}
	fail.Name = name
	fail.Namespace = cctx.Namespace

	fail.Spec.Action = chaos.PodFailureAction
	fail.Spec.Mode = chaos.AllMode
	fail.Spec.Selector = selectPods(pods)
	if err := cctx.Generic.Create(cctx, &fail); err != nil {
		return nil, err
	}
	return func(ctx context.Context) error {
		return cctx.Generic.Delete(ctx, &fail)
	}, nil
}