bug fix

bootstrap/bootstrap.go

@@ -2,6 +2,7 @@ package bootstrap
 
 import (
 	"spaceofcheng/services/models"
+	"spaceofcheng/services/routers"
 	"spaceofcheng/services/service/events"
 	"spaceofcheng/services/utils"
 )
@@ -22,4 +23,7 @@ func SystemInit() {
 
 	// 事件系统
 	events.EventInit()
+
+	// 路由系统
+	routers.RouteInit()
 }

main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"spaceofcheng/services/bootstrap"
-	_ "spaceofcheng/services/routers"
 
 	"github.com/astaxie/beego"
 )

routers/router.go

@@ -14,7 +14,7 @@ import (
 	"github.com/astaxie/beego"
 )
 
-func init() {
+func RouteInit() {
 	appRoot := utils.GetAppRoot()
 
 	appConf, _ := utils.GetConfiger(appRoot + "/conf/app.conf")

tests/utils.go

@@ -12,6 +12,7 @@ import (
 	"net/http/httptest"
 	"net/url"
 	"os"
+	"spaceofcheng/services/utils"
 	"strconv"
 	"strings"
 	"time"
@@ -49,6 +50,13 @@ func (t *RequestMock) AddWechatUA() *RequestMock {
 	return t
 }
 
+// AddToken 添加Token
+func (t *RequestMock) AddToken(dt map[string]interface{}) *RequestMock {
+	token, _ := utils.CreateToken(dt)
+	t.Headers[utils.TokenHeader] = utils.TokenSchema + " " + token
+	return t
+}
+
 // SetHost 设置 host 包含 port
 func (t *RequestMock) SetHost(host string) *RequestMock {
 	t.Host = host

utils/utils.go

@@ -4,8 +4,8 @@ import (
 	"encoding/base64"
 	"math/rand"
 	"net/http"
+	"os"
 	"path/filepath"
-	"runtime"
 	"strconv"
 	"strings"
 	"time"
@@ -161,7 +161,6 @@ func GetFiveSeconds(t time.Time) string {
 
 // GetAppRoot 获取系统根目录
 func GetAppRoot() string {
-	_, file, _, _ := runtime.Caller(1)
-	appRoot, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".."+string(filepath.Separator))))
+	appRoot, _ := filepath.Abs(filepath.Dir(os.Args[0]))
 	return appRoot
 }