func setup(t etcdrunner.TestingT, ec EtcdClient, dc discoverdClient) (discoverdClient, EtcdClient, func()) {
if *fake {
if ec == nil {
ec = newFakeEtcd()
}
if dc == nil {
dc = newFakeDiscoverd()
}
return dc, ec, nil
}
var killEtcd, killDiscoverd func()
if ec == nil {
killEtcd = etcdrunner.RunEtcdServer(t)
ec = etcd.NewClient(nil)
}
if dc == nil {
dc, killDiscoverd = testutil.BootDiscoverd(t, "")
}
return dc, ec, func() {
if killDiscoverd != nil {
dc.Close()
killDiscoverd()
}
if killEtcd != nil {
killEtcd()
}
}
}
func TestMain(m *testing.M) {
var cleanup func()
etcdAddr, cleanup = etcdrunner.RunEtcdServer(etcdLogger{log.New(os.Stderr, "", log.Lmicroseconds|log.Lshortfile)})
exitCode := m.Run()
cleanup()
os.Exit(exitCode)
}
func setup(t etcdrunner.TestingT, ec EtcdClient, dc discoverdClient) (discoverdClient, EtcdClient, func()) {
if *fake {
if ec == nil {
ec = newFakeEtcd()
}
if dc == nil {
dc = newFakeDiscoverd()
}
return dc, ec, nil
}
var killEtcd, killDiscoverd func()
var etcdAddr string
if ec == nil {
etcdAddr, killEtcd = etcdrunner.RunEtcdServer(t)
ec = etcd.NewClient([]string{etcdAddr})
} else if c, ok := ec.(*etcd.Client); ok {
etcdAddr = c.GetCluster()[0]
}
if dc == nil {
dc, killDiscoverd = testutil.BootDiscoverd(t, "", etcdAddr)
}
return dc, ec, func() {
if killDiscoverd != nil {
dc.Close()
killDiscoverd()
}
if killEtcd != nil {
killEtcd()
}
}
}
func newEtcd(t etcdrunner.TestingT) (EtcdClient, func()) {
if *fake {
return newFakeEtcd(), func() {}
}
cleanup := etcdrunner.RunEtcdServer(t)
return etcd.NewClient(nil), cleanup
}
func newEtcd(t etcdrunner.TestingT) (EtcdClient, string, func()) {
if *fake {
return newFakeEtcd(), "", func() {}
}
addr, cleanup := etcdrunner.RunEtcdServer(t)
return etcd.NewClient([]string{addr}), addr, cleanup
}
func setup(t etcdrunner.TestingT) (*discoverdWrapper, func()) {
etcdAddr, killEtcd := etcdrunner.RunEtcdServer(t)
dc, killDiscoverd := testutil.BootDiscoverd(t, "", etcdAddr)
dw := &discoverdWrapper{discoverdClient: dc}
return dw, func() {
killDiscoverd()
killEtcd()
}
}
func (s *S) SetUpSuite(c *C) {
etcdAddr, killEtcd := etcdrunner.RunEtcdServer(c)
discd, killDiscoverd := testutil.BootDiscoverd(c, "", etcdAddr)
s.discd = discd
s.cleanup = func() {
killDiscoverd()
killEtcd()
}
}
func runEtcdServer(t *testing.T) (*etcd.Client, func()) {
kill := etcdrunner.RunEtcdServer(t)
return etcd.NewClient(nil), kill
}
func runEtcdServer(t *testing.T) (*etcd.Client, func()) {
addr, kill := etcdrunner.RunEtcdServer(t)
return etcd.NewClient([]string{addr}), kill
}
