mirror of
https://github.com/mikefarah/yq.git
synced 2024-11-13 22:38:04 +00:00
8072e66d46
The delete (short option "d") will delete the YAML subtree at the provided path in the specified file (or STDIN), if it the node exists. More complex support is currently omitted, for example: - specify nodes to delete using an external script - deleting common elements from all elements of an array
388 lines
7.6 KiB
Go
388 lines
7.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"testing"
|
|
|
|
yaml "gopkg.in/yaml.v2"
|
|
)
|
|
|
|
func TestReadMap_simple(t *testing.T) {
|
|
var data = parseData(`
|
|
---
|
|
b:
|
|
c: 2
|
|
`)
|
|
got, _ := readMap(data, "b", []string{"c"})
|
|
assertResult(t, 2, got)
|
|
}
|
|
|
|
func TestReadMap_splat(t *testing.T) {
|
|
var data = parseData(`
|
|
---
|
|
mapSplat:
|
|
item1: things
|
|
item2: whatever
|
|
`)
|
|
res, _ := readMap(data, "mapSplat", []string{"*"})
|
|
result := res.([]interface{})
|
|
var actual = []string{result[0].(string), result[1].(string)}
|
|
sort.Strings(actual)
|
|
assertResult(t, "[things whatever]", fmt.Sprintf("%v", actual))
|
|
}
|
|
|
|
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{})
|
|
var actual = []string{result[0].(string), result[1].(string)}
|
|
sort.Strings(actual)
|
|
assertResult(t, "[apples bananas]", fmt.Sprintf("%v", actual))
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
func TestReadMap_recurse_against_string(t *testing.T) {
|
|
var data = parseData(`
|
|
---
|
|
a: cat
|
|
`)
|
|
got, _ := readMap(data, "a", []string{"b"})
|
|
assertResult(t, nil, got)
|
|
}
|
|
|
|
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")
|
|
}
|
|
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")
|
|
}
|
|
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")
|
|
}
|
|
expectedOutput := `Error accessing array: strconv.ParseInt: parsing "x": invalid syntax`
|
|
assertResult(t, expectedOutput, err.Error())
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
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))
|
|
}
|
|
|
|
func TestWrite_really_simple(t *testing.T) {
|
|
var data = parseData(`
|
|
b: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b"}, "4")
|
|
b := entryInSlice(updated, "b").Value
|
|
assertResult(t, "4", b)
|
|
}
|
|
|
|
func TestWrite_simple(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "c"}, "4")
|
|
b := entryInSlice(updated, "b").Value.(yaml.MapSlice)
|
|
c := entryInSlice(b, "c").Value
|
|
assertResult(t, "4", c)
|
|
}
|
|
|
|
func TestWrite_new(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "d"}, "4")
|
|
b := entryInSlice(updated, "b").Value.(yaml.MapSlice)
|
|
d := entryInSlice(b, "d").Value
|
|
assertResult(t, "4", d)
|
|
}
|
|
|
|
func TestWrite_new_deep(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "d", "f"}, "4")
|
|
got, _ := readMap(updated, "b", []string{"d", "f"})
|
|
assertResult(t, "4", got)
|
|
}
|
|
|
|
func TestWrite_array(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
- aa
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "0"}, "bb")
|
|
|
|
b := entryInSlice(updated, "b").Value.([]interface{})
|
|
assertResult(t, "bb", b[0].(string))
|
|
}
|
|
|
|
func TestWrite_new_array(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "0"}, "4")
|
|
got, _ := readMap(updated, "b", []string{"0"})
|
|
assertResult(t, "4", got)
|
|
}
|
|
|
|
func TestWrite_new_array_deep(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
var expected = `b:
|
|
- c: "4"`
|
|
|
|
updated := writeMap(data, []string{"b", "0", "c"}, "4")
|
|
got, _ := yamlToString(updated)
|
|
assertResult(t, expected, got)
|
|
}
|
|
|
|
func TestWrite_new_map_array_deep(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
|
|
updated := writeMap(data, []string{"b", "d", "0"}, "4")
|
|
got, _ := readMap(updated, "b", []string{"d", "0"})
|
|
assertResult(t, "4", got)
|
|
}
|
|
|
|
func TestWrite_add_to_array(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
- aa
|
|
`)
|
|
|
|
var expected = `b:
|
|
- aa
|
|
- bb`
|
|
|
|
updated := writeMap(data, []string{"b", "1"}, "bb")
|
|
got, _ := yamlToString(updated)
|
|
assertResult(t, expected, got)
|
|
}
|
|
|
|
func TestWrite_with_no_tail(t *testing.T) {
|
|
var data = parseData(`
|
|
b:
|
|
c: 2
|
|
`)
|
|
updated := writeMap(data, []string{"b"}, "4")
|
|
|
|
b := entryInSlice(updated, "b").Value
|
|
assertResult(t, "4", fmt.Sprintf("%v", b))
|
|
}
|
|
|
|
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
|
|
`)
|
|
|
|
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
|
|
`)
|
|
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]
|
|
`)
|
|
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]
|
|
`)
|
|
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]
|
|
`)
|
|
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
|
|
`)
|
|
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
|
|
`)
|
|
result := deleteMap(data, []string{"b", "0", "name"})
|
|
assertResult(t, fmt.Sprintf("%v", expected), fmt.Sprintf("%v", result))
|
|
}
|