Merge pull request #1262 from fenggw-fnst/work

test: add unit test case for func ReportMetrics

dfget/core/api/supernode_api_test.go

@@ -21,6 +21,7 @@ import (
 	"strings"
 	"testing"
 
+	api_types "github.com/dragonflyoss/Dragonfly/apis/types"
 	"github.com/dragonflyoss/Dragonfly/dfget/types"
 	"github.com/dragonflyoss/Dragonfly/pkg/constants"
 	"github.com/dragonflyoss/Dragonfly/pkg/httputils"
@@ -123,6 +124,26 @@ func (s *SupernodeAPITestSuite) TestSupernodeAPI_ReportClientError(c *check.C) {
 	c.Check(r.Code, check.Equals, 700)
 }
 
+func (s *SupernodeAPITestSuite) TestSupernodeAPI_ReportMetrics(c *check.C) {
+	s.mock.PostJSONFunc = s.mock.CreatePostJSONFunc(0, nil, nil)
+	r, e := s.api.ReportMetrics(localhost, &api_types.TaskMetricsRequest{})
+	c.Assert(r, check.IsNil)
+	c.Assert(e.Error(), check.Equals, "0:")
+
+	s.mock.PostJSONFunc = s.mock.CreatePostJSONFunc(0, nil,
+		fmt.Errorf("test"))
+	r, e = s.api.ReportMetrics(localhost, &api_types.TaskMetricsRequest{})
+	c.Assert(r, check.IsNil)
+	c.Assert(e.Error(), check.Equals, "test")
+
+	res := types.RegisterResponse{BaseResponse: &types.BaseResponse{}}
+	s.mock.PostJSONFunc = s.mock.CreatePostJSONFunc(200, []byte(res.String()), nil)
+	r, e = s.api.ReportMetrics(localhost, &api_types.TaskMetricsRequest{})
+	c.Assert(e, check.IsNil)
+	c.Assert(r, check.NotNil)
+	c.Assert(r.Code, check.Equals, 0)
+}
+
 func (s *SupernodeAPITestSuite) TestSupernodeAPI_get(c *check.C) {
 	type testRes struct {
 		A int