1package nodes // import "github.com/docker/docker/pkg/discovery/nodes"
2
3import (
4	"testing"
5
6	"github.com/docker/docker/internal/test/suite"
7	"github.com/docker/docker/pkg/discovery"
8	"gotest.tools/v3/assert"
9)
10
11// Hook up gocheck into the "go test" runner.
12func Test(t *testing.T) {
13	suite.Run(t, &DiscoverySuite{})
14}
15
16type DiscoverySuite struct{}
17
18func (s *DiscoverySuite) TestInitialize(c *testing.T) {
19	d := &Discovery{}
20	d.Initialize("1.1.1.1:1111,2.2.2.2:2222", 0, 0, nil)
21	assert.Equal(c, len(d.entries), 2)
22	assert.Equal(c, d.entries[0].String(), "1.1.1.1:1111")
23	assert.Equal(c, d.entries[1].String(), "2.2.2.2:2222")
24}
25
26func (s *DiscoverySuite) TestInitializeWithPattern(c *testing.T) {
27	d := &Discovery{}
28	d.Initialize("1.1.1.[1:2]:1111,2.2.2.[2:4]:2222", 0, 0, nil)
29	assert.Equal(c, len(d.entries), 5)
30	assert.Equal(c, d.entries[0].String(), "1.1.1.1:1111")
31	assert.Equal(c, d.entries[1].String(), "1.1.1.2:1111")
32	assert.Equal(c, d.entries[2].String(), "2.2.2.2:2222")
33	assert.Equal(c, d.entries[3].String(), "2.2.2.3:2222")
34	assert.Equal(c, d.entries[4].String(), "2.2.2.4:2222")
35}
36
37func (s *DiscoverySuite) TestWatch(c *testing.T) {
38	d := &Discovery{}
39	d.Initialize("1.1.1.1:1111,2.2.2.2:2222", 0, 0, nil)
40	expected := discovery.Entries{
41		&discovery.Entry{Host: "1.1.1.1", Port: "1111"},
42		&discovery.Entry{Host: "2.2.2.2", Port: "2222"},
43	}
44	ch, _ := d.Watch(nil)
45	assert.Equal(c, expected.Equals(<-ch), true)
46}
47
48func (s *DiscoverySuite) TestRegister(c *testing.T) {
49	d := &Discovery{}
50	assert.Assert(c, d.Register("0.0.0.0") != nil)
51}
52