func DeployConsulWithInstanceCount(count int, client bosh.Client, config Config) (manifest destiny.Manifest, kv consul.KV, err error) { guid, err := NewGUID() if err != nil { return } info, err := client.Info() if err != nil { return } manifestConfig := destiny.Config{ DirectorUUID: info.UUID, Name: fmt.Sprintf("consul-%s", guid), } switch info.CPI { case "aws_cpi": manifestConfig.IAAS = destiny.AWS if config.AWS.Subnet != "" { manifestConfig.AWS.Subnet = config.AWS.Subnet } else { err = errors.New("AWSSubnet is required for AWS IAAS deployment") return } case "warden_cpi": manifestConfig.IAAS = destiny.Warden default: err = errors.New("unknown infrastructure type") return } manifest = destiny.NewConsul(manifestConfig) manifest.Jobs[0], manifest.Properties = destiny.SetJobInstanceCount(manifest.Jobs[0], manifest.Networks[0], manifest.Properties, count) yaml, err := manifest.ToYAML() if err != nil { return } yaml, err = client.ResolveManifestVersions(yaml) if err != nil { return } manifest, err = destiny.FromYAML(yaml) if err != nil { return } err = client.Deploy(yaml) if err != nil { return } kv, err = NewKV(manifest, count) return }
func DeployConsulWithInstanceCount(count int, client bosh.Client, config Config) (manifest destiny.Manifest, kv consul.KV, err error) { guid, err := NewGUID() if err != nil { return } info, err := client.Info() if err != nil { return } manifestConfig := destiny.Config{ DirectorUUID: info.UUID, Name: fmt.Sprintf("consul-%s", guid), } switch info.CPI { case "aws_cpi": manifestConfig.IAAS = destiny.AWS if config.AWS.Subnet != "" { manifestConfig.AWS.Subnet = config.AWS.Subnet } else { err = errors.New("AWSSubnet is required for AWS IAAS deployment") return } case "warden_cpi": manifestConfig.IAAS = destiny.Warden default: err = errors.New("unknown infrastructure type") return } manifest = destiny.NewConsul(manifestConfig) manifest.Jobs[0], manifest.Properties = destiny.SetJobInstanceCount(manifest.Jobs[0], manifest.Networks[0], manifest.Properties, count) yaml, err := manifest.ToYAML() if err != nil { return } yaml, err = client.ResolveManifestVersions(yaml) if err != nil { return } manifest, err = destiny.FromYAML(yaml) if err != nil { return } err = client.Deploy(yaml) if err != nil { return } members := manifest.ConsulMembers() if len(members) != count { err = fmt.Errorf("expected %d consul members, found %d", count, len(members)) return } consulMemberAddresses := []string{} for _, member := range members { consulMemberAddresses = append(consulMemberAddresses, member.Address) } dataDir, err := ioutil.TempDir("", "consul") if err != nil { return } configDir, err := ioutil.TempDir("", "consul-config") if err != nil { return } var encryptKey string if len(manifest.Properties.Consul.EncryptKeys) > 0 { encryptKey = manifest.Properties.Consul.EncryptKeys[0] } agent := consul.NewAgent(consul.AgentOptions{ DataDir: dataDir, RetryJoin: consulMemberAddresses, ConfigDir: configDir, Domain: "cf.internal", Key: manifest.Properties.Consul.AgentKey, Cert: manifest.Properties.Consul.AgentCert, CACert: manifest.Properties.Consul.CACert, Encrypt: encryptKey, ServerName: "consul agent", }) agentLocation := "http://127.0.0.1:8500" kv = consul.NewManagedKV(consul.ManagedKVConfig{ Agent: agent, KV: consul.NewHTTPKV(agentLocation), Catalog: consul.NewHTTPCatalog(agentLocation), }) return }
} case "warden_cpi": manifestConfig.IAAS = destiny.Warden default: Fail("unknown infrastructure type") } 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,
Password: "******", }, }) yaml, err := manifest.ToYAML() Expect(err).NotTo(HaveOccurred()) Expect(yaml).To(MatchYAML(turbulenceManifest)) }) }) Describe("FromYAML", func() { It("returns a Manifest matching the given YAML", func() { consulManifest, err := ioutil.ReadFile("fixtures/consul_manifest.yml") Expect(err).NotTo(HaveOccurred()) manifest, err := destiny.FromYAML(consulManifest) Expect(err).NotTo(HaveOccurred()) Expect(manifest).To(Equal(destiny.Manifest{ DirectorUUID: "some-director-uuid", Name: "consul", Releases: []destiny.Release{{ Name: "consul", Version: "latest", }}, Compilation: destiny.Compilation{ Network: "consul1", ReuseCompilationVMs: true, Workers: 3, }, Update: destiny.Update{