feat: collector 单测调整 --story=119054819 (#474)

TencentBlueKing · Aug 6, 2024 · 2b45548 · 2b45548
1 parent d975f20
commit 2b45548
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pkg/collector/.gitignore b/pkg/collector/.gitignore
@@ -21,4 +21,5 @@ bk-collector
 logs/
 data/
 coverage.out
+coverage.tmp
 build/
diff --git a/pkg/collector/control.sh b/pkg/collector/control.sh
@@ -17,7 +17,8 @@ MODULE=bk-collector
 TEST_COVERAGE_THRESHOLD=75
 
 function unittest() {
-  go test ./... -coverprofile coverage.out -covermode count
+  go test ./... -coverprofile coverage.tmp -covermode count
+  cat coverage.tmp | grep -v '/gen/' > coverage.out
   go tool cover -func coverage.out
   echo "Quality Gate: checking test coverage is above threshold ..."
   echo "Threshold             : $TEST_COVERAGE_THRESHOLD%"

diff --git a/pkg/collector/receiver/pyroscope/http.go b/pkg/collector/receiver/pyroscope/http.go
@@ -29,10 +29,9 @@ import (
 	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/collector/internal/utils"
 	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/collector/pipeline"
 	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/collector/receiver"
-	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/utils/logger"
-
 	pushv1 "github.com/TencentBlueKing/bkmonitor-datalink/pkg/collector/receiver/pyroscope/gen/proto/go/push/v1"
 	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/collector/receiver/pyroscope/gen/proto/go/push/v1/pushv1connect"
+	"github.com/TencentBlueKing/bkmonitor-datalink/pkg/utils/logger"
 )
 
 const (
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,4 +21,5 @@ bk-collector @@
     logs/
     data/
     coverage.out
+    coverage.tmp
     build/