mirror of
https://github.com/fatedier/frp.git
synced 2024-12-14 18:28:59 +08:00
182 lines
4.5 KiB
Go
182 lines
4.5 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 (
|
|
"math"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/astaxie/beego/context"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
func TestGetInt(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", "40")
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetInt("age")
|
|
if val != 40 {
|
|
t.Errorf("TestGetInt expect 40,get %T,%v", val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetInt8(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", "40")
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetInt8("age")
|
|
if val != 40 {
|
|
t.Errorf("TestGetInt8 expect 40,get %T,%v", val, val)
|
|
}
|
|
//Output: int8
|
|
}
|
|
|
|
func TestGetInt16(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", "40")
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetInt16("age")
|
|
if val != 40 {
|
|
t.Errorf("TestGetInt16 expect 40,get %T,%v", val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetInt32(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", "40")
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetInt32("age")
|
|
if val != 40 {
|
|
t.Errorf("TestGetInt32 expect 40,get %T,%v", val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetInt64(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", "40")
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetInt64("age")
|
|
if val != 40 {
|
|
t.Errorf("TestGeetInt64 expect 40,get %T,%v", val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetUint8(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", strconv.FormatUint(math.MaxUint8, 10))
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetUint8("age")
|
|
if val != math.MaxUint8 {
|
|
t.Errorf("TestGetUint8 expect %v,get %T,%v", math.MaxUint8, val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetUint16(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", strconv.FormatUint(math.MaxUint16, 10))
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetUint16("age")
|
|
if val != math.MaxUint16 {
|
|
t.Errorf("TestGetUint16 expect %v,get %T,%v", math.MaxUint16, val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetUint32(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", strconv.FormatUint(math.MaxUint32, 10))
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetUint32("age")
|
|
if val != math.MaxUint32 {
|
|
t.Errorf("TestGetUint32 expect %v,get %T,%v", math.MaxUint32, val, val)
|
|
}
|
|
}
|
|
|
|
func TestGetUint64(t *testing.T) {
|
|
i := context.NewInput()
|
|
i.SetParam("age", strconv.FormatUint(math.MaxUint64, 10))
|
|
ctx := &context.Context{Input: i}
|
|
ctrlr := Controller{Ctx: ctx}
|
|
val, _ := ctrlr.GetUint64("age")
|
|
if val != math.MaxUint64 {
|
|
t.Errorf("TestGetUint64 expect %v,get %T,%v", uint64(math.MaxUint64), val, val)
|
|
}
|
|
}
|
|
|
|
func TestAdditionalViewPaths(t *testing.T) {
|
|
dir1 := "_beeTmp"
|
|
dir2 := "_beeTmp2"
|
|
defer os.RemoveAll(dir1)
|
|
defer os.RemoveAll(dir2)
|
|
|
|
dir1file := "file1.tpl"
|
|
dir2file := "file2.tpl"
|
|
|
|
genFile := func(dir string, name string, content string) {
|
|
os.MkdirAll(filepath.Dir(filepath.Join(dir, name)), 0777)
|
|
if f, err := os.Create(filepath.Join(dir, name)); err != nil {
|
|
t.Fatal(err)
|
|
} else {
|
|
defer f.Close()
|
|
f.WriteString(content)
|
|
f.Close()
|
|
}
|
|
|
|
}
|
|
genFile(dir1, dir1file, `<div>{{.Content}}</div>`)
|
|
genFile(dir2, dir2file, `<html>{{.Content}}</html>`)
|
|
|
|
AddViewPath(dir1)
|
|
AddViewPath(dir2)
|
|
|
|
ctrl := Controller{
|
|
TplName: "file1.tpl",
|
|
ViewPath: dir1,
|
|
}
|
|
ctrl.Data = map[interface{}]interface{}{
|
|
"Content": "value2",
|
|
}
|
|
if result, err := ctrl.RenderString(); err != nil {
|
|
t.Fatal(err)
|
|
} else {
|
|
if result != "<div>value2</div>" {
|
|
t.Fatalf("TestAdditionalViewPaths expect %s got %s", "<div>value2</div>", result)
|
|
}
|
|
}
|
|
|
|
func() {
|
|
ctrl.TplName = "file2.tpl"
|
|
defer func() {
|
|
if r := recover(); r == nil {
|
|
t.Fatal("TestAdditionalViewPaths expected error")
|
|
}
|
|
}()
|
|
ctrl.RenderString();
|
|
}()
|
|
|
|
ctrl.TplName = "file2.tpl"
|
|
ctrl.ViewPath = dir2
|
|
ctrl.RenderString();
|
|
}
|