1
0
镜像自地址 https://github.com/tuna/tunasync.git 已同步 2025-12-06 22:46:47 +00:00

tests(manager): add tests for server.go, validate workerID in middleware

这个提交包含在:
walkerning
2016-04-25 10:47:29 +08:00
提交者 bigeagle
父节点 02bb8c16ab
当前提交 401b6a694e
共有 3 个文件被更改,包括 190 次插入42 次删除

查看文件

@@ -1,6 +1,9 @@
package manager
import (
"fmt"
"net/http"
"github.com/gin-gonic/gin"
)
@@ -14,3 +17,17 @@ func contextErrorLogger(c *gin.Context) {
// pass on to the next middleware in chain
c.Next()
}
func (s *managerServer) workerIDValidator(c *gin.Context) {
workerID := c.Param("id")
_, err := s.adapter.GetWorker(workerID)
if err != nil {
// no worker named `workerID` exists
err := fmt.Errorf("invalid workerID %s", workerID)
s.returnErrJSON(c, http.StatusBadRequest, err)
c.Abort()
return
}
// pass on to the next middleware in chain
c.Next()
}