test: add unit test case for func TransRate

Signed-off-by: Guangwen Feng <[email protected]>
dragonflyoss · Dec 30, 2019 · d1d7b0a · d1d7b0a
1 parent df32ce2
commit d1d7b0a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/pkg/ratelimiter/ratelimiter_test.go b/pkg/ratelimiter/ratelimiter_test.go
@@ -115,3 +115,20 @@ func (suite *RateLimiterSuite) TestRateLimiter_AcquireNonBlocking(c *check.C) {
 	rl.blocking(1000)
 	c.Assert(rl.AcquireNonBlocking(1000), check.Equals, int64(1000))
 }
+
+func (suite *RateLimiterSuite) TestTransRate(c *check.C) {
+	var cases = []struct {
+		r int64
+		e int64
+	}{
+		{666, 1000},
+		{2048, 3000},
+		{123456, 124000},
+		{0, 10486000},
+		{-233, 10486000},
+	}
+	for _, cc := range cases {
+		v := TransRate(cc.r)
+		c.Assert(v, check.Equals, cc.e)
+	}
+}