Thomas Pelletier
191 lines
2.9 KiB
Go
191 lines
2.9 KiB
Go
package toml
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUnmarshalSimple(t *testing.T) {
|
|
x := struct{ Foo string }{}
|
|
err := Unmarshal([]byte(`Foo = "hello"`), &x)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "hello", x.Foo)
|
|
}
|
|
|
|
func TestUnmarshalNestedStructs(t *testing.T) {
|
|
x := struct{ Foo struct{ Bar string } }{}
|
|
err := Unmarshal([]byte(`Foo.Bar = "hello"`), &x)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "hello", x.Foo.Bar)
|
|
}
|
|
|
|
func TestUnmarshalNestedStructsMultipleExpressions(t *testing.T) {
|
|
x := struct {
|
|
A struct{ B string }
|
|
C string
|
|
}{}
|
|
err := Unmarshal([]byte(`A.B = "hello"
|
|
C = "test"`), &x)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "hello", x.A.B)
|
|
assert.Equal(t, "test", x.C)
|
|
}
|
|
|
|
func TestUnmarshalTable(t *testing.T) {
|
|
x := struct {
|
|
Foo struct {
|
|
A string
|
|
B string
|
|
C string
|
|
}
|
|
Bar struct {
|
|
D string
|
|
}
|
|
E string
|
|
}{}
|
|
err := Unmarshal([]byte(`
|
|
|
|
E = "E"
|
|
Foo.C = "C"
|
|
|
|
[Foo]
|
|
A = "A"
|
|
B = 'B'
|
|
|
|
[Bar]
|
|
D = "D"
|
|
|
|
`), &x)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "A", x.Foo.A)
|
|
assert.Equal(t, "B", x.Foo.B)
|
|
assert.Equal(t, "C", x.Foo.C)
|
|
assert.Equal(t, "D", x.Bar.D)
|
|
assert.Equal(t, "E", x.E)
|
|
}
|
|
|
|
func TestUnmarshalDoesNotEraseBaseStruct(t *testing.T) {
|
|
x := struct {
|
|
A string
|
|
B string
|
|
}{
|
|
A: "preset",
|
|
}
|
|
err := Unmarshal([]byte(`B = "data"`), &x)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "preset", x.A)
|
|
assert.Equal(t, "data", x.B)
|
|
}
|
|
|
|
func TestArrayTableSimple(t *testing.T) {
|
|
doc := `
|
|
[[Products]]
|
|
Name = "Hammer"
|
|
|
|
[[Products]]
|
|
Name = "Nail"
|
|
`
|
|
|
|
type Product struct {
|
|
Name string
|
|
}
|
|
|
|
type Data struct {
|
|
Products []Product
|
|
}
|
|
|
|
x := Data{}
|
|
err := Unmarshal([]byte(doc), &x)
|
|
|
|
require.NoError(t, err)
|
|
|
|
expected := Data{
|
|
Products: []Product{
|
|
{
|
|
Name: "Hammer",
|
|
},
|
|
{
|
|
Name: "Nail",
|
|
},
|
|
},
|
|
}
|
|
|
|
assert.Equal(t, expected, x)
|
|
}
|
|
|
|
//func TestUnmarshalArrayTablesMultiple(t *testing.T) {
|
|
// doc := `
|
|
//[[Products]]
|
|
//Name = "Hammer"
|
|
//Sku = "738594937"
|
|
//
|
|
//[[Products]] # empty table within the array
|
|
//
|
|
//[[Products]]
|
|
//Name = "Nail"
|
|
//Sku = "284758393"
|
|
//
|
|
//Color = "gray"
|
|
//`
|
|
//
|
|
// type Product struct {
|
|
// Name string
|
|
// Sku string
|
|
// Color string
|
|
// }
|
|
//
|
|
// type Data struct {
|
|
// Products []Product
|
|
// }
|
|
//
|
|
// x := Data{}
|
|
// err := Unmarshal([]byte(doc), &x)
|
|
//
|
|
// require.NoError(t, err)
|
|
//
|
|
// expected := Data{
|
|
// Products: []Product{
|
|
// {
|
|
// Name: "Hammer",
|
|
// Sku: "738594937",
|
|
// },
|
|
// {},
|
|
// {
|
|
// Name: "Nail",
|
|
// Sku: "284758393",
|
|
// Color: "gray",
|
|
// },
|
|
// },
|
|
// }
|
|
//
|
|
// assert.Equal(t, expected, x)
|
|
//}
|
|
|
|
//func TestUnmarshalArrayTablesNested(t *testing.T) {
|
|
// doc := `
|
|
//[[Fruits]]
|
|
//Name = "apple"
|
|
//
|
|
//[Fruits.Physical] # subtable
|
|
//Color = "red"
|
|
//Shape = "round"
|
|
//
|
|
//[[Fruits.Varieties]] # nested array of tables
|
|
//Name = "red delicious"
|
|
//
|
|
//[[fruits.varieties]]
|
|
//Name = "granny smith"
|
|
//
|
|
//
|
|
//[[fruits]]
|
|
//Name = "banana"
|
|
//
|
|
//[[fruits.varieties]]
|
|
//Name = "plantain"
|
|
//`
|
|
//
|
|
//}
|