1.在controllers文件目录中建立文件base_test.go，代码如下
package controllers
import (
"fmt"
"github.com/astaxie/beego"
"github.com/astaxie/beego/context"
"net/http"
"net/url"
)
type testResponseWriter struct{}
func (f *testResponseWriter) Header() http.Header {
return http.Header{}
}
func (f *testResponseWriter) Write(b []byte) (int, error) {
//输出打印到终端
fmt.Println(string(b))
return 0, nil
}
func (f *testResponseWriter) WriteHeader(n int) {}
//构造控制器 BaseController是我项目的基础控制器，你可以换成你的，但必须继承自beego.Controller
func (c *BaseController) makeCtx() *BaseController {
c.Ctx = context.NewContext()
c.Ctx.Request = &http.Request{URL: &url.URL{Scheme: "http", Host: "localhost", Path: "/"}}
c.Ctx.ResponseWriter = &context.Response{ResponseWriter: &testResponseWriter{}, Started: true}
c.Ctx.Input.Context = c.Ctx
c.Ctx.Input.RequestBody = []byte{}
c.Ctx.Output.Context = c.Ctx
c.Data = map[interface{}]interface{}{}
return c
}
//阻止控制器中调用this.StopRun 而导致的panic
func recoverUserStop() {
if err := recover(); err == beego.ErrAbort {
return
} else {
panic(err)
}
}
2.调用示例
func TestClientController_List(t *testing.T) {
defer recoverUserStop()
this := new(ClientController)
this.makeCtx()
this.List()
}
}
