mirror of
https://github.com/astaxie/beego.git
synced 2024-07-05 08:55:10 +00:00
67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
var yamlcontext = `
|
|
"appname": beeapi
|
|
"httpport": 8080
|
|
"mysqlport": 3600
|
|
"PI": 3.1415976
|
|
"runmode": dev
|
|
"autorender": false
|
|
"copyrequestbody": true
|
|
`
|
|
|
|
func TestYaml(t *testing.T) {
|
|
f, err := os.Create("testyaml.conf")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = f.WriteString(yamlcontext)
|
|
if err != nil {
|
|
f.Close()
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
defer os.Remove("testyaml.conf")
|
|
yamlconf, err := NewConfig("yaml", "testyaml.conf")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if yamlconf.String("appname") != "beeapi" {
|
|
t.Fatal("appname not equal to beeapi")
|
|
}
|
|
if port, err := yamlconf.Int("httpport"); err != nil || port != 8080 {
|
|
t.Error(port)
|
|
t.Fatal(err)
|
|
}
|
|
if port, err := yamlconf.Int64("mysqlport"); err != nil || port != 3600 {
|
|
t.Error(port)
|
|
t.Fatal(err)
|
|
}
|
|
if pi, err := yamlconf.Float("PI"); err != nil || pi != 3.1415976 {
|
|
t.Error(pi)
|
|
t.Fatal(err)
|
|
}
|
|
if yamlconf.String("runmode") != "dev" {
|
|
t.Fatal("runmode not equal to dev")
|
|
}
|
|
if v, err := yamlconf.Bool("autorender"); err != nil || v != false {
|
|
t.Error(v)
|
|
t.Fatal(err)
|
|
}
|
|
if v, err := yamlconf.Bool("copyrequestbody"); err != nil || v != true {
|
|
t.Error(v)
|
|
t.Fatal(err)
|
|
}
|
|
if err = yamlconf.Set("name", "astaxie"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if yamlconf.String("name") != "astaxie" {
|
|
t.Fatal("get name error")
|
|
}
|
|
}
|