supernode/config: add UT for IsSuperCID

supernode/config: add UT for GetSuperPID supernode/config: add UT for IsSuperPID Signed-off-by: Zhou Hao <[email protected]>
dragonflyoss · May 13, 2020 · 4ca7667 · 4ca7667
1 parent 003b73d
commit 4ca7667
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/supernode/config/config_test.go b/supernode/config/config_test.go
@@ -92,3 +92,32 @@ func (s *SupernodeConfigTestSuite) TestGetSuperCID(c *check.C) {
 
 	c.Assert(conf.GetSuperCID("taskID"), check.DeepEquals, "CIDPrefixtaskID")
 }
+
+func (s *SupernodeConfigTestSuite) TestIsSuperCID(c *check.C) {
+	conf := Config{
+		BaseProperties: &BaseProperties{cIDPrefix: "CIDPrefix"},
+	}
+
+	c.Assert(conf.IsSuperCID("CIDPrefixTest"), check.DeepEquals, true)
+	c.Assert(conf.IsSuperCID("Test"), check.DeepEquals, false)
+}
+
+func (s *SupernodeConfigTestSuite) TestGetSuperPID(c *check.C) {
+	conf := Config{
+		BaseProperties: &BaseProperties{superNodePID: "superNodePID"},
+	}
+
+	c.Assert(conf.GetSuperPID(), check.DeepEquals, "superNodePID")
+
+	conf.SetSuperPID("Test")
+	c.Assert(conf.GetSuperPID(), check.DeepEquals, "Test")
+}
+
+func (s *SupernodeConfigTestSuite) TestIsSuperPID(c *check.C) {
+	conf := Config{
+		BaseProperties: &BaseProperties{superNodePID: "superNodePID"},
+	}
+
+	c.Assert(conf.IsSuperPID("superNodePID"), check.DeepEquals, true)
+	c.Assert(conf.IsSuperPID("Test"), check.DeepEquals, false)
+}