generator_test.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. package discovery
  2. import (
  3. "github.com/go-check/check"
  4. )
  5. func (s *DiscoverySuite) TestGeneratorNotGenerate(c *check.C) {
  6. ips := Generate("127.0.0.1")
  7. c.Assert(len(ips), check.Equals, 1)
  8. c.Assert(ips[0], check.Equals, "127.0.0.1")
  9. }
  10. func (s *DiscoverySuite) TestGeneratorWithPortNotGenerate(c *check.C) {
  11. ips := Generate("127.0.0.1:8080")
  12. c.Assert(len(ips), check.Equals, 1)
  13. c.Assert(ips[0], check.Equals, "127.0.0.1:8080")
  14. }
  15. func (s *DiscoverySuite) TestGeneratorMatchFailedNotGenerate(c *check.C) {
  16. ips := Generate("127.0.0.[1]")
  17. c.Assert(len(ips), check.Equals, 1)
  18. c.Assert(ips[0], check.Equals, "127.0.0.[1]")
  19. }
  20. func (s *DiscoverySuite) TestGeneratorWithPort(c *check.C) {
  21. ips := Generate("127.0.0.[1:11]:2375")
  22. c.Assert(len(ips), check.Equals, 11)
  23. c.Assert(ips[0], check.Equals, "127.0.0.1:2375")
  24. c.Assert(ips[1], check.Equals, "127.0.0.2:2375")
  25. c.Assert(ips[2], check.Equals, "127.0.0.3:2375")
  26. c.Assert(ips[3], check.Equals, "127.0.0.4:2375")
  27. c.Assert(ips[4], check.Equals, "127.0.0.5:2375")
  28. c.Assert(ips[5], check.Equals, "127.0.0.6:2375")
  29. c.Assert(ips[6], check.Equals, "127.0.0.7:2375")
  30. c.Assert(ips[7], check.Equals, "127.0.0.8:2375")
  31. c.Assert(ips[8], check.Equals, "127.0.0.9:2375")
  32. c.Assert(ips[9], check.Equals, "127.0.0.10:2375")
  33. c.Assert(ips[10], check.Equals, "127.0.0.11:2375")
  34. }
  35. func (s *DiscoverySuite) TestGenerateWithMalformedInputAtRangeStart(c *check.C) {
  36. malformedInput := "127.0.0.[x:11]:2375"
  37. ips := Generate(malformedInput)
  38. c.Assert(len(ips), check.Equals, 1)
  39. c.Assert(ips[0], check.Equals, malformedInput)
  40. }
  41. func (s *DiscoverySuite) TestGenerateWithMalformedInputAtRangeEnd(c *check.C) {
  42. malformedInput := "127.0.0.[1:x]:2375"
  43. ips := Generate(malformedInput)
  44. c.Assert(len(ips), check.Equals, 1)
  45. c.Assert(ips[0], check.Equals, malformedInput)
  46. }