我有一个使用Gin框架构建的Go API。
阅读这里测试部分的文档,我尝试实现类似的东西:
main.go
package main
import (
"mes/routes"
"github.com/gin-gonic/gin"
)
func setupMainRoutes(engine *gin.Engine) {
engine.GET("/mesg/:language/services", routes.AllServices)
engine.GET("/mesg/:language/service/:id", routes.OneService)
engine.GET("/mesg/:language/services/search", routes.SearchService)
}
func setupErrorRoutes(engine *gin.Engine) {
engine.NoRoute(routes.Error404Handler)
}
func setupServer() *gin.Engine {
// Gin Mode
gin.SetMode(gin.ReleaseMode)
// Creates the Gin Engine
engine := gin.New()
// Setup the API Routes
setupMainRoutes(engine)
// Setup Error Routes
setupErrorRoutes(engine)
// Return engine
return engine
}
func main() {
// Run the engine
setupServer().Run()
}
main_test.go
package main
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
/************** Error Handling Tests **************/
func TestPing404Errors1Of3(t *testing.T) {
router := setupServer()
w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/", nil)
router.ServeHTTP(w, req)
assert.Equal(t, 404, w.Code)
}
但是,我似乎无法将main_test.go
移动到另一个文件夹,因为setupServer()
函数变得未定义。
有没有一种方法可以将我的所有测试分类到一个子文件夹中?
将main_test.go
移动到另一个文件夹意味着您将从不同的包测试/调用setupServer()
。为此,您需要导出为SetupServer()
,并导入测试文件中包含SetupServer()
的包。不幸的是,您无法对当前的文件结构执行此操作,因为我们无法导入package main
。因此,如果你真的想这样做,你需要重命名包。
话虽如此,将测试文件_test.go
放在同一个目录(因此是同一个包(中是Go中的常见做法。