yq/data_navigator_test.go

400 lines
7.8 KiB
Go
Raw Normal View History

2015-09-28 01:39:16 +00:00
package main
import (
2015-09-28 02:00:38 +00:00
"fmt"
2015-10-05 23:01:33 +00:00
"sort"
2015-09-28 02:00:38 +00:00
"testing"
2015-09-28 01:39:16 +00:00
)
func TestReadMap_simple(t *testing.T) {
var data = parseData(`
---
b:
c: 2
`)
got, _ := readMap(data, "b", []string{"c"})
assertResult(t, 2, got)
2015-09-28 01:39:16 +00:00
}
func TestReadMap_numberKey(t *testing.T) {
var data = parseData(`
---
200: things
`)
got, _ := readMap(data, "200", []string{})
assertResult(t, "things", got)
}
2015-10-05 23:01:33 +00:00
func TestReadMap_splat(t *testing.T) {
var data = parseData(`
---
mapSplat:
item1: things
item2: whatever
otherThing: cat
2015-10-05 23:01:33 +00:00
`)
res, _ := readMap(data, "mapSplat", []string{"*"})
assertResult(t, "[things whatever cat]", fmt.Sprintf("%v", res))
}
func TestReadMap_prefixSplat(t *testing.T) {
var data = parseData(`
---
mapSplat:
item1: things
item2: whatever
otherThing: cat
`)
res, _ := readMap(data, "mapSplat", []string{"item*"})
assertResult(t, "[things whatever]", fmt.Sprintf("%v", res))
2015-10-05 23:01:33 +00:00
}
func TestReadMap_deep_splat(t *testing.T) {
var data = parseData(`
---
mapSplatDeep:
item1:
cats: bananas
item2:
cats: apples
`)
res, _ := readMap(data, "mapSplatDeep", []string{"*", "cats"})
result := res.([]interface{})
2015-10-05 23:01:33 +00:00
var actual = []string{result[0].(string), result[1].(string)}
sort.Strings(actual)
assertResult(t, "[apples bananas]", fmt.Sprintf("%v", actual))
}
2015-10-03 07:06:33 +00:00
func TestReadMap_key_doesnt_exist(t *testing.T) {
var data = parseData(`
---
b:
c: 2
`)
got, _ := readMap(data, "b.x.f", []string{"c"})
assertResult(t, nil, got)
2015-10-03 07:06:33 +00:00
}
func TestReadMap_recurse_against_string(t *testing.T) {
var data = parseData(`
---
a: cat
`)
got, _ := readMap(data, "a", []string{"b"})
assertResult(t, nil, got)
}
2015-10-03 07:06:33 +00:00
func TestReadMap_with_array(t *testing.T) {
var data = parseData(`
---
b:
d:
- 3
- 4
`)
got, _ := readMap(data, "b", []string{"d", "1"})
assertResult(t, 4, got)
}
func TestReadMap_with_array_and_bad_index(t *testing.T) {
var data = parseData(`
---
b:
d:
- 3
- 4
`)
_, err := readMap(data, "b", []string{"d", "x"})
if err == nil {
t.Fatal("Expected error due to invalid path")
}
2019-01-20 22:33:14 +00:00
expectedOutput := `error accessing array: strconv.ParseInt: parsing "x": invalid syntax`
assertResult(t, expectedOutput, err.Error())
}
func TestReadMap_with_mapsplat_array_and_bad_index(t *testing.T) {
var data = parseData(`
---
b:
d:
e:
- 3
- 4
f:
- 1
- 2
`)
_, err := readMap(data, "b", []string{"d", "*", "x"})
if err == nil {
t.Fatal("Expected error due to invalid path")
}
2019-01-20 22:33:14 +00:00
expectedOutput := `error accessing array: strconv.ParseInt: parsing "x": invalid syntax`
assertResult(t, expectedOutput, err.Error())
}
func TestReadMap_with_arraysplat_map_array_and_bad_index(t *testing.T) {
var data = parseData(`
---
b:
d:
- names:
- fred
- smith
- names:
- sam
- bo
`)
_, err := readMap(data, "b", []string{"d", "*", "names", "x"})
if err == nil {
t.Fatal("Expected error due to invalid path")
}
2019-01-20 22:33:14 +00:00
expectedOutput := `error accessing array: strconv.ParseInt: parsing "x": invalid syntax`
assertResult(t, expectedOutput, err.Error())
2015-09-28 01:39:16 +00:00
}
2015-10-03 07:06:33 +00:00
func TestReadMap_with_array_out_of_bounds(t *testing.T) {
var data = parseData(`
---
b:
d:
- 3
- 4
`)
got, _ := readMap(data, "b", []string{"d", "3"})
assertResult(t, nil, got)
}
2016-03-16 05:11:06 +00:00
func TestReadMap_with_array_out_of_bounds_by_1(t *testing.T) {
var data = parseData(`
---
b:
d:
- 3
- 4
`)
got, _ := readMap(data, "b", []string{"d", "2"})
assertResult(t, nil, got)
2016-03-16 05:11:06 +00:00
}
2015-10-05 03:41:01 +00:00
func TestReadMap_with_array_splat(t *testing.T) {
var data = parseData(`
e:
-
name: Fred
thing: cat
-
name: Sam
thing: dog
`)
got, _ := readMap(data, "e", []string{"*", "name"})
assertResult(t, "[Fred Sam]", fmt.Sprintf("%v", got))
2015-10-05 03:41:01 +00:00
}
2017-02-26 22:01:52 +00:00
func TestWrite_really_simple(t *testing.T) {
var data = parseData(`
b: 2
`)
2017-04-11 23:16:54 +00:00
updated := writeMap(data, []string{"b"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b 4}]", fmt.Sprintf("%v", updated))
2017-02-26 22:01:52 +00:00
}
func TestWrite_simple(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
2017-04-11 23:16:54 +00:00
updated := writeMap(data, []string{"b", "c"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b [{c 4}]}]", fmt.Sprintf("%v", updated))
2015-10-01 04:43:57 +00:00
}
2017-04-11 23:16:54 +00:00
func TestWrite_new(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
updated := writeMap(data, []string{"b", "d"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b [{c 2} {d 4}]}]", fmt.Sprintf("%v", updated))
2017-04-11 23:16:54 +00:00
}
func TestWrite_new_deep(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
updated := writeMap(data, []string{"b", "d", "f"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b [{c 2} {d [{f 4}]}]}]", fmt.Sprintf("%v", updated))
2017-04-11 23:16:54 +00:00
}
func TestWrite_array(t *testing.T) {
var data = parseData(`
b:
- aa
`)
2017-04-11 23:16:54 +00:00
updated := writeMap(data, []string{"b", "0"}, "bb")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b [bb]}]", fmt.Sprintf("%v", updated))
}
2017-04-11 23:16:54 +00:00
func TestWrite_new_array(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
updated := writeMap(data, []string{"b", "0"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b [{c 2} {0 4}]}]", fmt.Sprintf("%v", updated))
2017-04-11 23:16:54 +00:00
}
func TestWrite_new_array_deep(t *testing.T) {
var data = parseData(`
a: apple
2017-04-11 23:16:54 +00:00
`)
var expected = `a: apple
b:
2017-04-11 23:16:54 +00:00
- c: "4"`
updated := writeMap(data, []string{"b", "+", "c"}, "4")
got, _ := yamlToString(updated)
assertResult(t, expected, got)
2017-04-11 23:16:54 +00:00
}
func TestWrite_new_map_array_deep(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
2019-05-12 23:13:45 +00:00
var expected = `b:
c: 2
d:
- "4"`
2017-04-11 23:16:54 +00:00
updated := writeMap(data, []string{"b", "d", "+"}, "4")
2019-05-12 23:13:45 +00:00
got, _ := yamlToString(updated)
assertResult(t, expected, got)
2017-04-11 23:16:54 +00:00
}
2017-04-17 22:53:27 +00:00
func TestWrite_add_to_array(t *testing.T) {
2017-04-17 23:14:06 +00:00
var data = parseData(`
2017-04-17 22:53:27 +00:00
b:
- aa
`)
2017-04-17 23:14:06 +00:00
var expected = `b:
2017-04-17 22:53:27 +00:00
- aa
- bb`
2017-04-17 23:14:06 +00:00
updated := writeMap(data, []string{"b", "1"}, "bb")
got, _ := yamlToString(updated)
assertResult(t, expected, got)
2017-04-17 22:53:27 +00:00
}
func TestWrite_with_no_tail(t *testing.T) {
var data = parseData(`
b:
c: 2
`)
2017-04-11 23:16:54 +00:00
updated := writeMap(data, []string{"b"}, "4")
2019-05-12 23:13:45 +00:00
assertResult(t, "[{b 4}]", fmt.Sprintf("%v", updated))
}
2017-04-14 02:39:55 +00:00
func TestWriteMap_no_paths(t *testing.T) {
var data = parseData(`
b: 5
`)
result := writeMap(data, []string{}, 4)
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestWriteArray_no_paths(t *testing.T) {
var data = make([]interface{}, 1)
data[0] = "mike"
result := writeArray(data, []string{}, 4)
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestDelete_MapItem(t *testing.T) {
var data = parseData(`
a: 123
b: 456
`)
var expected = parseData(`
b: 456
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"a"})
assertResult(t, fmt.Sprintf("%v", expected), fmt.Sprintf("%v", result))
}
// Ensure deleting an index into a string does nothing
func TestDelete_index_to_string(t *testing.T) {
var data = parseData(`
a: mystring
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"a", "0"})
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestDelete_list_index(t *testing.T) {
var data = parseData(`
a: [3, 4]
`)
var expected = parseData(`
a: [3]
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"a", "1"})
assertResult(t, fmt.Sprintf("%v", expected), fmt.Sprintf("%v", result))
}
func TestDelete_list_index_beyond_bounds(t *testing.T) {
var data = parseData(`
a: [3, 4]
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"a", "5"})
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestDelete_list_index_out_of_bounds_by_1(t *testing.T) {
var data = parseData(`
a: [3, 4]
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"a", "2"})
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestDelete_no_paths(t *testing.T) {
var data = parseData(`
a: [3, 4]
b:
- name: test
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{})
assertResult(t, fmt.Sprintf("%v", data), fmt.Sprintf("%v", result))
}
func TestDelete_array_map_item(t *testing.T) {
var data = parseData(`
b:
- name: fred
value: blah
- name: john
value: test
`)
var expected = parseData(`
b:
- value: blah
- name: john
value: test
`)
2019-05-14 01:20:41 +00:00
result, _ := deleteMap(data, []string{"b", "0", "name"})
assertResult(t, fmt.Sprintf("%v", expected), fmt.Sprintf("%v", result))
}