polish the code and add unit test

modules/gittar/auth/auth.go

@@ -216,8 +216,8 @@ func doAuth(c *webcontext.Context, repo *models.Repo, repoName string) {
 			return
 		}
 		logrus.Infof("repo: %s userId: %v, username: %s", repoName, userIdStr, userInfoDto.Username)
-		//校验通过缓存5分钟结果
-		//校验失败每次都会请求
+
+		// if success, caches the results for 5 minutes
 		_, validateError := ValidaUserRepoWithCache(c, userIdStr, repo)
 		if validateError != nil {
 			logrus.Infof("openapi auth fail repo:%s user:%s", repoName, userInfoDto.Username)

modules/gittar/auth/auth_test.go

@@ -0,0 +1,88 @@
+// Copyright (c) 2021 Terminus, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package auth
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"bou.ke/monkey"
+	"github.com/labstack/echo"
+
+	"github.com/erda-project/erda/modules/gittar/models"
+	"github.com/erda-project/erda/modules/gittar/pkg/gitmodule"
+	"github.com/erda-project/erda/modules/gittar/webcontext"
+)
+
+func TestIsGitProtocolRequest(t *testing.T) {
+	e := echo.New()
+	req1 := httptest.NewRequest(http.MethodGet, "/", nil)
+	req2 := httptest.NewRequest(http.MethodGet, "/", nil)
+	req1.Header.Add("Git-Protocol", "version")
+	res := httptest.NewRecorder()
+	ctx1 := &webcontext.Context{
+		EchoContext: e.NewContext(req1, res),
+		Repository:  nil,
+		User:        nil,
+		Service:     nil,
+		DBClient:    nil,
+		Bundle:      nil,
+		UCAuth:      nil,
+		EtcdClient:  nil,
+	}
+	ctx2 := &webcontext.Context{
+		EchoContext: e.NewContext(req2, res),
+		Repository:  nil,
+		User:        nil,
+		Service:     nil,
+		DBClient:    nil,
+		Bundle:      nil,
+		UCAuth:      nil,
+		EtcdClient:  nil,
+	}
+
+	if isGitProtocolRequest(ctx1) != true {
+		t.Error("fail")
+	}
+	if isGitProtocolRequest(ctx2) == true {
+		t.Error("fail")
+	}
+}
+
+func TestDoAuthWithHttpProtocolWithoutUserID(t *testing.T) {
+	e := echo.New()
+	req := httptest.NewRequest(http.MethodGet, "/", nil)
+	res := httptest.NewRecorder()
+	ctx := &webcontext.Context{
+		EchoContext: e.NewContext(req, res),
+		Repository:  nil,
+		User:        nil,
+		Service:     nil,
+		DBClient:    nil,
+		Bundle:      nil,
+		UCAuth:      nil,
+		EtcdClient:  nil,
+	}
+	monkey.Patch(openRepository, func(ctx *webcontext.Context, repo *models.Repo) (*gitmodule.Repository, error) {
+		return nil, nil
+	})
+	defer monkey.UnpatchAll()
+
+	doAuth(ctx, nil, "")
+	if ctx.EchoContext.Response().Status == 500 {
+		t.Error("fail")
+	}
+}

modules/gittar/initialize.go

@@ -139,6 +139,9 @@ func (p *provider) Initialize() error {
 	// cron task to git gc all repository
 	go gc.ScheduledExecuteClean()
 
+	// start hook task consumer
+	models.Init(dbClient)
+
 	return e.Start(":" + conf.ListenPort())
 }
 

modules/gittar/models/web_hook.go

@@ -120,11 +120,7 @@ func (svc *Service) CreateHookTask(task *WebHookTask) error {
 
 }
 
-func init() {
-	db, err := OpenDB()
-	if err != nil {
-		panic(err)
-	}
+func Init(db *DBClient) {
 	for i := 0; i < 5; i += 1 {
 		go StartHookTaskConsumer(db)
 	}