toml_test.go

// Testing support for go-toml

package toml

import (
	"reflect"
	"testing"
)

func TestTomlHas(t *testing.T) {
	tree, _ := Load(`
		[test]
		key = "value"
	`)

	if !tree.Has("test.key") {
		t.Errorf("Has - expected test.key to exists")
	}

	if tree.Has("") {
		t.Errorf("Should return false if the key is not provided")
	}
}

func TestTomlGet(t *testing.T) {
	tree, _ := Load(`
		[test]
		key = "value"
	`)

	if tree.Get("") != tree {
		t.Errorf("Get should return the tree itself when given an empty path")
	}

	if tree.Get("test.key") != "value" {
		t.Errorf("Get should return the value")
	}
	if tree.Get(`\`) != nil {
		t.Errorf("should return nil when the key is malformed")
	}
}

func TestTomlGetArray(t *testing.T) {
	tree, _ := Load(`
		[test]
		key = ["one", "two"]
		key2 = [true, false, false]
		key3 = [1.5,2.5]
	`)

	if tree.GetArray("") != tree {
		t.Errorf("GetArray should return the tree itself when given an empty path")
	}

	expect := []string{"one", "two"}
	actual := tree.GetArray("test.key").([]string)
	if !reflect.DeepEqual(actual, expect) {
		t.Errorf("GetArray should return the []string value")
	}

	expect2 := []bool{true, false, false}
	actual2 := tree.GetArray("test.key2").([]bool)
	if !reflect.DeepEqual(actual2, expect2) {
		t.Errorf("GetArray should return the []bool value")
	}

	expect3 := []float64{1.5, 2.5}
	actual3 := tree.GetArray("test.key3").([]float64)
	if !reflect.DeepEqual(actual3, expect3) {
		t.Errorf("GetArray should return the []float64 value")
	}

	if tree.GetArray(`\`) != nil {
		t.Errorf("should return nil when the key is malformed")
	}
}

func TestTomlGetDefault(t *testing.T) {
	tree, _ := Load(`
		[test]
		key = "value"
	`)

	if tree.GetDefault("", "hello") != tree {
		t.Error("GetDefault should return the tree itself when given an empty path")
	}

	if tree.GetDefault("test.key", "hello") != "value" {
		t.Error("Get should return the value")
	}

	if tree.GetDefault("whatever", "hello") != "hello" {
		t.Error("GetDefault should return the default value if the key does not exist")
	}
}

func TestTomlHasPath(t *testing.T) {
	tree, _ := Load(`
		[test]
		key = "value"
	`)

	if !tree.HasPath([]string{"test", "key"}) {
		t.Errorf("HasPath - expected test.key to exists")
	}
}

func TestTomlDelete(t *testing.T) {
	tree, _ := Load(`
        key = "value"
    `)
	err := tree.Delete("key")
	if err != nil {
		t.Errorf("Delete - unexpected error while deleting key: %s", err.Error())
	}

	if tree.Get("key") != nil {
		t.Errorf("Delete should have removed key but did not.")
	}

}

func TestTomlDeleteUnparsableKey(t *testing.T) {
	tree, _ := Load(`
        key = "value"
    `)
	err := tree.Delete(".")
	if err == nil {
		t.Errorf("Delete should error")
	}
}

func TestTomlDeleteNestedKey(t *testing.T) {
	tree, _ := Load(`
		[foo]
        [foo.bar]
        key = "value"
    `)
	err := tree.Delete("foo.bar.key")
	if err != nil {
		t.Errorf("Error while deleting nested key: %s", err.Error())
	}

	if tree.Get("key") != nil {
		t.Errorf("Delete should have removed nested key but did not.")
	}

}

func TestTomlDeleteNonexistentNestedKey(t *testing.T) {
	tree, _ := Load(`
		[foo]
        [foo.bar]
        key = "value"
    `)
	err := tree.Delete("foo.not.there.key")
	if err == nil {
		t.Errorf("Delete should have thrown an error trying to delete key in nonexistent tree")
	}
}

func TestTomlGetPath(t *testing.T) {
	node := newTree()
	//TODO: set other node data

	for idx, item := range []struct {
		Path     []string
		Expected *Tree
	}{
		{ // empty path test
			[]string{},
			node,
		},
	} {
		result := node.GetPath(item.Path)
		if result != item.Expected {
			t.Errorf("GetPath[%d] %v - expected %v, got %v instead.", idx, item.Path, item.Expected, result)
		}
	}

	tree, _ := Load("[foo.bar]\na=1\nb=2\n[baz.foo]\na=3\nb=4\n[gorf.foo]\na=5\nb=6")
	if tree.GetPath([]string{"whatever"}) != nil {
		t.Error("GetPath should return nil when the key does not exist")
	}
}

func TestTomlGetArrayPath(t *testing.T) {
	for idx, item := range []struct {
		Name string
		Path []string
		Make func() (tree *Tree, expected interface{})
	}{
		{
			Name: "empty",
			Path: []string{},
			Make: func() (tree *Tree, expected interface{}) {
				tree = newTree()
				expected = tree
				return
			},
		},
		{
			Name: "int64",
			Path: []string{"a"},
			Make: func() (tree *Tree, expected interface{}) {
				var err error
				tree, err = Load(`a = [1,2,3]`)
				if err != nil {
					panic(err)
				}
				expected = []int64{1, 2, 3}
				return
			},
		},
	} {
		t.Run(item.Name, func(t *testing.T) {
			tree, expected := item.Make()
			result := tree.GetArrayPath(item.Path)
			if !reflect.DeepEqual(result, expected) {
				t.Errorf("GetArrayPath[%d] %v - expected %#v, got %#v instead.", idx, item.Path, expected, result)
			}
		})
	}

	tree, _ := Load("[foo.bar]\na=1\nb=2\n[baz.foo]\na=3\nb=4\n[gorf.foo]\na=5\nb=6")
	if tree.GetArrayPath([]string{"whatever"}) != nil {
		t.Error("GetArrayPath should return nil when the key does not exist")
	}

}

func TestTomlFromMap(t *testing.T) {
	simpleMap := map[string]interface{}{"hello": 42}
	tree, err := TreeFromMap(simpleMap)
	if err != nil {
		t.Fatal("unexpected error:", err)
	}
	if tree.Get("hello") != int64(42) {
		t.Fatal("hello should be 42, not", tree.Get("hello"))
	}
}

func TestLoadBytesBOM(t *testing.T) {
	payloads := [][]byte{
		[]byte("\xFE\xFFhello=1"),
		[]byte("\xFF\xFEhello=1"),
		[]byte("\xEF\xBB\xBFhello=1"),
		[]byte("\x00\x00\xFE\xFFhello=1"),
		[]byte("\xFF\xFE\x00\x00hello=1"),
	}
	for _, data := range payloads {
		tree, err := LoadBytes(data)
		if err != nil {
			t.Fatal("unexpected error:", err, "for:", data)
		}
		v := tree.Get("hello")
		if v != int64(1) {
			t.Fatal("hello should be 1, not", v)
		}
	}
}