123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137 |
- package discovery
- import (
- "testing"
- "github.com/go-check/check"
- )
- // Hook up gocheck into the "go test" runner.
- func Test(t *testing.T) { check.TestingT(t) }
- type DiscoverySuite struct{}
- var _ = check.Suite(&DiscoverySuite{})
- func (s *DiscoverySuite) TestNewEntry(c *check.C) {
- entry, err := NewEntry("127.0.0.1:2375")
- c.Assert(err, check.IsNil)
- c.Assert(entry.Equals(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)
- c.Assert(entry.String(), check.Equals, "127.0.0.1:2375")
- entry, err = NewEntry("[2001:db8:0:f101::2]:2375")
- c.Assert(err, check.IsNil)
- c.Assert(entry.Equals(&Entry{Host: "2001:db8:0:f101::2", Port: "2375"}), check.Equals, true)
- c.Assert(entry.String(), check.Equals, "[2001:db8:0:f101::2]:2375")
- _, err = NewEntry("127.0.0.1")
- c.Assert(err, check.NotNil)
- }
- func (s *DiscoverySuite) TestParse(c *check.C) {
- scheme, uri := parse("127.0.0.1:2375")
- c.Assert(scheme, check.Equals, "nodes")
- c.Assert(uri, check.Equals, "127.0.0.1:2375")
- scheme, uri = parse("localhost:2375")
- c.Assert(scheme, check.Equals, "nodes")
- c.Assert(uri, check.Equals, "localhost:2375")
- scheme, uri = parse("scheme://127.0.0.1:2375")
- c.Assert(scheme, check.Equals, "scheme")
- c.Assert(uri, check.Equals, "127.0.0.1:2375")
- scheme, uri = parse("scheme://localhost:2375")
- c.Assert(scheme, check.Equals, "scheme")
- c.Assert(uri, check.Equals, "localhost:2375")
- scheme, uri = parse("")
- c.Assert(scheme, check.Equals, "nodes")
- c.Assert(uri, check.Equals, "")
- }
- func (s *DiscoverySuite) TestCreateEntries(c *check.C) {
- entries, err := CreateEntries(nil)
- c.Assert(entries, check.DeepEquals, Entries{})
- c.Assert(err, check.IsNil)
- entries, err = CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", "[2001:db8:0:f101::2]:2375", ""})
- c.Assert(err, check.IsNil)
- expected := Entries{
- &Entry{Host: "127.0.0.1", Port: "2375"},
- &Entry{Host: "127.0.0.2", Port: "2375"},
- &Entry{Host: "2001:db8:0:f101::2", Port: "2375"},
- }
- c.Assert(entries.Equals(expected), check.Equals, true)
- _, err = CreateEntries([]string{"127.0.0.1", "127.0.0.2"})
- c.Assert(err, check.NotNil)
- }
- func (s *DiscoverySuite) TestContainsEntry(c *check.C) {
- entries, err := CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", ""})
- c.Assert(err, check.IsNil)
- c.Assert(entries.Contains(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)
- c.Assert(entries.Contains(&Entry{Host: "127.0.0.3", Port: "2375"}), check.Equals, false)
- }
- func (s *DiscoverySuite) TestEntriesEquality(c *check.C) {
- entries := Entries{
- &Entry{Host: "127.0.0.1", Port: "2375"},
- &Entry{Host: "127.0.0.2", Port: "2375"},
- }
- // Same
- c.Assert(entries.Equals(Entries{
- &Entry{Host: "127.0.0.1", Port: "2375"},
- &Entry{Host: "127.0.0.2", Port: "2375"},
- }), check.
- Equals, true)
- // Different size
- c.Assert(entries.Equals(Entries{
- &Entry{Host: "127.0.0.1", Port: "2375"},
- &Entry{Host: "127.0.0.2", Port: "2375"},
- &Entry{Host: "127.0.0.3", Port: "2375"},
- }), check.
- Equals, false)
- // Different content
- c.Assert(entries.Equals(Entries{
- &Entry{Host: "127.0.0.1", Port: "2375"},
- &Entry{Host: "127.0.0.42", Port: "2375"},
- }), check.
- Equals, false)
- }
- func (s *DiscoverySuite) TestEntriesDiff(c *check.C) {
- entry1 := &Entry{Host: "1.1.1.1", Port: "1111"}
- entry2 := &Entry{Host: "2.2.2.2", Port: "2222"}
- entry3 := &Entry{Host: "3.3.3.3", Port: "3333"}
- entries := Entries{entry1, entry2}
- // No diff
- added, removed := entries.Diff(Entries{entry2, entry1})
- c.Assert(added, check.HasLen, 0)
- c.Assert(removed, check.HasLen, 0)
- // Add
- added, removed = entries.Diff(Entries{entry2, entry3, entry1})
- c.Assert(added, check.HasLen, 1)
- c.Assert(added.Contains(entry3), check.Equals, true)
- c.Assert(removed, check.HasLen, 0)
- // Remove
- added, removed = entries.Diff(Entries{entry2})
- c.Assert(added, check.HasLen, 0)
- c.Assert(removed, check.HasLen, 1)
- c.Assert(removed.Contains(entry1), check.Equals, true)
- // Add and remove
- added, removed = entries.Diff(Entries{entry1, entry3})
- c.Assert(added, check.HasLen, 1)
- c.Assert(added.Contains(entry3), check.Equals, true)
- c.Assert(removed, check.HasLen, 1)
- c.Assert(removed.Contains(entry2), check.Equals, true)
- }
|