func jobIndexOfLeader(kv consulclient.HTTPKV, client bosh.Client, deploymentName string) (int, error) { resp, err := http.Get(fmt.Sprintf("%s/v1/status/leader", kv.Address())) if err != nil { return -1, err } var leader string if err := json.NewDecoder(resp.Body).Decode(&leader); err != nil { return -1, err } vms, err := client.DeploymentVMs(deploymentName) if err != nil { return -1, err } for _, vm := range vms { if len(vm.IPs) > 0 { if vm.IPs[0] == strings.Split(leader, ":")[0] { return vm.Index, nil } } } return -1, errors.New("could not determine leader") }
func DeploymentVMs(boshClient bosh.Client, deploymentName string) ([]bosh.VM, error) { vms, err := boshClient.DeploymentVMs(deploymentName) if err != nil { return nil, err } for index := range vms { vms[index].IPs = nil } return vms, nil }
turbulenceManifest = destiny.NewTurbulence(manifestConfig) yaml, err := turbulenceManifest.ToYAML() Expect(err).NotTo(HaveOccurred()) yaml, err = client.ResolveManifestVersions(yaml) Expect(err).NotTo(HaveOccurred()) turbulenceManifest, err = destiny.FromYAML(yaml) Expect(err).NotTo(HaveOccurred()) err = client.Deploy(yaml) Expect(err).NotTo(HaveOccurred()) Eventually(func() ([]bosh.VM, error) { return client.DeploymentVMs(turbulenceManifest.Name) }, "1m", "10s").Should(ConsistOf([]bosh.VM{ {"running"}, })) }) By("preparing turbulence client", func() { turbulenceUrl := fmt.Sprintf("https://*****:*****@%s:8080", turbulenceManifest.Properties.TurbulenceAPI.Password, turbulenceManifest.Jobs[0].Networks[0].StaticIPs[0]) turbulenceClient = turbulence.NewClient(turbulenceUrl, 5*time.Minute, 2*time.Second) }) }) var _ = AfterSuite(func() {