second commit

Author: abhi

systemtests/init_test.go

@@ -3,6 +3,7 @@ package systemtests
 import (
 	"flag"
 	"os"
+	"os/exec"
 	"strconv"
 	"strings"
 	. "testing"
@@ -26,6 +27,7 @@ type systemtestSuite struct {
 	fwdMode      string
 	clusterStore string
 	enableDNS    bool
+	scheduler    string
 	// user       string
 	// password   string
 	// nodes      []string
@@ -58,6 +60,9 @@ func TestMain(m *M) {
 	} else {
 		flag.StringVar(&sts.fwdMode, "fwd-mode", "routing", "forwarding mode to start the test ")
 	}
+	if os.Getenv("CONTIV_K8") != "" {
+		flag.StringVar(&sts.scheduler, "scheduler", "k8", "scheduler used for testing")
+	}
 	flag.Parse()
 
 	logrus.Infof("Running system test with params: %+v", sts)
@@ -92,6 +97,10 @@ func (s *systemtestSuite) SetUpSuite(c *C) {
 
 	s.nodes = []*node{}
 
+	if s.scheduler == "k8" {
+		s.KubeNodeSetup(c)
+	}
+
 	if s.fwdMode == "routing" {
 		contivL3Nodes := 2
 		c.Assert(s.vagrant.Setup(false, "CONTIV_NODES=3 CONTIV_L3=2", contivNodes+contivL3Nodes), IsNil)
@@ -191,3 +200,8 @@ func (s *systemtestSuite) Test00SSH(c *C) {
 		return node.RunCommand("true")
 	}), IsNil)
 }
+
+func (s *systemtestSuite) KubeNodeSetup(c *C) {
+	cmd := exec.Command("/bin/sh", "./vagrant/k8s/setup_cluster.sh")
+	c.Assert(cmd.Run(), IsNil)
+}