Add glide.yaml and vendor deps

vendor/k8s.io/kubernetes/test/e2e/ha_master.go

@@ -0,0 +1,194 @@
+/*
+Copyright 2015 The Kubernetes Authors.
+
+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 e2e
+
+import (
+	"bytes"
+	"fmt"
+	"os/exec"
+	"path"
+	"strconv"
+	"strings"
+	"time"
+
+	. "github.com/onsi/ginkgo"
+	clientset "k8s.io/kubernetes/pkg/client/clientset_generated/release_1_5"
+	"k8s.io/kubernetes/test/e2e/framework"
+)
+
+func addMasterReplica(zone string) error {
+	framework.Logf(fmt.Sprintf("Adding a new master replica, zone: %s", zone))
+	v, _, err := framework.RunCmd(path.Join(framework.TestContext.RepoRoot, "hack/e2e-internal/e2e-add-master.sh"), zone)
+	framework.Logf("%s", v)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func removeMasterReplica(zone string) error {
+	framework.Logf(fmt.Sprintf("Removing an existing master replica, zone: %s", zone))
+	v, _, err := framework.RunCmd(path.Join(framework.TestContext.RepoRoot, "hack/e2e-internal/e2e-remove-master.sh"), zone)
+	framework.Logf("%s", v)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func verifyRCs(c clientset.Interface, ns string, names []string) {
+	for _, name := range names {
+		framework.ExpectNoError(framework.VerifyPods(c, ns, name, true, 1))
+	}
+}
+
+func createNewRC(c clientset.Interface, ns string, name string) {
+	_, err := newRCByName(c, ns, name, 1, nil)
+	framework.ExpectNoError(err)
+}
+
+func verifyNumberOfMasterReplicas(expected int) {
+	output, err := exec.Command("gcloud", "compute", "instances", "list",
+		"--project="+framework.TestContext.CloudConfig.ProjectID,
+		"--regexp="+framework.GenerateMasterRegexp(framework.TestContext.CloudConfig.MasterName),
+		"--filter=status=RUNNING",
+		"--format=[no-heading]").CombinedOutput()
+	framework.Logf("%s", output)
+	framework.ExpectNoError(err)
+	newline := []byte("\n")
+	replicas := bytes.Count(output, newline)
+	framework.Logf("Num master replicas/expected: %d/%d", replicas, expected)
+	if replicas != expected {
+		framework.Failf("Wrong number of master replicas %d expected %d", replicas, expected)
+	}
+}
+
+func findRegionForZone(zone string) string {
+	region, err := exec.Command("gcloud", "compute", "zones", "list", zone, "--quiet", "--format=[no-heading](region)").CombinedOutput()
+	framework.ExpectNoError(err)
+	if string(region) == "" {
+		framework.Failf("Region not found; zone: %s", zone)
+	}
+	return string(region)
+}
+
+func findZonesForRegion(region string) []string {
+	output, err := exec.Command("gcloud", "compute", "zones", "list", "--filter=region="+region,
+		"--quiet", "--format=[no-heading](name)").CombinedOutput()
+	framework.ExpectNoError(err)
+	zones := strings.Split(string(output), "\n")
+	return zones
+}
+
+// removeZoneFromZones removes zone from zones slide.
+// Please note that entries in zones can be repeated. In such situation only one replica is removed.
+func removeZoneFromZones(zones []string, zone string) []string {
+	idx := -1
+	for j, z := range zones {
+		if z == zone {
+			idx = j
+			break
+		}
+	}
+	if idx >= 0 {
+		return zones[:idx+copy(zones[idx:], zones[idx+1:])]
+	}
+	return zones
+}
+
+var _ = framework.KubeDescribe("HA-master [Feature:HAMaster]", func() {
+	f := framework.NewDefaultFramework("ha-master")
+	var c clientset.Interface
+	var ns string
+	var additionalReplicaZones []string
+	var existingRCs []string
+
+	BeforeEach(func() {
+		framework.SkipUnlessProviderIs("gce")
+		c = f.ClientSet
+		ns = f.Namespace.Name
+		verifyNumberOfMasterReplicas(1)
+		additionalReplicaZones = make([]string, 0)
+		existingRCs = make([]string, 0)
+	})
+
+	AfterEach(func() {
+		// Clean-up additional master replicas if the test execution was broken.
+		for _, zone := range additionalReplicaZones {
+			removeMasterReplica(zone)
+		}
+		framework.WaitForMasters(framework.TestContext.CloudConfig.MasterName, c, 1, 10*time.Minute)
+		verifyNumberOfMasterReplicas(1)
+	})
+
+	type Action int
+	const (
+		None Action = iota
+		AddReplica
+		RemoveReplica
+	)
+
+	step := func(action Action, zone string) {
+		switch action {
+		case None:
+		case AddReplica:
+			framework.ExpectNoError(addMasterReplica(zone))
+			additionalReplicaZones = append(additionalReplicaZones, zone)
+		case RemoveReplica:
+			framework.ExpectNoError(removeMasterReplica(zone))
+			additionalReplicaZones = removeZoneFromZones(additionalReplicaZones, zone)
+		}
+		verifyNumberOfMasterReplicas(len(additionalReplicaZones) + 1)
+		framework.WaitForMasters(framework.TestContext.CloudConfig.MasterName, c, len(additionalReplicaZones)+1, 10*time.Minute)
+
+		// Verify that API server works correctly with HA master.
+		rcName := "ha-master-" + strconv.Itoa(len(existingRCs))
+		createNewRC(c, ns, rcName)
+		existingRCs = append(existingRCs, rcName)
+		verifyRCs(c, ns, existingRCs)
+	}
+
+	It("survive addition/removal replicas same zone [Serial][Disruptive]", func() {
+		zone := framework.TestContext.CloudConfig.Zone
+		step(None, "")
+		numAdditionalReplicas := 2
+		for i := 0; i < numAdditionalReplicas; i++ {
+			step(AddReplica, zone)
+		}
+		for i := 0; i < numAdditionalReplicas; i++ {
+			step(RemoveReplica, zone)
+		}
+	})
+
+	It("survive addition/removal replicas different zones [Serial][Disruptive]", func() {
+		zone := framework.TestContext.CloudConfig.Zone
+		region := findRegionForZone(zone)
+		zones := findZonesForRegion(region)
+		zones = removeZoneFromZones(zones, zone)
+
+		step(None, "")
+		// If numAdditionalReplicas is larger then the number of remaining zones in the region,
+		// we create a few masters in the same zone and zone entry is repeated in additionalReplicaZones.
+		numAdditionalReplicas := 2
+		for i := 0; i < numAdditionalReplicas; i++ {
+			step(AddReplica, zones[i%len(zones)])
+		}
+		for i := 0; i < numAdditionalReplicas; i++ {
+			step(RemoveReplica, zones[i%len(zones)])
+		}
+	})
+})