stack/ee/agent/internal/informer_stacks_test.go

package internal_test

import (
	"fmt"
	"testing"

	"github.com/formancehq/operator/api/formance.com/v1beta1"
	"github.com/formancehq/stack/components/agent/internal"
	"github.com/formancehq/stack/libs/go-libs/logging"
	"github.com/google/uuid"
	"github.com/stretchr/testify/require"
	"go.uber.org/mock/gomock"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"k8s.io/apimachinery/pkg/runtime"
)

func TestDeleteFunc(t *testing.T) {
	t.Parallel()
	ctrl := gomock.NewController(t)
	membershipClientMock := internal.NewMockMembershipClient(ctrl)
	resourceInformer := internal.NewStackEventHandler(logging.Testing(), membershipClientMock)

	stack := &v1beta1.Stack{
		ObjectMeta: v1.ObjectMeta{
			Name: uuid.NewString(),
		},
	}

	membershipClientMock.EXPECT().Send(gomock.Any())
	unstructuredStack, err := runtime.DefaultUnstructuredConverter.ToUnstructured(stack)
	if err != nil {
		t.Fatalf("failed to convert stack to unstructured: %v", err)
	}

	resourceInformer.DeleteFunc(&unstructured.Unstructured{
		Object: unstructuredStack,
	})

	require.True(t, ctrl.Satisfied())
}

func TestAddStack(t *testing.T) {
	t.Parallel()
	ctrl := gomock.NewController(t)
	membershipClientMock := internal.NewMockMembershipClient(ctrl)
	resourceInformer := internal.NewStackEventHandler(logging.Testing(), membershipClientMock)

	stack := &v1beta1.Stack{
		ObjectMeta: v1.ObjectMeta{
			Name: uuid.NewString(),
		},
		Spec: v1beta1.StackSpec{
			Disabled: true,
		},
	}

	membershipClientMock.EXPECT().Send(gomock.Any())
	unstructuredStack, err := runtime.DefaultUnstructuredConverter.ToUnstructured(stack)
	if err != nil {
		t.Fatalf("failed to convert stack to unstructured: %v", err)
	}
	resourceInformer.AddFunc(&unstructured.Unstructured{
		Object: unstructuredStack,
	})

	require.True(t, ctrl.Satisfied())
}

// We are watching .Status and .Spec fields of the stack resource.
// Simulating a change in the status or spec of the stack resource should trigger a call to the membership client.
func TestUpdateStatus(t *testing.T) {
	type testCase struct {
		isReady    bool
		isDisabled bool

		wasReady    bool
		wasDisabled bool
	}
	testCases := []testCase{}

	for _, b := range []bool{true, false} {
		for _, c := range []bool{true, false} {
			for _, d := range []bool{true, false} {
				for _, e := range []bool{true, false} {
					testCases = append(testCases, testCase{
						isReady:    b,
						isDisabled: c,

						wasReady:    d,
						wasDisabled: e,
					})
				}
			}
		}
	}

	for _, tc := range testCases {
		tc := tc
		t.Run(fmt.Sprintf("isReady: %t isDisabled: %t wasReady: %t wasDisabled: %t", tc.isReady, tc.isDisabled, tc.wasReady, tc.wasDisabled), func(t *testing.T) {
			t.Parallel()

			ctrl := gomock.NewController(t)
			membershipClientMock := internal.NewMockMembershipClient(ctrl)
			resourceInformer := internal.NewStackEventHandler(logging.Testing(), membershipClientMock)

			oldStack := &v1beta1.Stack{
				ObjectMeta: v1.ObjectMeta{
					Name: uuid.NewString(),
				},
				Spec: v1beta1.StackSpec{
					Disabled: tc.wasDisabled,
				},
				Status: v1beta1.StackStatus{
					Status: v1beta1.Status{
						Ready: tc.wasReady,
					},
				},
			}

			newStack := oldStack.DeepCopy()
			newStack.Status.Ready = tc.isReady
			newStack.Spec.Disabled = tc.isDisabled

			if tc.isReady != tc.wasReady || tc.isDisabled != tc.wasDisabled {
				membershipClientMock.EXPECT().Send(gomock.Any())
			}

			unstructuredOldStack, err := runtime.DefaultUnstructuredConverter.ToUnstructured(oldStack)
			if err != nil {
				t.Fatalf("failed to convert old stack to unstructured: %v", err)
			}

			unstructuredNewStack, err := runtime.DefaultUnstructuredConverter.ToUnstructured(newStack)
			if err != nil {
				t.Fatalf("failed to convert new stack to unstructured: %v", err)
			}
			resourceInformer.UpdateFunc(&unstructured.Unstructured{
				Object: unstructuredOldStack,
			}, &unstructured.Unstructured{
				Object: unstructuredNewStack,
			})

			require.True(t, ctrl.Satisfied())
		})
	}
}