9bf05b525d
We're in effect testing the K8s scheduler logic in the test by setting taints, as opposed to the actual VK itself. If we want to make sure the taint is set, we can just observe the node object. Instead, bind to the pod to the VK node explicitly.
61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
// +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"
|
|
)
|
|
|
|
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
|
|
)
|
|
|
|
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.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)
|
|
// 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
|
|
}
|
|
}
|