velero/pkg/backup/pv_skip_tracker_test.go

package backup

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestSummary(t *testing.T) {
	tracker := NewSkipPVTracker()
	tracker.Track("pv5", "", "skipped due to policy")
	tracker.Track("pv3", podVolumeApproach, "it's set to opt-out")
	tracker.Track("pv3", csiSnapshotApproach, "not applicable for CSI ")
	// shouldn't be added
	tracker.Track("", podVolumeApproach, "pvc3 is set to be skipped")
	tracker.Track("pv10", volumeSnapshotApproach, "added by mistake")
	tracker.Untrack("pv10")
	expected := []SkippedPV{
		{
			Name: "pv3",
			Reasons: []PVSkipReason{
				{
					Approach: csiSnapshotApproach,
					Reason:   "not applicable for CSI ",
				},
				{
					Approach: podVolumeApproach,
					Reason:   "it's set to opt-out",
				},
			},
		},
		{
			Name: "pv5",
			Reasons: []PVSkipReason{
				{
					Approach: anyApproach,
					Reason:   "skipped due to policy",
				},
			},
		},
	}
	assert.Equal(t, expected, tracker.Summary())
}