builder

type InitEnvFn func(
	ctx context.Context,
	mgr pkgmgr.Manager,
	client client.Client)

type InitVCSimEnvFn func(ctx *IntegrationTestContextForVCSim)

type IntegrationTestContext struct {
	context.Context
	Client       client.Client
	Namespace    string
	PodNamespace string
}

type IntegrationTestContextForVCSim struct {
	TestContextForVCSim

	Suite  *TestSuite
	NSInfo WorkloadNamespaceInfo
}

type NetworkEnv string

type NewReconcilerFunc func() Reconciler

type Reconciler interface{}

type TestContextForVCSim struct {
	// NOTE: Unit test in the context of test suite framework means we use
	// the fake k8s client, which is sufficient for our needs. Otherwise,
	// unit testing is a little misleading here since we're using vcsim.
	*UnitTestContext

	PodNamespace   string
	VCClient       *govmomi.Client
	VCClientConfig pkgclient.Config
	Datacenter     *object.Datacenter
	Finder         *find.Finder
	RestClient     *rest.Client
	Recorder       record.Recorder
	Datastore      *object.Datastore

	ZoneCount       int
	ClustersPerZone int
	ZoneNames       []string

	// When WithContentLibrary is true:
	LocalContentLibraryID       string
	ContentLibraryImageName     string
	ContentLibraryID            string
	ContentLibraryItemID        string
	ContentLibraryItemVersion   string
	ContentLibraryItemDiskPath  string
	ContentLibraryItemNVRAMPath string

	ContentLibraryIsoImageName   string
	ContentLibraryIsoItemID      string
	ContentLibraryIsoItemVersion string

	// When WithoutStorageClass is false:
	StorageClassName          string
	StorageProfileID          string
	EncryptedStorageClassName string
	EncryptedStorageProfileID string

	// When WithoutEncryptionClass is false:
	EncryptionClass1Name       string
	EncryptionClass1ProviderID string
	EncryptionClass2Name       string
	EncryptionClass2ProviderID string

	// When WithoutNativeKeyProvider is false:
	NativeKeyProviderID string

	NetworkRef  object.NetworkReference
	NetworkRefs []object.NetworkReference
	// contains filtered or unexported fields
}

type TestSuite struct {
	context.Context

	// StartErrExpected may be set to true to expect an error from
	// manager.Start.
	StartErrExpected bool
	// contains filtered or unexported fields
}

type UnitTestContext struct {
	context.Context
	Client client.Client
	Scheme *runtime.Scheme
}

type UnitTestContextForController struct {
	// context is the pkgctx.ControllerManagerContext for being tested.
	pkgctx.ControllerManagerContext

	// Client is the k8s client to access resources.
	Client client.Client

	// reconciler is the reconcile.Reconciler being unit tested.
	Reconciler reconcile.Reconciler

	// Events is a channel that fake recorder records events to.
	Events chan string
}

type UnitTestContextForMutatingWebhook struct {
	// WebhookRequestContext is initialized with fake.NewWebhookRequestContext
	// and is used for unit testing.
	pkgctx.WebhookRequestContext

	// Client is the k8s client to access resources.
	Client client.Client

	// Key may be used to lookup Ctx.Cluster with Ctx.Client.Get.
	Key client.ObjectKey

	// mutator is the builder.Mutator being unit tested.
	builder.Mutator
}

type UnitTestContextForValidatingWebhook struct {
	// WebhookRequestContext is initialized with fake.NewWebhookRequestContext
	// and is used for unit testing.
	pkgctx.WebhookRequestContext

	// Client is the k8s client to access resources.
	Client client.Client

	// Key may be used to lookup Ctx.Obj with Ctx.Client.Get.
	Key client.ObjectKey

	// validator is the builder.Validator being unit tested.
	builder.Validator
}

type VCSimTestConfig struct {
	// NumFaultDomains is the number of zones.
	NumFaultDomains int

	// NumNetworks is the number of networks.
	NumNetworks int

	// WithContentLibrary configures a Content Library, populated with one image's
	// name available in the TestContextForVCSim.ContentLibraryImageName.
	WithContentLibrary bool

	// WithInstanceStorage enables the WCP_INSTANCE_STORAGE FSS.
	WithInstanceStorage bool

	// WithVMResize enables the FSS_WCP_VMSERVICE_RESIZE FSS.
	WithVMResize bool
	// WithVMResize enables the FSS_WCP_VMSERVICE_RESIZE_CPU_MEMORY FSS.
	WithVMResizeCPUMemory bool

	// WithVMIncrementalRestore enables the FSS_WCP_VMSERVICE_INCREMENTAL_RESTORE FSS.
	WithVMIncrementalRestore bool

	// WithVMSnapshots enables the FSS_WCP_VMSERVICE_VM_SNAPSHOTS FSS.
	WithVMSnapshots bool

	// WithoutStorageClass disables the storage class required, meaning that the
	// Datastore will be used instead. In WCP production the storage class is
	// always required; the Datastore is only needed for gce2e.
	WithoutStorageClass bool

	// WithoutWorkloadDomainIsolation disables the WorkloadDomainIsolation capability.
	WithoutWorkloadDomainIsolation bool

	// WithJSONExtraConfig enables additional ExtraConfig that is included when
	// creating a VM.
	WithJSONExtraConfig string

	// WithDefaultNetwork string sets the default network VM NICs will use.
	// In WCP production this is never set; it only exists for current
	// limitations of gce2e.
	WithDefaultNetwork string

	// WithNetworkEnv is the network environment type.
	WithNetworkEnv NetworkEnv

	// WithoutEncryptionClass disables the creation of the EncryptionClass
	// resource in each workload namespace.
	WithoutEncryptionClass bool

	// WithoutNativeKeyProvider disables the creation of the native key provider
	// in vcsim.
	WithoutNativeKeyProvider bool
}

type WorkloadNamespaceInfo struct {
	Namespace             string
	Folder                *object.Folder
	EncryptionClass1KeyID string
	EncryptionClass2KeyID string
}