579823e6a5
* mock: implement GetStatsSummary Signed-off-by: Paulo Pires <pjpires@gmail.com> * make: use skaffold to deploy vk Signed-off-by: Paulo Pires <pjpires@gmail.com> * test: add an e2e test suite Signed-off-by: Paulo Pires <pjpires@gmail.com> * test: add vendored code Signed-off-by: Paulo Pires <pjpires@gmail.com> * docs: update README.md Signed-off-by: Paulo Pires <pjpires@gmail.com> * ci: run e2e on circleci Signed-off-by: Paulo Pires <pjpires@gmail.com> * make: improve the skaffold target Signed-off-by: Paulo Pires <pjpires@gmail.com> * e2e: fix defer pod deletion Signed-off-by: Paulo Pires <pjpires@gmail.com> * e2e: improve instructions Signed-off-by: Paulo Pires <pjpires@gmail.com> * makefile: default shell is bash Signed-off-by: Paulo Pires <pjpires@gmail.com>
82 lines
2.9 KiB
Go
82 lines
2.9 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"
|
|
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
|
|
}
|
|
}
|