func Register(internal bool, zone, host, ip string) (*datamodel.ZkRouter, error) {
// create ZkRouter
zkRouter := datamodel.Router(internal, zone, host, ip)
zkRouter.RecordIDs = []string{}
if dns.Provider == nil {
// if we have no dns provider then just save here
return zkRouter, zkRouter.Save()
}
// get record sets
cnames, err := createRecordSets(internal, zone, host, zkRouter)
if err != nil {
return zkRouter, err
}
if len(cnames) == 0 { // no need to do anything, there are no cnames to save
return zkRouter, nil
}
err = dns.Provider.CreateRecords(Region, "CREATE_ROUTER "+host+" with ip "+ip+" in "+zone, cnames)
if err != nil {
return zkRouter, err
}
// add RecordIDs
for _, cname := range cnames {
zkRouter.RecordIDs = append(zkRouter.RecordIDs, cname.ID())
}
return zkRouter, zkRouter.Save()
}
func (s *DeployHelperSuite) TestResolveDepValues(c *C) {
datamodel.Zk.RecursiveDelete(helper.GetBaseEnvPath())
datamodel.Zk.RecursiveDelete(helper.GetBaseRouterPath(true))
datamodel.Zk.RecursiveDelete(helper.GetBaseRouterPath(false))
datamodel.Zk.RecursiveDelete(helper.GetBaseRouterPortsPath(true))
datamodel.Zk.RecursiveDelete(helper.GetBaseRouterPortsPath(false))
datamodel.CreateEnvPath()
datamodel.CreateRouterPaths()
datamodel.Router(true, "dev", "somehost", "1.2.3.4").Save()
zkEnv := datamodel.Env("root")
err := zkEnv.Save()
c.Assert(err, IsNil)
deps, err := ResolveDepValues("app", zkEnv, []string{"hello-go"}, false, &Task{})
c.Assert(err, Not(IsNil))
_, err = datamodel.CreateInstance("hello-go", "1234567890", "root", "myhost")
c.Assert(err, IsNil)
_, err = datamodel.CreateOrUpdateApp(false, false, "app", "ssh://github.com/ooyala/apo", "/", "*****@*****.**")
c.Assert(err, IsNil)
zkApp, err := datamodel.CreateOrUpdateApp(false, true, "hello-go", "ssh://github.com/ooyala/hello-go", "/", "*****@*****.**")
c.Assert(err, IsNil)
c.Assert(zkApp.AddDependerAppData(&DependerAppData{Name: "app", DependerEnvData: map[string]*DependerEnvData{"root": &DependerEnvData{Name: "root"}}}), IsNil)
deps, err = ResolveDepValues("app", zkEnv, []string{"hello-go"}, false, &Task{})
c.Assert(err, IsNil)
c.Assert(deps["dev1"]["hello-go"].DataMap["address"], Equals, fmt.Sprintf("internal-router.1.%s.suffix.com:%d", Region, datamodel.MinRouterPort))
deps, err = ResolveDepValues("app", zkEnv, []string{"hello-go"}, true, &Task{})
c.Assert(err, IsNil)
c.Assert(deps["dev1"]["hello-go"].EncryptedData, Not(Equals), "")
c.Assert(deps["dev1"]["hello-go"].DataMap, IsNil)
scrypto.DecryptAppDep(deps["dev1"]["hello-go"])
c.Assert(deps["dev1"]["hello-go"].DataMap, Not(IsNil))
c.Assert(deps["dev1"]["hello-go"].DataMap["address"], Equals, fmt.Sprintf("internal-router.1.%s.suffix.com:%d", Region, datamodel.MinRouterPort))
}