218 lines
6.9 KiB
Go
218 lines
6.9 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 context
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestBind(t *testing.T) {
|
|
type testItem struct {
|
|
field string
|
|
empty interface{}
|
|
want interface{}
|
|
}
|
|
type Human struct {
|
|
ID int
|
|
Nick string
|
|
Pwd string
|
|
Ms bool
|
|
}
|
|
|
|
cases := []struct {
|
|
request string
|
|
valueGp []testItem
|
|
}{
|
|
{"/?p=str", []testItem{{"p", interface{}(""), interface{}("str")}}},
|
|
|
|
{"/?p=", []testItem{{"p", "", ""}}},
|
|
{"/?p=str", []testItem{{"p", "", "str"}}},
|
|
|
|
{"/?p=123", []testItem{{"p", 0, 123}}},
|
|
{"/?p=123", []testItem{{"p", uint(0), uint(123)}}},
|
|
|
|
{"/?p=1.0", []testItem{{"p", 0.0, 1.0}}},
|
|
{"/?p=1", []testItem{{"p", false, true}}},
|
|
|
|
{"/?p=true", []testItem{{"p", false, true}}},
|
|
{"/?p=ON", []testItem{{"p", false, true}}},
|
|
{"/?p=on", []testItem{{"p", false, true}}},
|
|
{"/?p=1", []testItem{{"p", false, true}}},
|
|
{"/?p=2", []testItem{{"p", false, false}}},
|
|
{"/?p=false", []testItem{{"p", false, false}}},
|
|
|
|
{"/?p[a]=1&p[b]=2&p[c]=3", []testItem{{"p", map[string]int{}, map[string]int{"a": 1, "b": 2, "c": 3}}}},
|
|
{"/?p[a]=v1&p[b]=v2&p[c]=v3", []testItem{{"p", map[string]string{}, map[string]string{"a": "v1", "b": "v2", "c": "v3"}}}},
|
|
|
|
{"/?p[]=8&p[]=9&p[]=10", []testItem{{"p", []int{}, []int{8, 9, 10}}}},
|
|
{"/?p[0]=8&p[1]=9&p[2]=10", []testItem{{"p", []int{}, []int{8, 9, 10}}}},
|
|
{"/?p[0]=8&p[1]=9&p[2]=10&p[5]=14", []testItem{{"p", []int{}, []int{8, 9, 10, 0, 0, 14}}}},
|
|
{"/?p[0]=8.0&p[1]=9.0&p[2]=10.0", []testItem{{"p", []float64{}, []float64{8.0, 9.0, 10.0}}}},
|
|
|
|
{"/?p[]=10&p[]=9&p[]=8", []testItem{{"p", []string{}, []string{"10", "9", "8"}}}},
|
|
{"/?p[0]=8&p[1]=9&p[2]=10", []testItem{{"p", []string{}, []string{"8", "9", "10"}}}},
|
|
|
|
{"/?p[0]=true&p[1]=false&p[2]=true&p[5]=1&p[6]=ON&p[7]=other", []testItem{{"p", []bool{}, []bool{true, false, true, false, false, true, true, false}}}},
|
|
|
|
{"/?human.Nick=astaxie", []testItem{{"human", Human{}, Human{Nick: "astaxie"}}}},
|
|
{"/?human.ID=888&human.Nick=astaxie&human.Ms=true&human[Pwd]=pass", []testItem{{"human", Human{}, Human{ID: 888, Nick: "astaxie", Ms: true, Pwd: "pass"}}}},
|
|
{"/?human[0].ID=888&human[0].Nick=astaxie&human[0].Ms=true&human[0][Pwd]=pass01&human[1].ID=999&human[1].Nick=ysqi&human[1].Ms=On&human[1].Pwd=pass02",
|
|
[]testItem{{"human", []Human{}, []Human{
|
|
{ID: 888, Nick: "astaxie", Ms: true, Pwd: "pass01"},
|
|
{ID: 999, Nick: "ysqi", Ms: true, Pwd: "pass02"},
|
|
}}}},
|
|
|
|
{
|
|
"/?id=123&isok=true&ft=1.2&ol[0]=1&ol[1]=2&ul[]=str&ul[]=array&human.Nick=astaxie",
|
|
[]testItem{
|
|
{"id", 0, 123},
|
|
{"isok", false, true},
|
|
{"ft", 0.0, 1.2},
|
|
{"ol", []int{}, []int{1, 2}},
|
|
{"ul", []string{}, []string{"str", "array"}},
|
|
{"human", Human{}, Human{Nick: "astaxie"}},
|
|
},
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
r, _ := http.NewRequest("GET", c.request, nil)
|
|
beegoInput := NewInput()
|
|
beegoInput.Context = NewContext()
|
|
beegoInput.Context.Reset(httptest.NewRecorder(), r)
|
|
|
|
for _, item := range c.valueGp {
|
|
got := item.empty
|
|
err := beegoInput.Bind(&got, item.field)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(got, item.want) {
|
|
t.Fatalf("Bind %q error,should be:\n%#v \ngot:\n%#v", item.field, item.want, got)
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func TestSubDomain(t *testing.T) {
|
|
r, _ := http.NewRequest("GET", "http://www.example.com/?id=123&isok=true&ft=1.2&ol[0]=1&ol[1]=2&ul[]=str&ul[]=array&user.Name=astaxie", nil)
|
|
beegoInput := NewInput()
|
|
beegoInput.Context = NewContext()
|
|
beegoInput.Context.Reset(httptest.NewRecorder(), r)
|
|
|
|
subdomain := beegoInput.SubDomains()
|
|
if subdomain != "www" {
|
|
t.Fatal("Subdomain parse error, got" + subdomain)
|
|
}
|
|
|
|
r, _ = http.NewRequest("GET", "http://localhost/", nil)
|
|
beegoInput.Context.Request = r
|
|
if beegoInput.SubDomains() != "" {
|
|
t.Fatal("Subdomain parse error, should be empty, got " + beegoInput.SubDomains())
|
|
}
|
|
|
|
r, _ = http.NewRequest("GET", "http://aa.bb.example.com/", nil)
|
|
beegoInput.Context.Request = r
|
|
if beegoInput.SubDomains() != "aa.bb" {
|
|
t.Fatal("Subdomain parse error, got " + beegoInput.SubDomains())
|
|
}
|
|
|
|
/* TODO Fix this
|
|
r, _ = http.NewRequest("GET", "http://127.0.0.1/", nil)
|
|
beegoInput.Context.Request = r
|
|
if beegoInput.SubDomains() != "" {
|
|
t.Fatal("Subdomain parse error, got " + beegoInput.SubDomains())
|
|
}
|
|
*/
|
|
|
|
r, _ = http.NewRequest("GET", "http://example.com/", nil)
|
|
beegoInput.Context.Request = r
|
|
if beegoInput.SubDomains() != "" {
|
|
t.Fatal("Subdomain parse error, got " + beegoInput.SubDomains())
|
|
}
|
|
|
|
r, _ = http.NewRequest("GET", "http://aa.bb.cc.dd.example.com/", nil)
|
|
beegoInput.Context.Request = r
|
|
if beegoInput.SubDomains() != "aa.bb.cc.dd" {
|
|
t.Fatal("Subdomain parse error, got " + beegoInput.SubDomains())
|
|
}
|
|
}
|
|
|
|
func TestParams(t *testing.T) {
|
|
inp := NewInput()
|
|
|
|
inp.SetParam("p1", "val1_ver1")
|
|
inp.SetParam("p2", "val2_ver1")
|
|
inp.SetParam("p3", "val3_ver1")
|
|
if l := inp.ParamsLen(); l != 3 {
|
|
t.Fatalf("Input.ParamsLen wrong value: %d, expected %d", l, 3)
|
|
}
|
|
|
|
if val := inp.Param("p1"); val != "val1_ver1" {
|
|
t.Fatalf("Input.Param wrong value: %s, expected %s", val, "val1_ver1")
|
|
}
|
|
if val := inp.Param("p3"); val != "val3_ver1" {
|
|
t.Fatalf("Input.Param wrong value: %s, expected %s", val, "val3_ver1")
|
|
}
|
|
vals := inp.Params()
|
|
expected := map[string]string{
|
|
"p1": "val1_ver1",
|
|
"p2": "val2_ver1",
|
|
"p3": "val3_ver1",
|
|
}
|
|
if !reflect.DeepEqual(vals, expected) {
|
|
t.Fatalf("Input.Params wrong value: %s, expected %s", vals, expected)
|
|
}
|
|
|
|
// overwriting existing params
|
|
inp.SetParam("p1", "val1_ver2")
|
|
inp.SetParam("p2", "val2_ver2")
|
|
expected = map[string]string{
|
|
"p1": "val1_ver2",
|
|
"p2": "val2_ver2",
|
|
"p3": "val3_ver1",
|
|
}
|
|
vals = inp.Params()
|
|
if !reflect.DeepEqual(vals, expected) {
|
|
t.Fatalf("Input.Params wrong value: %s, expected %s", vals, expected)
|
|
}
|
|
|
|
if l := inp.ParamsLen(); l != 3 {
|
|
t.Fatalf("Input.ParamsLen wrong value: %d, expected %d", l, 3)
|
|
}
|
|
|
|
if val := inp.Param("p1"); val != "val1_ver2" {
|
|
t.Fatalf("Input.Param wrong value: %s, expected %s", val, "val1_ver2")
|
|
}
|
|
|
|
if val := inp.Param("p2"); val != "val2_ver2" {
|
|
t.Fatalf("Input.Param wrong value: %s, expected %s", val, "val1_ver2")
|
|
}
|
|
|
|
}
|
|
func BenchmarkQuery(b *testing.B) {
|
|
beegoInput := NewInput()
|
|
beegoInput.Context = NewContext()
|
|
beegoInput.Context.Request, _ = http.NewRequest("POST", "http://www.example.com/?q=foo", nil)
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
beegoInput.Query("q")
|
|
}
|
|
})
|
|
}
|