brokercell.go

/*
Copyright 2020 Google LLC

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package testing

import (
	"context"
	"time"

	"github.com/google/knative-gcp/pkg/apis/intevents/v1alpha1"
	intv1alpha1 "github.com/google/knative-gcp/pkg/apis/intevents/v1alpha1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// BrokerCellOption enables further configuration of a BrokerCell.
type BrokerCellOption func(*intv1alpha1.BrokerCell)

// NewBrokerCell creates a BrokerCell with BrokerCellOptions.
func NewBrokerCell(name, namespace string, o ...BrokerCellOption) *intv1alpha1.BrokerCell {
	bc := &intv1alpha1.BrokerCell{
		ObjectMeta: metav1.ObjectMeta{
			Namespace: namespace,
			Name:      name,
		},
	}
	for _, opt := range o {
		opt(bc)
	}
	return bc
}

func WithBrokerCellAnnotations(annotations map[string]string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.SetAnnotations(annotations)
	}
}

// WithInitBrokerCellConditions initializes the BrokerCell's conditions.
func WithInitBrokerCellConditions(bc *intv1alpha1.BrokerCell) {
	bc.Status.InitializeConditions()
}

func WithBrokerCellFinalizers(finalizers ...string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Finalizers = finalizers
	}
}

func WithBrokerCellGeneration(gen int64) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Generation = gen
	}
}

func WithBrokerCellStatusObservedGeneration(gen int64) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.ObservedGeneration = gen
	}
}

func WithBrokerCellDeletionTimestamp(bc *intv1alpha1.BrokerCell) {
	t := metav1.NewTime(time.Unix(1e9, 0))
	bc.ObjectMeta.SetDeletionTimestamp(&t)
}

func WithIngressTemplate(address string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.SetIngressTemplate(address)
	}
}

func WithBrokerCellReady(bc *intv1alpha1.BrokerCell) {
	bc.Status = *intv1alpha1.TestHelper.ReadyBrokerCellStatus()
}

func WithBrokerCellIngressFailed(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkIngressFailed(reason, msg)
	}
}

func WithBrokerCellIngressUnknown(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkIngressUnknown(reason, msg)
	}
}

func WithBrokerCellIngressAvailable() BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.PropagateIngressAvailability(v1alpha1.TestHelper.AvailableEndpoints(), v1alpha1.TestHelper.AvailableDeployment())
	}
}

func WithBrokerCellFanoutAvailable() BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.PropagateFanoutAvailability(v1alpha1.TestHelper.AvailableDeployment())
	}
}

func WithBrokerCellFanoutFailed(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkFanoutFailed(reason, msg)
	}
}

func WithBrokerCellFanoutUnknown(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkFanoutUnknown(reason, msg)
	}
}

func WithBrokerCellRetryAvailable() BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.PropagateFanoutAvailability(v1alpha1.TestHelper.AvailableDeployment())
	}
}

func WithBrokerCellRetryFailed(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkRetryFailed(reason, msg)
	}
}

func WithBrokerCellRetryUnknown(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkRetryUnknown(reason, msg)
	}
}

func WithTargetsCofigReady() BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkTargetsConfigReady()
	}
}

func WithTargetsCofigFailed(reason, msg string) BrokerCellOption {
	return func(bc *intv1alpha1.BrokerCell) {
		bc.Status.MarkTargetsConfigFailed(reason, msg)
	}
}

func WithBrokerCellSetDefaults(bc *intv1alpha1.BrokerCell) {
	bc.SetDefaults(context.Background())
}