kind.go

//
// Copyright © 2021 Kris Nóva <kris@nivenly.com>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//   ███╗   ██╗ █████╗ ███╗   ███╗██╗
//   ████╗  ██║██╔══██╗████╗ ████║██║
//   ██╔██╗ ██║███████║██╔████╔██║██║
//   ██║╚██╗██║██╔══██║██║╚██╔╝██║██║
//   ██║ ╚████║██║  ██║██║ ╚═╝ ██║███████╗
//   ╚═╝  ╚═══╝╚═╝  ╚═╝╚═╝     ╚═╝╚══════╝
//

package naml

import (
	"fmt"
	"path"

	"k8s.io/client-go/util/homedir"
	"sigs.k8s.io/kind/pkg/cluster"
	"sigs.k8s.io/kind/pkg/cmd"
)

const (
	// TestClusterName is used to identify the test cluster with Kind.
	TestClusterName string = "namltestcluster"
)

var (
	isStarted      bool   = false
	kubeConfigPath string = path.Join(homedir.HomeDir(), ".kube", "naml.conf")
)

// TestClusterStart can be used to start the test cluster in the TestMain() function.
func TestClusterStart() error {
	if isStarted {
		return nil
	}
	provider := cluster.NewProvider(cluster.ProviderWithDocker(), cluster.ProviderWithLogger(cmd.NewLogger()))
	err := provider.Create(TestClusterName)
	if err != nil {
		defer TestClusterStop()
		return fmt.Errorf("unable to create kind test cluster: %v", err)
	}
	err = provider.ExportKubeConfig(TestClusterName, kubeConfigPath)
	if err != nil {
		return fmt.Errorf("unable to export test kube config: %v", err)
	}
	isStarted = true
	return nil
}

// TestClusterKubeConfigPath will export the kubeconfig path to this directory to use
// for the client in the tests.
func TestClusterKubeConfigPath() string {
	return kubeConfigPath
}

// TestClusterStop can be used to stop the test cluster in the TestMain() function.
func TestClusterStop() error {
	provider := cluster.NewProvider(cluster.ProviderWithDocker())
	err := provider.Delete(TestClusterName, kubeConfigPath)
	if err != nil {
		return fmt.Errorf("unable to delete kind test cluster: %v", err)
	}
	return nil
}