porter/internal/repository/gorm/event_test.go

package gorm_test

import (
	"fmt"
	"testing"
	"time"

	"github.com/go-test/deep"
	"github.com/porter-dev/porter/api/types"
	"github.com/porter-dev/porter/internal/encryption"
	"github.com/porter-dev/porter/internal/models"
	"gorm.io/gorm"
)

func TestCreateKubeEvent(t *testing.T) {
	tester := &tester{
		dbFileName: "./porter_create_event.db",
	}

	setupTestEnv(tester, t)
	initProject(tester, t)
	initCluster(tester, t)
	defer cleanup(tester, t)

	event := &models.KubeEvent{
		ProjectID: tester.initProjects[0].Model.ID,
		ClusterID: tester.initClusters[0].Model.ID,
		Name:      "pod-example-1",
		Namespace: "default",
	}

	copyKubeEvent := *event

	event, err := tester.repo.KubeEvent().CreateEvent(event)
	if err != nil {
		t.Fatalf("%v\n", err)
	}

	// append a sub event as well
	subEvent := &models.KubeSubEvent{
		EventType: "pod",
		Message:   "Pod killed",
		Reason:    "OOM: memory limit exceeded",
		Timestamp: time.Now(),
	}

	copySubEvent := *subEvent
	copySubEvent.KubeEventID = 1

	err = tester.repo.KubeEvent().AppendSubEvent(event, subEvent)

	if err != nil {
		t.Fatalf("%v\n", err)
	}

	copyKubeEvent.SubEvents = []models.KubeSubEvent{copySubEvent}

	event, err = tester.repo.KubeEvent().ReadEvent(event.Model.ID, 1, 1)

	if err != nil {
		t.Fatalf("%v\n", err)
	}

	// make sure id is 1 and name is "ecr"
	if event.Model.ID != 1 {
		t.Errorf("incorrect registry ID: expected %d, got %d\n", 1, event.Model.ID)
	}

	event.Model = gorm.Model{}
	event.SubEvents[0].Model = gorm.Model{}

	if diff := deep.Equal(event, &copyKubeEvent); diff != nil {
		t.Errorf("events not equal:")
		t.Error(diff)
	}
}

func TestReadKubeEventsByGroup(t *testing.T) {
	suffix, _ := encryption.GenerateRandomBytes(4)

	tester := &tester{
		dbFileName: fmt.Sprintf("./porter_read_event_%s.db", suffix),
	}

	setupTestEnv(tester, t)
	initProject(tester, t)
	initCluster(tester, t)
	initKubeEvents(tester, t)
	defer cleanup(tester, t)

	event, err := tester.repo.KubeEvent().ReadEventByGroup(
		tester.initProjects[0].Model.ID,
		tester.initClusters[0].Model.ID,
		&types.GroupOptions{
			Name:          "pod-example-1",
			Namespace:     "default",
			ResourceType:  "pod",
			ThresholdTime: time.Now().Add(-15 * time.Minute),
		},
	)
	if err != nil {
		t.Fatalf("%v\n", err)
	}

	expKubeEvent := tester.initKubeEvents[1]

	if diff := deep.Equal(expKubeEvent, event); diff != nil {
		t.Errorf("incorrect events")
		t.Error(diff)
	}
}

func TestListKubeEventsByProjectIDWithLimit(t *testing.T) {
	suffix, _ := encryption.GenerateRandomBytes(4)

	tester := &tester{
		dbFileName: fmt.Sprintf("./porter_list_events_%s.db", suffix),
	}

	setupTestEnv(tester, t)
	initProject(tester, t)
	initCluster(tester, t)
	initKubeEvents(tester, t)
	defer cleanup(tester, t)

	testListKubeEventsByProjectID(tester, t, 1, true, &types.ListKubeEventRequest{
		Limit:        10,
		ResourceType: "node",
	}, tester.initKubeEvents[50:60])
}

func TestListKubeEventsByProjectIDWithSkip(t *testing.T) {
	suffix, _ := encryption.GenerateRandomBytes(4)

	tester := &tester{
		dbFileName: fmt.Sprintf("./porter_list_events_%s.db", suffix),
	}

	setupTestEnv(tester, t)
	initProject(tester, t)
	initCluster(tester, t)
	initKubeEvents(tester, t)
	defer cleanup(tester, t)

	testListKubeEventsByProjectID(tester, t, 1, true, &types.ListKubeEventRequest{
		Limit: 25,
		Skip:  10,
	}, tester.initKubeEvents[10:35])
}

func TestDeleteKubeEvents(t *testing.T) {
	suffix, _ := encryption.GenerateRandomBytes(4)

	tester := &tester{
		dbFileName: fmt.Sprintf("./porter_delete_events_%s.db", suffix),
	}

	setupTestEnv(tester, t)
	initProject(tester, t)
	initCluster(tester, t)
	initKubeEvents(tester, t)
	defer cleanup(tester, t)

	// delete a specific event and then test list again
	err := tester.repo.KubeEvent().DeleteEvent(11)
	if err != nil {
		t.Fatalf("%v\n", err)
	}

	testListKubeEventsByProjectID(tester, t, 1, true, &types.ListKubeEventRequest{
		Limit: 25,
		Skip:  10,
	}, tester.initKubeEvents[11:36])
}

func testListKubeEventsByProjectID(tester *tester, t *testing.T, clusterID uint, decrypt bool, opts *types.ListKubeEventRequest, expKubeEvents []*models.KubeEvent) {
	t.Helper()

	events, _, err := tester.repo.KubeEvent().ListEventsByProjectID(
		tester.initProjects[0].Model.ID,
		clusterID,
		opts,
	)
	if err != nil {
		t.Fatalf("%v\n", err)
	}

	// make sure data is correct
	if len(events) != len(expKubeEvents) {
		t.Fatalf("length of events incorrect: expected %d, got %d\n", len(expKubeEvents), len(events))
	}

	if diff := deep.Equal(expKubeEvents, events); diff != nil {
		t.Errorf("incorrect events")
		t.Error(diff)
	}
}