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 TestCreateEntries(t *testing.T) {
	service := &ZkDiscoveryService{}

	entries, err := service.createEntries(nil)
	assert.Equal(t, entries, []*discovery.Entry{})
	assert.NoError(t, err)

	entries, err = service.createEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375"})
	assert.Equal(t, entries[0].String(), "127.0.0.1:2375")
	assert.Equal(t, entries[1].String(), "127.0.0.2:2375")
	assert.NoError(t, err)

	_, err = service.createEntries([]string{"127.0.0.1", "127.0.0.2"})
	assert.Error(t, err)
}