Skip to content
This repository has been archived by the owner on Feb 27, 2023. It is now read-only.

Commit

Permalink
pkg/syncmap: add unit test for GetAsAtomicInt
Browse files Browse the repository at this point in the history
Signed-off-by: Zhou Hao <zhouhao@cn.fujitsu.com>
  • Loading branch information
Zhou Hao committed Feb 14, 2020
1 parent 790e620 commit 11ce918
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions pkg/syncmap/syncmap_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ import (
"testing"
"time"

"github.com/dragonflyoss/Dragonfly/pkg/atomiccount"

"github.com/go-check/check"
"github.com/willf/bitset"
)
Expand Down Expand Up @@ -131,3 +133,16 @@ 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)

result, err := mmap.GetAsAtomicInt("nonexist")
c.Check(err, check.NotNil)
c.Check(result, check.IsNil)
}

0 comments on commit 11ce918

Please sign in to comment.