test: add unit test case for func Remove

Signed-off-by: Hu Shuai <[email protected]>
dragonflyoss · Jan 9, 2020 · a021b8a · a021b8a
1 parent 7894da1
commit a021b8a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pkg/syncmap/syncmap_test.go b/pkg/syncmap/syncmap_test.go
@@ -54,3 +54,16 @@ func (suite *SyncMapUtilSuite) TestListKeyAsIntSlice(c *check.C) {
 	sort.Ints(result)
 	c.Check(result, check.DeepEquals, []int{1, 2})
 }
+
+func (suite *SyncMapUtilSuite) TestRemove(c *check.C) {
+	mmap := NewSyncMap()
+	mmap.Add("aaa", true)
+	mmap.Add("bbb", true)
+	mmap.Add("ccc", true)
+
+	mmap.Remove("ccc")
+
+	result := mmap.ListKeyAsStringSlice()
+	sort.Strings(result)
+	c.Check(result, check.DeepEquals, []string{"aaa", "bbb"})
+}