add a few tests

Signed-off-by: Victor Vieux <vieux@docker.com>

discovery/discovery.go

@@ -34,8 +34,9 @@ type DiscoveryService interface {
 }
 
 var (
-	discoveries     map[string]func() DiscoveryService
+	discoveries       map[string]func() DiscoveryService
-	ErrNotSupported = errors.New("discovery service not supported")
+	ErrNotSupported   = errors.New("discovery service not supported")
+	ErrNotImplemented = errors.New("not implemented in this discovery service")
 )
 
 func init() {

discovery/etcd/etcd_test.go

@@ -0,0 +1,16 @@
+package etcd
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInitialize(t *testing.T) {
+	discovery := &EtcdDiscoveryService{}
+	assert.Error(t, discovery.Initialize("127.0.0.1/path", 0))
+	assert.Equal(t, discovery.path, "/path/")
+
+	assert.Error(t, discovery.Initialize("127.0.0.1,127.0.0.2,127.0.0.3/path", 0))
+	assert.Equal(t, discovery.path, "/path/")
+}

discovery/file/file.go

@@ -1,7 +1,6 @@
 package file
 
 import (
-	"errors"
 	"io/ioutil"
 	"strings"
 	"time"
@@ -54,5 +53,5 @@ func (s *FileDiscoveryService) Watch(callback discovery.WatchCallback) {
 }
 
 func (s *FileDiscoveryService) Register(addr string) error {
-	return errors.New("unimplemented")
+	return discovery.ErrNotImplemented
 }

discovery/file/file_test.go

@@ -0,0 +1,18 @@
+package file
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInitialize(t *testing.T) {
+	discovery := &FileDiscoveryService{}
+	discovery.Initialize("/path/to/file", 0)
+	assert.Equal(t, discovery.path, "/path/to/file")
+}
+
+func TestRegister(t *testing.T) {
+	discovery := &FileDiscoveryService{path: "/path/to/file"}
+	assert.Error(t, discovery.Register("0.0.0.0"))
+}

discovery/token/token_test.go

@@ -6,7 +6,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestInit(t *testing.T) {
+func TestInitialize(t *testing.T) {
 	discovery := &TokenDiscoveryService{}
 	discovery.Initialize("token", 0)
 	assert.Equal(t, discovery.token, "token")
@@ -18,7 +18,7 @@ func TestInit(t *testing.T) {
 }
 
 func TestRegister(t *testing.T) {
-	discovery := TokenDiscoveryService{token: "TEST_TOKEN", url: DISCOVERY_URL}
+	discovery := &TokenDiscoveryService{token: "TEST_TOKEN", url: DISCOVERY_URL}
 	expected := "127.0.0.1:2675"
 	assert.NoError(t, discovery.Register(expected))