mirror of
https://github.com/astaxie/beego.git
synced 2024-12-23 08:50:50 +00:00
172 lines
4.6 KiB
Go
172 lines
4.6 KiB
Go
// Copyright 2014 beego Author. All Rights Reserved.
|
|
//
|
|
// 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 beego
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/astaxie/beego/context"
|
|
)
|
|
|
|
func TestNamespaceGet(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/user", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Get("/user", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte("v1_user"))
|
|
})
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "v1_user" {
|
|
t.Errorf("TestNamespaceGet can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespacePost(t *testing.T) {
|
|
r, _ := http.NewRequest("POST", "/v1/user/123", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Post("/user/:id", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte(ctx.Input.Param(":id")))
|
|
})
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "123" {
|
|
t.Errorf("TestNamespacePost can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespaceNest(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/admin/order", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Namespace(
|
|
NewNamespace("/admin").
|
|
Get("/order", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte("order"))
|
|
}),
|
|
)
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "order" {
|
|
t.Errorf("TestNamespaceNest can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespaceNestParam(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/admin/order/123", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Namespace(
|
|
NewNamespace("/admin").
|
|
Get("/order/:id", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte(ctx.Input.Param(":id")))
|
|
}),
|
|
)
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "123" {
|
|
t.Errorf("TestNamespaceNestParam can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespaceRouter(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/api/list", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Router("/api/list", &TestController{}, "*:List")
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "i am list" {
|
|
t.Errorf("TestNamespaceRouter can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespaceAutoFunc(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/test/list", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.AutoRouter(&TestController{})
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "i am list" {
|
|
t.Errorf("user define func can't run")
|
|
}
|
|
}
|
|
|
|
func TestNamespaceFilter(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v1/user/123", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v1")
|
|
ns.Filter("before", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte("this is Filter"))
|
|
}).
|
|
Get("/user/:id", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte(ctx.Input.Param(":id")))
|
|
})
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "this is Filter" {
|
|
t.Errorf("TestNamespaceFilter can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestNamespaceCond(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v2/test/list", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
ns := NewNamespace("/v2")
|
|
ns.Cond(func(ctx *context.Context) bool {
|
|
if ctx.Input.Domain() == "beego.me" {
|
|
return true
|
|
}
|
|
return false
|
|
}).
|
|
AutoRouter(&TestController{})
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Code != 405 {
|
|
t.Errorf("TestNamespaceCond can't run get the result " + strconv.Itoa(w.Code))
|
|
}
|
|
}
|
|
|
|
func TestNamespaceInside(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "/v3/shop/order/123", nil)
|
|
w := httptest.NewRecorder()
|
|
ns := NewNamespace("/v3",
|
|
NSAutoRouter(&TestController{}),
|
|
NSNamespace("/shop",
|
|
NSGet("/order/:id", func(ctx *context.Context) {
|
|
ctx.Output.Body([]byte(ctx.Input.Param(":id")))
|
|
}),
|
|
),
|
|
)
|
|
AddNamespace(ns)
|
|
BeeApp.Handlers.ServeHTTP(w, r)
|
|
if w.Body.String() != "123" {
|
|
t.Errorf("TestNamespaceInside can't run, get the response is " + w.Body.String())
|
|
}
|
|
}
|