add unit test for Set

Signed-off-by: Zhou Hao <[email protected]>
dragonflyoss · Jan 14, 2020 · 733a6e3 · 733a6e3
1 parent b936211
commit 733a6e3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/atomiccount/atomiccount_test.go b/pkg/atomiccount/atomiccount_test.go
@@ -40,3 +40,13 @@ func (suite *AtomicCountUtilSuite) TestAdd(c *check.C) {
 	result := acount.Get()
 	c.Check(result, check.Equals, (int32)(12))
 }
+
+func (suite *AtomicCountUtilSuite) TestSet(c *check.C) {
+	acount := NewAtomicInt(3)
+	acount.Add(4)
+	acount.Add(5)
+
+	_ = acount.Set(1)
+	result := acount.Get()
+	c.Check(result, check.Equals, (int32)(1))
+}