docs/discovery/zookeeper/zookeeper_test.go

26 lines
723 B
Go

package zookeeper
import (
"testing"
"github.com/docker/swarm/discovery"
"github.com/stretchr/testify/assert"
)
func TestInitialize(t *testing.T) {
service := &ZkDiscoveryService{}
assert.Error(t, service.Initialize("127.0.0.1/path", 0))
assert.Equal(t, service.path, "/path")
assert.Error(t, service.Initialize("127.0.0.1,127.0.0.2,127.0.0.3/path", 0))
assert.Equal(t, service.path, "/path")
}
func TestCreateNodes(t *testing.T) {
service := &ZkDiscoveryService{}
assert.Equal(t, service.createNodes(nil), []*discovery.Node{})
nodes := service.createNodes([]string{"127.0.0.1", "127.0.0.2"})
assert.Equal(t, nodes[0].String(), "http://127.0.0.1")
assert.Equal(t, nodes[1].String(), "http://127.0.0.2")
}