func (s *LeaderSuite) SetUpSuite(c *C) {
machines := []string{"http://127.0.0.1:4001"}
s.client = etcd.NewClient(machines)
s.registry = NewLeaderRegistry("customkey", "groupid", 15)
s.masterRegistration = &AppRegistration{Name: "name", Host: "master", Port: 12345}
s.slaveRegistration = &AppRegistration{Name: "name", Host: "slave", Port: 67890}
s.handlerRegistration = &HandlerRegistration{
Name: "name",
Host: "host",
Path: "/path/to/server",
Methods: []string{"PUT"},
Middlewares: []middleware.Middleware{middleware.Middleware{Type: "test", ID: "id", Spec: "hi"}},
}
}
func (s *LBRegistrySuite) SetUpSuite(c *C) {
machines := []string{"http://127.0.0.1:4001"}
s.client = etcd.NewClient(machines)
s.client.Delete("customkey", true)
s.registry, _ = NewLBRegistry("customkey", 15)
s.appRegistration = &AppRegistration{Name: "name", Host: "host", Port: 12345}
s.handlerRegistration = &HandlerRegistration{
Name: "name",
Host: "host",
Path: "/path/to/server",
Methods: []string{"PUT"},
Middlewares: []middleware.Middleware{middleware.Middleware{Type: "test", ID: "id", Spec: "hi"}},
}
}
func (n *ng) reconnect() error {
n.Close()
var client *etcd.Client
if n.options.EtcdCertFile == "" && n.options.EtcdKeyFile == "" {
client = etcd.NewClient(n.nodes)
} else {
var err error
if client, err = etcd.NewTLSClient(n.nodes, n.options.EtcdCertFile, n.options.EtcdKeyFile, n.options.EtcdCaFile); err != nil {
return err
}
}
if err := client.SetConsistency(n.options.EtcdConsistency); err != nil {
return err
}
n.client = client
return nil
}
func (s *VESuite) SetUpSuite(c *C) {
log.InitWithConfig(log.Config{Name: "console"})
etcdNodes := os.Getenv("VULCAND_TEST_ETCD_NODES")
if etcdNodes == "" {
c.Skip("This test requires running Etcd, please provide url via VULCAND_TEST_ETCD_NODES environment variable")
return
}
s.etcdNodes = strings.Split(etcdNodes, ",")
s.client = etcd.NewClient(s.etcdNodes)
s.etcdPrefix = os.Getenv("VULCAND_TEST_ETCD_PREFIX")
if s.etcdPrefix == "" {
c.Skip("This test requires Etcd prefix, please provide url via VULCAND_TEST_ETCD_PREFIX environment variable")
return
}
s.apiUrl = os.Getenv("VULCAND_TEST_API_URL")
if s.apiUrl == "" {
c.Skip("This test requires running vulcand daemon, provide API url via VULCAND_TEST_API_URL environment variable")
return
}
s.serviceUrl = os.Getenv("VULCAND_TEST_SERVICE_URL")
if s.serviceUrl == "" {
c.Skip("This test requires running vulcand daemon, provide API url via VULCAND_TEST_SERVICE_URL environment variable")
return
}
s.sealKey = os.Getenv("VULCAND_TEST_SEAL_KEY")
if s.sealKey == "" {
c.Skip("This test requires running vulcand daemon, provide API url via VULCAND_TEST_SEAL_KEY environment variable")
return
}
key, err := secret.KeyFromString(s.sealKey)
c.Assert(err, IsNil)
box, err := secret.NewBox(key)
c.Assert(err, IsNil)
s.box = box
}
func NewRegistry(config Config) *Registry {
return &Registry{
etcdClient: etcd.NewClient([]string{"http://127.0.0.1:4001"}),
config: config,
}
}
func (s *ClientSuite) SetUpSuite(c *C) {
machines := []string{"http://127.0.0.1:4001"}
s.etcd = etcd.NewClient(machines)
s.client = NewClient("clienttest")
}
func NewClient(key string) *Client {
etcd := etcd.NewClient([]string{etcdMachine})
return &Client{Key: key, etcd: etcd}
}