mirror of https://github.com/docker/docs.git
32 lines
1004 B
Go
32 lines
1004 B
Go
package nodes
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestInitialise(t *testing.T) {
|
|
discovery := &DiscoveryService{}
|
|
discovery.Initialize("1.1.1.1:1111,2.2.2.2:2222", 0)
|
|
assert.Equal(t, len(discovery.entries), 2)
|
|
assert.Equal(t, discovery.entries[0].String(), "1.1.1.1:1111")
|
|
assert.Equal(t, discovery.entries[1].String(), "2.2.2.2:2222")
|
|
}
|
|
|
|
func TestInitialiseWithPattern(t *testing.T) {
|
|
discovery := &DiscoveryService{}
|
|
discovery.Initialize("1.1.1.[1:2]:1111,2.2.2.[2:4]:2222", 0)
|
|
assert.Equal(t, len(discovery.entries), 5)
|
|
assert.Equal(t, discovery.entries[0].String(), "1.1.1.1:1111")
|
|
assert.Equal(t, discovery.entries[1].String(), "1.1.1.2:1111")
|
|
assert.Equal(t, discovery.entries[2].String(), "2.2.2.2:2222")
|
|
assert.Equal(t, discovery.entries[3].String(), "2.2.2.3:2222")
|
|
assert.Equal(t, discovery.entries[4].String(), "2.2.2.4:2222")
|
|
}
|
|
|
|
func TestRegister(t *testing.T) {
|
|
discovery := &DiscoveryService{}
|
|
assert.Error(t, discovery.Register("0.0.0.0"))
|
|
}
|