github.com/verrazzano/verrazzano-monitoring-operator@v0.0.30/pkg/resources/pvcs/pvc_test.go

// Copyright (C) 2020, 2022, Oracle and/or its affiliates.
// Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl.

package pvcs

import (
	"testing"

	"github.com/verrazzano/verrazzano-monitoring-operator/pkg/constants"

	"github.com/stretchr/testify/assert"
	vmcontrollerv1 "github.com/verrazzano/verrazzano-monitoring-operator/pkg/apis/vmcontroller/v1"
)

func TestVMONoStorageVolumes(t *testing.T) {
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			Grafana: vmcontrollerv1.Grafana{
				Enabled: true,
			},
			Kibana: vmcontrollerv1.Kibana{
				Enabled: true,
			},
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, 0, len(pvcs), "Length of generated PVCs")
}

func TestVMOWithStorageVolumes1(t *testing.T) {
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			Grafana: vmcontrollerv1.Grafana{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size:               "50Gi",
					AvailabilityDomain: "AD1",
					PvcNames:           []string{"grafana-pvc"},
				},
			},
			// An empty size element is interpreted as no storage
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size:               "",
					AvailabilityDomain: "AD1",
				},
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, 1, len(pvcs), "Length of generated PVCs")
}

func TestVMOWithStorageVolumes2(t *testing.T) {
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
				DataNode: vmcontrollerv1.ElasticsearchNode{
					Storage: &vmcontrollerv1.Storage{
						Size:               "50Gi",
						AvailabilityDomain: "AD1",
						PvcNames:           []string{"elasticsearch-pvc"},
					},
				},
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, 1, len(pvcs), "Length of generated PVCs")
}

func TestVMOWithStorageVolumes3(t *testing.T) {
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
				Nodes: []vmcontrollerv1.ElasticsearchNode{
					{
						Name: "data",
						Roles: []vmcontrollerv1.NodeRole{
							vmcontrollerv1.DataRole,
						},
						Storage: &vmcontrollerv1.Storage{
							Size:               "100Gi",
							AvailabilityDomain: "AD1",
							PvcNames: []string{
								"p1",
								"p2",
							},
						},
					},
				},
				DataNode: vmcontrollerv1.ElasticsearchNode{
					Storage: &vmcontrollerv1.Storage{
						Size:               "50Gi",
						AvailabilityDomain: "AD1",
						PvcNames:           []string{"elasticsearch-pvc"},
					},
				},
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, 3, len(pvcs), "Length of generated PVCs")
}

func TestVMODevModeWithStorageVolumes(t *testing.T) {
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			Grafana: vmcontrollerv1.Grafana{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size:               "50Gi",
					AvailabilityDomain: "AD1",
					PvcNames:           []string{"grafana-pvc"},
				},
			},
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size: "",
				},
				IngestNode: vmcontrollerv1.ElasticsearchNode{Replicas: 0},
				DataNode:   vmcontrollerv1.ElasticsearchNode{Replicas: 0},
				MasterNode: vmcontrollerv1.ElasticsearchNode{Replicas: 1},
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.Equal(t, 1, len(pvcs), "Length of generated PVCs")
	for _, pvc := range pvcs {
		pvcName := pvc.ObjectMeta.Name
		t.Log("Checking pvc ", pvcName)
		assert.NotContains(t, pvcName, "elasticsearch", "Found unexpected elasticsearch volume")
	}
}

func TestVMOWithCascadingDelete(t *testing.T) {
	// With CascadingDelete
	vmo := &vmcontrollerv1.VerrazzanoMonitoringInstance{
		Spec: vmcontrollerv1.VerrazzanoMonitoringInstanceSpec{
			CascadingDelete: true,
			Grafana: vmcontrollerv1.Grafana{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size:               "50Gi",
					AvailabilityDomain: "AD1",
					PvcNames:           []string{"grafana-pvc"},
				},
			},
			// An empty size element is interpreted as no storage
			Elasticsearch: vmcontrollerv1.Elasticsearch{
				Enabled: true,
				Storage: vmcontrollerv1.Storage{
					Size:               "50Gi",
					AvailabilityDomain: "AD1",
				},
			},
		},
	}
	pvcs, err := New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.True(t, len(pvcs) > 0, "Non-zero length generated PVCs")
	for _, pvs := range pvcs {
		assert.Equal(t, 1, len(pvs.ObjectMeta.OwnerReferences), "OwnerReferences is not set with CascadingDelete true")
	}

	// Without CascadingDelete
	vmo.Spec.CascadingDelete = false
	pvcs, err = New(vmo, constants.OciFlexVolumeProvisioner)
	if err != nil {
		t.Error(err)
	}
	assert.True(t, len(pvcs) > 0, "Non-zero length generated ingresses")
	for _, pvc := range pvcs {
		assert.Equal(t, 0, len(pvc.ObjectMeta.OwnerReferences), "OwnerReferences is set even with CascadingDelete false")
	}
}