Merge pull request #1260 from fenggw-fnst/work

test: add unit test case for core.go

Author: lowzj

dfget/core/core_test.go

@@ -27,6 +27,7 @@ import (
 	"path/filepath"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/dragonflyoss/Dragonfly/dfget/config"
 	. "github.com/dragonflyoss/Dragonfly/dfget/core/helper"
@@ -139,6 +140,24 @@ func (s *CoreTestSuite) TestCheckConnectSupernode(c *check.C) {
 	c.Assert(ip, check.Equals, "")
 }
 
+func (s *CoreTestSuite) TestCalculateTimeout(c *check.C) {
+	cfg := s.createConfig(&bytes.Buffer{})
+
+	timeout := calculateTimeout(nil)
+	c.Assert(timeout, check.Equals, config.DefaultDownloadTimeout)
+
+	timeout = calculateTimeout(cfg)
+	c.Assert(timeout, check.Equals, config.DefaultDownloadTimeout)
+
+	cfg.RV.FileLength = 1000
+	timeout = calculateTimeout(cfg)
+	c.Assert(timeout, check.Equals, 10*time.Second)
+
+	cfg.Timeout = time.Minute
+	timeout = calculateTimeout(cfg)
+	c.Assert(timeout, check.Equals, time.Minute)
+}
+
 // ----------------------------------------------------------------------------
 // helper functions