test: add unit test case for static_locator.go

Signed-off-by: Guangwen Feng <[email protected]>
dragonflyoss · Jun 2, 2020 · 3e51fda · 3e51fda
1 parent 5981693
commit 3e51fda
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/dfget/locator/static_locator_test.go b/dfget/locator/static_locator_test.go
@@ -170,6 +170,24 @@ func (s *StaticLocatorTestSuite) Test_Refresh(c *check.C) {
 	c.Assert(l.load(), check.Equals, -1)
 }
 
+func (s *StaticLocatorTestSuite) Test_String(c *check.C) {
+	cases := []struct {
+		locator  *StaticLocator
+		expected string
+	}{
+		{createLocator("a:80=1"), "empty"},
+		{createLocator("a:80=1"), "empty"},
+		{createLocator("a:80=1"), "test-group:[a:80=1]"},
+	}
+	cases[0].locator.Group = nil
+	cases[1].locator.inc()
+	cases[1].locator.inc()
+
+	for _, v := range cases {
+		c.Assert(v.locator.String(), check.Equals, v.expected)
+	}
+}
+
 func create(ip string, port, weight int) *Supernode {
 	return &Supernode{
 		Schema:    config.DefaultSupernodeSchema,