policy-reporter/pkg/kubernetes/policy_report_client_test.go

package kubernetes_test

import (
	"context"
	"sync"
	"testing"
	"time"

	"github.com/kyverno/policy-reporter/pkg/kubernetes"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
)

func Test_PolicyWatcher(t *testing.T) {
	ctx := context.Background()

	kclient, rclient := NewFakeCilent()
	client := kubernetes.NewPolicyReportClient(kclient, NewMapper(), 100*time.Millisecond)

	eventChan := client.WatchPolicyReports(ctx)

	store := newStore(3)

	wg := sync.WaitGroup{}
	wg.Add(3)

	go func() {
		for event := range eventChan {
			store.Add(event)
			wg.Done()
		}
	}()

	rclient.Create(ctx, &unstructured.Unstructured{Object: policyMap}, metav1.CreateOptions{})
	time.Sleep(10 * time.Millisecond)
	rclient.Update(ctx, &unstructured.Unstructured{Object: policyMap}, metav1.UpdateOptions{})
	time.Sleep(10 * time.Millisecond)
	rclient.Delete(ctx, policyMap["metadata"].(map[string]interface{})["name"].(string), metav1.DeleteOptions{})

	wg.Wait()

	if len(store.List()) != 3 {
		t.Error("Should receive the Added, Updated and Deleted Event")
	}
}

func Test_GetFoundResources(t *testing.T) {
	ctx := context.Background()

	kclient, _ := NewFakeCilent()
	client := kubernetes.NewPolicyReportClient(kclient, NewMapper(), 100*time.Millisecond)

	client.WatchPolicyReports(ctx)

	time.Sleep(1 * time.Second)

	if len(client.GetFoundResources()) != 2 {
		t.Errorf("Should find PolicyReport and ClusterPolicyReport Resource")
	}
}