// +build e2e

package e2e

import (
	"flag"
	"os"
	"testing"

	"k8s.io/api/core/v1"

	"github.com/virtual-kubelet/virtual-kubelet/test/e2e/framework"
)

const (
	defaultNamespace   = v1.NamespaceDefault
	defaultNodeName    = "vkubelet-mock-0"
	defaultTaintKey    = "virtual-kubelet.io/provider"
	defaultTaintValue  = "mock"
	defaultTaintEffect = string(v1.TaintEffectNoSchedule)
)

var (
	// f is the testing framework used for running the test suite.
	f *framework.Framework

	// kubeconfig is the path to the kubeconfig file to use when running the test suite outside a Kubernetes cluster.
	kubeconfig string
	// namespace is the name of the Kubernetes namespace to use for running the test suite (i.e. where to create pods).
	namespace string
	// nodeName is the name of the virtual-kubelet node to test.
	nodeName string
	// taintKey is the key of the taint that is expected to be associated with the virtual-kubelet node to test.
	taintKey string
	// taintValue is the value of the taint that is expected to be associated with the virtual-kubelet node to test.
	taintValue string
	// taintEffect is the effect of the taint that is expected to be associated with the virtual-kubelet node to test.
	taintEffect string
)

func init() {
	flag.StringVar(&kubeconfig, "kubeconfig", "", "path to the kubeconfig file to use when running the test suite outside a kubernetes cluster")
	flag.StringVar(&namespace, "namespace", defaultNamespace, "the name of the kubernetes namespace to use for running the test suite (i.e. where to create pods)")
	flag.StringVar(&nodeName, "node-name", defaultNodeName, "the name of the virtual-kubelet node to test")
	flag.StringVar(&taintKey, "taint-key", defaultTaintKey, "the key of the taint that is expected to be associated with the virtual-kubelet node to test")
	flag.StringVar(&taintValue, "taint-value", defaultTaintValue, "the value of the taint that is expected to be associated with the virtual-kubelet node to test")
	flag.StringVar(&taintEffect, "taint-effect", defaultTaintEffect, "the effect of the taint that is expected to be associated with the virtual-kubelet node to test")
	flag.Parse()
}

func TestMain(m *testing.M) {
	// Set sane defaults in case no values (or empty ones) have been provided.
	setDefaults()
	// Create a new instance of the test framework targeting the specified node.
	f = framework.NewTestingFramework(kubeconfig, namespace, nodeName, taintKey, taintValue, taintEffect)
	// Wait for the virtual-kubelet pod to be ready.
	if err := f.WaitUntilPodReady(namespace, nodeName); err != nil {
		panic(err)
	}
	// Run the test suite.
	os.Exit(m.Run())
}

// setDefaults sets sane defaults in case no values (or empty ones) have been provided.
func setDefaults() {
	if namespace == "" {
		namespace = defaultNamespace
	}
	if nodeName == "" {
		nodeName = defaultNodeName
	}
	if taintKey == "" {
		taintKey = defaultTaintKey
	}
	if taintValue == "" {
		taintValue = defaultTaintValue
	}
	if taintEffect == "" {
		taintEffect = defaultTaintEffect
	}
}