Create unit tests for Audit Log state

internal/controller/runtime/fsm/runtime_fsm_configure_auditlog_test.go

@@ -0,0 +1,136 @@
+package fsm
+
+import (
+	"context"
+	gardener "github.com/gardener/gardener/pkg/apis/core/v1beta1"
+	v1 "github.com/kyma-project/infrastructure-manager/api/v1"
+	"github.com/kyma-project/infrastructure-manager/internal/auditlogging/mocks"
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"testing"
+)
+
+func TestEnableAuditLog(t *testing.T) {
+	t.Run("Should set status on Runtime CR when Audit Log was successfully configured", func(t *testing.T) {
+		// given
+		ctx := context.Background()
+		auditLog := &mocks.AuditLogging{}
+		shoot := shootForTest()
+		instance := runtimeForTest()
+		systemState := &systemState{
+			instance: instance,
+			shoot:    shoot,
+		}
+		expectedRuntimeConditions := []metav1.Condition{
+			{
+				Type:    string(v1.ConditionTypeRuntimeConfigured),
+				Status:  "True",
+				Reason:  string(v1.ConditionReasonConfigurationCompleted),
+				Message: "Audit Log configured",
+			},
+		}
+
+		auditLog.On("Enable", ctx, shoot).Return(true, nil).Once()
+
+		// when
+		fsm := &fsm{AuditLogging: auditLog}
+		stateFn, _, _ := sFnConfigureAuditLog(ctx, fsm, systemState)
+
+		// set the time to its zero value for comparison purposes
+		systemState.instance.Status.Conditions[0].LastTransitionTime = metav1.Time{}
+
+		// then
+		auditLog.AssertExpectations(t)
+		require.Contains(t, stateFn.name(), "sFnUpdateStatus")
+		assert.Equal(t, v1.RuntimeStateReady, string(systemState.instance.Status.State))
+		assert.Equal(t, expectedRuntimeConditions, systemState.instance.Status.Conditions)
+	})
+
+	t.Run("Should requeue in case of error during configuration and set status on Runtime CR", func(t *testing.T) {
+		// given
+		ctx := context.Background()
+		auditLog := &mocks.AuditLogging{}
+		shoot := shootForTest()
+		instance := runtimeForTest()
+		systemState := &systemState{
+			instance: instance,
+			shoot:    shoot,
+		}
+		expectedRuntimeConditions := []metav1.Condition{
+			{
+				Type:    string(v1.ConditionTypeRuntimeConfigured),
+				Status:  "False",
+				Reason:  string(v1.ConditionReasonAuditLogConfigured),
+				Message: "some error during configuration",
+			},
+		}
+
+		auditLog.On("Enable", ctx, shoot).Return(false, errors.New("some error during configuration")).Once()
+
+		// when
+		fsm := &fsm{AuditLogging: auditLog}
+		stateFn, _, _ := sFnConfigureAuditLog(ctx, fsm, systemState)
+
+		// set the time to its zero value for comparison purposes
+		systemState.instance.Status.Conditions[0].LastTransitionTime = metav1.Time{}
+
+		// then
+		auditLog.AssertExpectations(t)
+		require.Contains(t, stateFn.name(), "sFnUpdateStatus")
+		assert.Equal(t, v1.RuntimeStateFailed, string(systemState.instance.Status.State))
+		assert.Equal(t, expectedRuntimeConditions, systemState.instance.Status.Conditions)
+	})
+
+	t.Run("Should requeue if initial criteria of enabling Audit Log is not met", func(t *testing.T) {
+		// given
+		ctx := context.Background()
+		auditLog := &mocks.AuditLogging{}
+		shoot := shootForTest()
+		instance := runtimeForTest()
+		systemState := &systemState{
+			instance: instance,
+			shoot:    shoot,
+		}
+
+		auditLog.On("Enable", ctx, shoot).Return(false, nil).Once()
+
+		// when
+		fsm := &fsm{AuditLogging: auditLog}
+		stateFn, result, _ := sFnConfigureAuditLog(ctx, fsm, systemState)
+
+		// then
+		auditLog.AssertExpectations(t)
+		require.Nil(t, stateFn)
+		require.Equal(t, true, result.Requeue)
+	})
+}
+
+func shootForTest() *gardener.Shoot {
+	return &gardener.Shoot{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-shoot",
+			Namespace: "namespace",
+		},
+		Spec: gardener.ShootSpec{
+			Region:   "region",
+			Provider: gardener.Provider{Type: "aws"}},
+	}
+}
+
+func runtimeForTest() v1.Runtime {
+	return v1.Runtime{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-runtime",
+			Namespace: "namespace",
+		},
+		Spec: v1.RuntimeSpec{
+			Shoot: v1.RuntimeShoot{
+				Name:     "test-shoot",
+				Region:   "region",
+				Provider: v1.Provider{Type: "aws"},
+			},
+		},
+	}
+}