mirror of https://github.com/docker/docs.git
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package zookeeper
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/docker/swarm/discovery"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInitialize(t *testing.T) {
|
|
service := &ZkDiscoveryService{}
|
|
|
|
assert.Equal(t, service.Initialize("127.0.0.1", 0).Error(), "invalid format \"127.0.0.1\", missing <path>")
|
|
|
|
assert.Error(t, service.Initialize("127.0.0.1/path", 0))
|
|
assert.Equal(t, service.fullpath(), "/path")
|
|
|
|
assert.Error(t, service.Initialize("127.0.0.1,127.0.0.2,127.0.0.3/path", 0))
|
|
assert.Equal(t, service.fullpath(), "/path")
|
|
|
|
assert.Error(t, service.Initialize("127.0.0.1,127.0.0.2,127.0.0.3/path/sub1/sub2", 0))
|
|
assert.Equal(t, service.fullpath(), "/path/sub1/sub2")
|
|
}
|
|
|
|
func TestCreateNodes(t *testing.T) {
|
|
service := &ZkDiscoveryService{}
|
|
|
|
nodes, err := service.createNodes(nil)
|
|
assert.Equal(t, nodes, []*discovery.Node{})
|
|
assert.NoError(t, err)
|
|
|
|
nodes, err = service.createNodes([]string{"127.0.0.1:2375", "127.0.0.2:2375"})
|
|
assert.Equal(t, nodes[0].String(), "127.0.0.1:2375")
|
|
assert.Equal(t, nodes[1].String(), "127.0.0.2:2375")
|
|
assert.NoError(t, err)
|
|
|
|
_, err = service.createNodes([]string{"127.0.0.1", "127.0.0.2"})
|
|
assert.Error(t, err)
|
|
}
|