Kyle Isom
b893e99864
A customised config is an ini file with a [default] section and some other name sections; a config file is loaded from the default section with any keys in the named section being added in, overriding keys in the host. This allows for, e.g. setting different paths based on the host name or operating system.
143 lines
3.1 KiB
Go
143 lines
3.1 KiB
Go
package iniconf
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
// FailWithError is a utility for dumping errors and failing the test.
|
|
func FailWithError(t *testing.T, err error) {
|
|
fmt.Println("failed")
|
|
if err != nil {
|
|
fmt.Println("[!] ", err.Error())
|
|
}
|
|
t.FailNow()
|
|
}
|
|
|
|
// UnlinkIfExists removes a file if it exists.
|
|
func UnlinkIfExists(file string) {
|
|
_, err := os.Stat(file)
|
|
if err != nil && os.IsNotExist(err) {
|
|
panic("failed to remove " + file)
|
|
}
|
|
os.Remove(file)
|
|
}
|
|
|
|
// stringSlicesEqual compares two string lists, checking that they
|
|
// contain the same elements.
|
|
func stringSlicesEqual(slice1, slice2 []string) bool {
|
|
if len(slice1) != len(slice2) {
|
|
return false
|
|
}
|
|
|
|
for i := range slice1 {
|
|
if slice1[i] != slice2[i] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
for i := range slice2 {
|
|
if slice1[i] != slice2[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestGoodConfig(t *testing.T) {
|
|
testFile := "testdata/test.conf"
|
|
fmt.Printf("[+] validating known-good config... ")
|
|
cmap, err := ParseFile(testFile)
|
|
if err != nil {
|
|
FailWithError(t, err)
|
|
} else if len(cmap) != 2 {
|
|
FailWithError(t, err)
|
|
}
|
|
fmt.Println("ok")
|
|
}
|
|
|
|
func TestGoodConfig2(t *testing.T) {
|
|
testFile := "testdata/test2.conf"
|
|
fmt.Printf("[+] validating second known-good config... ")
|
|
cmap, err := ParseFile(testFile)
|
|
if err != nil {
|
|
FailWithError(t, err)
|
|
} else if len(cmap) != 1 {
|
|
FailWithError(t, err)
|
|
} else if len(cmap["default"]) != 3 {
|
|
FailWithError(t, err)
|
|
}
|
|
fmt.Println("ok")
|
|
}
|
|
|
|
func TestBadConfig(t *testing.T) {
|
|
testFile := "testdata/bad.conf"
|
|
fmt.Printf("[+] ensure invalid config file fails... ")
|
|
_, err := ParseFile(testFile)
|
|
if err == nil {
|
|
err = fmt.Errorf("invalid config file should fail")
|
|
FailWithError(t, err)
|
|
}
|
|
fmt.Println("ok")
|
|
}
|
|
|
|
func TestWriteConfigFile(t *testing.T) {
|
|
fmt.Printf("[+] ensure config file is written properly... ")
|
|
const testFile = "testdata/test.conf"
|
|
const testOut = "testdata/test.out"
|
|
|
|
cmap, err := ParseFile(testFile)
|
|
if err != nil {
|
|
FailWithError(t, err)
|
|
}
|
|
|
|
defer UnlinkIfExists(testOut)
|
|
err = cmap.WriteFile(testOut)
|
|
if err != nil {
|
|
FailWithError(t, err)
|
|
}
|
|
|
|
cmap2, err := ParseFile(testOut)
|
|
if err != nil {
|
|
FailWithError(t, err)
|
|
}
|
|
|
|
sectionList1 := cmap.ListSections()
|
|
sectionList2 := cmap2.ListSections()
|
|
sort.Strings(sectionList1)
|
|
sort.Strings(sectionList2)
|
|
if !stringSlicesEqual(sectionList1, sectionList2) {
|
|
err = fmt.Errorf("section lists don't match")
|
|
FailWithError(t, err)
|
|
}
|
|
|
|
for _, section := range sectionList1 {
|
|
for _, k := range cmap[section] {
|
|
if cmap[section][k] != cmap2[section][k] {
|
|
err = fmt.Errorf("config key doesn't match")
|
|
FailWithError(t, err)
|
|
}
|
|
}
|
|
}
|
|
fmt.Println("ok")
|
|
}
|
|
|
|
func TestQuotedValue(t *testing.T) {
|
|
testFile := "testdata/test.conf"
|
|
fmt.Printf("[+] validating quoted value... ")
|
|
cmap, _ := ParseFile(testFile)
|
|
val := cmap["sectionName"]["key4"]
|
|
if val != " space at beginning and end " {
|
|
FailWithError(t, errors.New("Wrong value in double quotes ["+val+"]"))
|
|
}
|
|
|
|
val = cmap["sectionName"]["key5"]
|
|
if val != " is quoted with single quotes " {
|
|
FailWithError(t, errors.New("Wrong value in single quotes ["+val+"]"))
|
|
}
|
|
fmt.Println("ok")
|
|
}
|