porter/cmd/migrate/enable_cluster_preview_envs/enable_test.go

package enable_cluster_preview_envs

import (
	"testing"

	"github.com/launchdarkly/go-sdk-common/v3/ldcontext"
	"github.com/porter-dev/porter/internal/features"
	lr "github.com/porter-dev/porter/pkg/logger"
)

type FeaturesTestClient struct {
	override bool
}

func (c FeaturesTestClient) BoolVariation(key string, context ldcontext.Context, defaultVal bool) (bool, error) {
	return c.override, nil
}

func TestEnableForProjectEnabled(t *testing.T) {
	logger := lr.NewConsole(true)

	tester := &tester{
		dbFileName: "./porter_cluster_preview_envs_enabled.db",
	}

	setupTestEnv(tester, t)

	defer cleanup(tester, t)

	initProjectPreviewEnabled(tester, t)
	initCluster(tester, t)

	err := EnableClusterPreviewEnvs(tester.DB, &features.Client{Client: FeaturesTestClient{true}}, logger)
	if err != nil {
		t.Fatalf("%v\n", err)
		return
	}

	cluster, err := tester.repo.Cluster().ReadCluster(1, 1)
	if err != nil {
		t.Fatalf("%v\n", err)
		return
	}

	if !cluster.PreviewEnvsEnabled {
		t.Fatalf("expected preview envs to be enabled, got disabled")
	}
}

func TestEnableForProjectDisabled(t *testing.T) {
	logger := lr.NewConsole(true)

	tester := &tester{
		dbFileName: "./porter_cluster_preview_envs_disabled.db",
	}

	setupTestEnv(tester, t)

	defer cleanup(tester, t)

	initProjectPreviewDisabled(tester, t)
	initCluster(tester, t)

	err := EnableClusterPreviewEnvs(tester.DB, &features.Client{Client: FeaturesTestClient{false}}, logger)
	if err != nil {
		t.Fatalf("%v\n", err)
		return
	}

	cluster, err := tester.repo.Cluster().ReadCluster(1, 1)
	if err != nil {
		t.Fatalf("%v\n", err)
		return
	}

	if cluster.PreviewEnvsEnabled {
		t.Fatalf("expected preview envs to be disabled, got enabled")
	}
}