mirror of
https://github.com/astaxie/beego.git
synced 2024-06-26 02:34:14 +00:00
cd31c816cc
get environment variable if config item has prefix "$ENV_" . e.g. ```ini [demo] password = $ENV_MyPWD ```
115 lines
2.7 KiB
Go
115 lines
2.7 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 xml
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/astaxie/beego/config"
|
|
)
|
|
|
|
func TestXML(t *testing.T) {
|
|
|
|
//xml parse should incluce in <config></config> tags
|
|
var xmlcontext = `<?xml version="1.0" encoding="UTF-8"?>
|
|
<config>
|
|
<appname>beeapi</appname>
|
|
<httpport>8080</httpport>
|
|
<mysqlport>3600</mysqlport>
|
|
<PI>3.1415976</PI>
|
|
<runmode>dev</runmode>
|
|
<autorender>false</autorender>
|
|
<copyrequestbody>true</copyrequestbody>
|
|
<path>$ENV_GOROOT</path>
|
|
<dbinfo>
|
|
<db>beego</db>
|
|
<pwd>$ENV_GOROOT</pwd>
|
|
<url>localhost</url>
|
|
<detail>
|
|
<d1>value1</d1>
|
|
<d2>$ENV_GOROOT</d2>
|
|
<d3></d3>
|
|
</detail>
|
|
<group>
|
|
<id>001</id>
|
|
<name>gp2</name>
|
|
</group>
|
|
</dbinfo>
|
|
</config>
|
|
`
|
|
|
|
f, err := os.Create("testxml.conf")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = f.WriteString(xmlcontext)
|
|
if err != nil {
|
|
f.Close()
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
defer os.Remove("testxml.conf")
|
|
xmlconf, err := config.NewConfig("xml", "testxml.conf")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if xmlconf.String("appname") != "beeapi" {
|
|
t.Fatal("appname not equal to beeapi")
|
|
}
|
|
if port, err := xmlconf.Int("httpport"); err != nil || port != 8080 {
|
|
t.Error(port)
|
|
t.Fatal(err)
|
|
}
|
|
if port, err := xmlconf.Int64("mysqlport"); err != nil || port != 3600 {
|
|
t.Error(port)
|
|
t.Fatal(err)
|
|
}
|
|
if pi, err := xmlconf.Float("PI"); err != nil || pi != 3.1415976 {
|
|
t.Error(pi)
|
|
t.Fatal(err)
|
|
}
|
|
if xmlconf.String("runmode") != "dev" {
|
|
t.Fatal("runmode not equal to dev")
|
|
}
|
|
if v, err := xmlconf.Bool("autorender"); err != nil || v != false {
|
|
t.Error(v)
|
|
t.Fatal(err)
|
|
}
|
|
if v, err := xmlconf.Bool("copyrequestbody"); err != nil || v != true {
|
|
t.Error(v)
|
|
t.Fatal(err)
|
|
}
|
|
if err = xmlconf.Set("name", "astaxie"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if xmlconf.String("name") != "astaxie" {
|
|
t.Fatal("get name error")
|
|
}
|
|
|
|
if xmlconf.String("path") == os.Getenv("GOROOT") {
|
|
t.Fatal("get path error")
|
|
}
|
|
|
|
if dbinfo, err := xmlconf.GetSection("dbinfo"); err != nil {
|
|
t.Fatal(err)
|
|
} else if dbinfo["pwd"] != os.Getenv("GOROOT") {
|
|
t.Fatal("get pwd error")
|
|
} else if strings.Contains(dbinfo["detail"], os.Getenv("GOROOT")) == false {
|
|
t.Fatal("get goroot path error")
|
|
}
|
|
}
|