discovery_test.go 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
  1. package discovery
  2. import (
  3. "testing"
  4. "github.com/go-check/check"
  5. )
  6. // Hook up gocheck into the "go test" runner.
  7. func Test(t *testing.T) { check.TestingT(t) }
  8. type DiscoverySuite struct{}
  9. var _ = check.Suite(&DiscoverySuite{})
  10. func (s *DiscoverySuite) TestNewEntry(c *check.C) {
  11. entry, err := NewEntry("127.0.0.1:2375")
  12. c.Assert(err, check.IsNil)
  13. c.Assert(entry.Equals(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)
  14. c.Assert(entry.String(), check.Equals, "127.0.0.1:2375")
  15. entry, err = NewEntry("[2001:db8:0:f101::2]:2375")
  16. c.Assert(err, check.IsNil)
  17. c.Assert(entry.Equals(&Entry{Host: "2001:db8:0:f101::2", Port: "2375"}), check.Equals, true)
  18. c.Assert(entry.String(), check.Equals, "[2001:db8:0:f101::2]:2375")
  19. _, err = NewEntry("127.0.0.1")
  20. c.Assert(err, check.NotNil)
  21. }
  22. func (s *DiscoverySuite) TestParse(c *check.C) {
  23. scheme, uri := parse("127.0.0.1:2375")
  24. c.Assert(scheme, check.Equals, "nodes")
  25. c.Assert(uri, check.Equals, "127.0.0.1:2375")
  26. scheme, uri = parse("localhost:2375")
  27. c.Assert(scheme, check.Equals, "nodes")
  28. c.Assert(uri, check.Equals, "localhost:2375")
  29. scheme, uri = parse("scheme://127.0.0.1:2375")
  30. c.Assert(scheme, check.Equals, "scheme")
  31. c.Assert(uri, check.Equals, "127.0.0.1:2375")
  32. scheme, uri = parse("scheme://localhost:2375")
  33. c.Assert(scheme, check.Equals, "scheme")
  34. c.Assert(uri, check.Equals, "localhost:2375")
  35. scheme, uri = parse("")
  36. c.Assert(scheme, check.Equals, "nodes")
  37. c.Assert(uri, check.Equals, "")
  38. }
  39. func (s *DiscoverySuite) TestCreateEntries(c *check.C) {
  40. entries, err := CreateEntries(nil)
  41. c.Assert(entries, check.DeepEquals, Entries{})
  42. c.Assert(err, check.IsNil)
  43. entries, err = CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", "[2001:db8:0:f101::2]:2375", ""})
  44. c.Assert(err, check.IsNil)
  45. expected := Entries{
  46. &Entry{Host: "127.0.0.1", Port: "2375"},
  47. &Entry{Host: "127.0.0.2", Port: "2375"},
  48. &Entry{Host: "2001:db8:0:f101::2", Port: "2375"},
  49. }
  50. c.Assert(entries.Equals(expected), check.Equals, true)
  51. _, err = CreateEntries([]string{"127.0.0.1", "127.0.0.2"})
  52. c.Assert(err, check.NotNil)
  53. }
  54. func (s *DiscoverySuite) TestContainsEntry(c *check.C) {
  55. entries, err := CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", ""})
  56. c.Assert(err, check.IsNil)
  57. c.Assert(entries.Contains(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)
  58. c.Assert(entries.Contains(&Entry{Host: "127.0.0.3", Port: "2375"}), check.Equals, false)
  59. }
  60. func (s *DiscoverySuite) TestEntriesEquality(c *check.C) {
  61. entries := Entries{
  62. &Entry{Host: "127.0.0.1", Port: "2375"},
  63. &Entry{Host: "127.0.0.2", Port: "2375"},
  64. }
  65. // Same
  66. c.Assert(entries.Equals(Entries{
  67. &Entry{Host: "127.0.0.1", Port: "2375"},
  68. &Entry{Host: "127.0.0.2", Port: "2375"},
  69. }), check.
  70. Equals, true)
  71. // Different size
  72. c.Assert(entries.Equals(Entries{
  73. &Entry{Host: "127.0.0.1", Port: "2375"},
  74. &Entry{Host: "127.0.0.2", Port: "2375"},
  75. &Entry{Host: "127.0.0.3", Port: "2375"},
  76. }), check.
  77. Equals, false)
  78. // Different content
  79. c.Assert(entries.Equals(Entries{
  80. &Entry{Host: "127.0.0.1", Port: "2375"},
  81. &Entry{Host: "127.0.0.42", Port: "2375"},
  82. }), check.
  83. Equals, false)
  84. }
  85. func (s *DiscoverySuite) TestEntriesDiff(c *check.C) {
  86. entry1 := &Entry{Host: "1.1.1.1", Port: "1111"}
  87. entry2 := &Entry{Host: "2.2.2.2", Port: "2222"}
  88. entry3 := &Entry{Host: "3.3.3.3", Port: "3333"}
  89. entries := Entries{entry1, entry2}
  90. // No diff
  91. added, removed := entries.Diff(Entries{entry2, entry1})
  92. c.Assert(added, check.HasLen, 0)
  93. c.Assert(removed, check.HasLen, 0)
  94. // Add
  95. added, removed = entries.Diff(Entries{entry2, entry3, entry1})
  96. c.Assert(added, check.HasLen, 1)
  97. c.Assert(added.Contains(entry3), check.Equals, true)
  98. c.Assert(removed, check.HasLen, 0)
  99. // Remove
  100. added, removed = entries.Diff(Entries{entry2})
  101. c.Assert(added, check.HasLen, 0)
  102. c.Assert(removed, check.HasLen, 1)
  103. c.Assert(removed.Contains(entry1), check.Equals, true)
  104. // Add and remove
  105. added, removed = entries.Diff(Entries{entry1, entry3})
  106. c.Assert(added, check.HasLen, 1)
  107. c.Assert(added.Contains(entry3), check.Equals, true)
  108. c.Assert(removed, check.HasLen, 1)
  109. c.Assert(removed.Contains(entry2), check.Equals, true)
  110. }