pkg/syncmap: add unit test for GetAsAtomicInt

Signed-off-by: Zhou Hao <[email protected]>
dragonflyoss · Feb 12, 2020 · 31003ca · 31003ca
1 parent 790e620
commit 31003ca
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/syncmap/syncmap_test.go b/pkg/syncmap/syncmap_test.go
@@ -21,6 +21,8 @@ import (
 	"testing"
 	"time"
 
+	"github.com/dragonflyoss/Dragonfly/pkg/atomiccount"
+
 	"github.com/go-check/check"
 	"github.com/willf/bitset"
 )
@@ -131,3 +133,12 @@ func (suite *SyncMapUtilSuite) TestGetAsTime(c *check.C) {
 	result, _ := mmap.GetAsTime("aaa")
 	c.Check(result, check.DeepEquals, expected)
 }
+
+func (suite *SyncMapUtilSuite) TestGetAsAtomicInt(c *check.C) {
+	expected := atomiccount.NewAtomicInt(10)
+	mmap := NewSyncMap()
+	mmap.Add("aaa", expected)
+
+	result, _ := mmap.GetAsAtomicInt("aaa")
+	c.Check(result, check.DeepEquals, expected)
+}