kyverno/pkg/cel/utils/convert_test.go

package utils

import (
	"testing"

	"github.com/google/cel-go/common/types"
	"github.com/google/cel-go/common/types/ref"
	"github.com/stretchr/testify/assert"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
)

func TestConvertToNative(t *testing.T) {
	tests := []struct {
		name    string
		value   ref.Val
		want    any
		wantErr bool
	}{{
		name:  "bool ok",
		value: types.False,
		want:  false,
	}, {
		name:    "string ko",
		value:   types.String("false"),
		wantErr: true,
	}}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := ConvertToNative[bool](tt.value)
			if tt.wantErr {
				assert.Error(t, err)
			} else {
				assert.NoError(t, err)
				assert.Equal(t, tt.want, got)
			}
		})
	}
}

func TestConvertObjectToUnstructured(t *testing.T) {
	tests := []struct {
		name    string
		obj     any
		want    *unstructured.Unstructured
		wantErr bool
	}{{
		name: "nil",
		obj:  nil,
		want: &unstructured.Unstructured{},
	}, {
		name: "error",
		obj: map[string]string{
			"foo": "bar",
		},
		wantErr: true,
	}, {
		name: "ok",
		obj: &corev1.Namespace{
			TypeMeta: metav1.TypeMeta{
				APIVersion: corev1.SchemeGroupVersion.String(),
				Kind:       "Namespace",
			},
			ObjectMeta: metav1.ObjectMeta{
				Name: "foo",
			},
		},
		want: &unstructured.Unstructured{
			Object: map[string]any{
				"apiVersion": "v1",
				"kind":       "Namespace",
				"metadata": map[string]any{
					"name":              "foo",
					"creationTimestamp": nil,
				},
				"spec":   map[string]any{},
				"status": map[string]any{},
			},
		},
	},
	// TODO: Add test cases.
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := ConvertObjectToUnstructured(tt.obj)
			if tt.wantErr {
				assert.Error(t, err)
			} else {
				assert.NoError(t, err)
				assert.Equal(t, tt.want, got)
			}
		})
	}
}