render_test.go

// Copyright (c) 2021 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License.AGPL.txt in the project root for license information.

package migrations

import (
	"testing"

	"github.com/stretchr/testify/require"
	"k8s.io/utils/pointer"

	"github.com/gitpod-io/gitpod/installer/pkg/common"
	config "github.com/gitpod-io/gitpod/installer/pkg/config/v1"
	"github.com/gitpod-io/gitpod/installer/pkg/config/v1/experimental"
	"github.com/gitpod-io/gitpod/installer/pkg/config/versions"
)

func TestJob_IsNotRenderedWhenDisableMigrationIsTrue(t *testing.T) {
	ctx := renderContextWithDisableMigration(t, true)

	objects, err := job(ctx)
	require.NoError(t, err)

	require.Len(t, objects, 0, "must not render any objects")
}

func TestJob_IsRenderedWhenDisableMigrationIsFalse(t *testing.T) {
	ctx := renderContextWithDisableMigration(t, false)

	objects, err := job(ctx)
	require.NoError(t, err)

	require.Len(t, objects, 1, "must render one object")
}

func renderContextWithDisableMigration(t *testing.T, disableMigration bool) *common.RenderContext {
	ctx, err := common.NewRenderContext(config.Config{
		Database: config.Database{
			InCluster: pointer.Bool(true),
		},
		Experimental: &experimental.Config{
			WebApp: &experimental.WebAppConfig{
				DisableMigration: disableMigration,
			},
		},
	}, versions.Manifest{
		Components: versions.Components{
			ServiceWaiter: versions.Versioned{
				Version: "arbitary",
			},
			DBMigrations: versions.Versioned{
				Version: "arbitary",
			},
		},
	}, "test-namespace")
	require.NoError(t, err)

	return ctx
}