fix(cloudevents): fix UT cases

Signed-off-by: wuhuizuo <wuhuizuo@126.com>
PingCAP-QE · Dec 26, 2023 · e1b2391 · e1b2391
1 parent d0148db
commit e1b2391
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/cloudevents-server/pkg/events/custom/tekton/handler_pipelinerun_test.go b/cloudevents-server/pkg/events/custom/tekton/handler_pipelinerun_test.go
@@ -50,6 +50,7 @@ func Test_pipelineRunHandler_Handle(t *testing.T) {
 
 	h := &pipelineRunHandler{
 		LarkClient:       lark.NewClient(larkAppID, larkAppSecret, lark.WithLogReqAtDebug(true), lark.WithEnableTokenCache(true)),
+		Receiver:         receiver,
 		RunDetailBaseURL: baseURL,
 	}
 	for _, tt := range tests {

diff --git a/cloudevents-server/pkg/events/custom/tekton/handler_taskrun_test.go b/cloudevents-server/pkg/events/custom/tekton/handler_taskrun_test.go
@@ -46,6 +46,7 @@ func Test_taskRunHandler_Handle(t *testing.T) {
 
 	h := &taskRunHandler{
 		LarkClient:       lark.NewClient(larkAppID, larkAppSecret, lark.WithLogReqAtDebug(true), lark.WithEnableTokenCache(true)),
+		Receiver:         receiver,
 		RunDetailBaseURL: baseURL,
 	}
 	for _, tt := range tests {

diff --git a/cloudevents-server/pkg/events/custom/tekton/handler_test.go b/cloudevents-server/pkg/events/custom/tekton/handler_test.go
@@ -1,8 +1,10 @@
 package tekton
 
-const (
-	larkAppID     = "<Chagne me>"
-	larkAppSecret = "<Chagne me>"
-	receiver      = "<Change me>@xxx.com"
-	baseURL       = "https://chagne.me.com/"
+import "os"
+
+var (
+	larkAppID     = os.Getenv("LARK_APP_ID")
+	larkAppSecret = os.Getenv("LARK_APP_SECRET")
+	receiver      = os.Getenv("LARK_RECEIVER")
+	baseURL       = os.Getenv("LINK_BASE_URL")
 )