Add more tests for a and e commands

1 files changed, 24 insertions, 0 deletions

diff --git a/main/main_test.go b/main/main_test.go
index 8512b62..94e1f04 100644
--- a/main/main_test.go
+++ b/main/main_test.go
@@ -107,6 +107,30 @@ func TestMain(t *testing.T) {
 			input: miscInput,
 			expected: `"Hello world these are values"`,
 		},
+		{
+			name: "Drop first element of array",
+			program: `s/#( "people" @( 0 . )@ )#/d`,
+			input: miscInput,
+			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
+		},
+		{
+			name: "Drop last element of array",
+			program: `M/#( "people" @( . #()# )@ )#/{ Ed }`,
+			input: miscInput,
+			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122}]}`,
+		},
+		{
+			name: "Prepend to array",
+			program: "as/#( \"array\" :( `\"First\"` ): )#/",
+			input: miscInput,
+			expected: `{"something":{"nested":"Here is my test value"},"array":["First","Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
+		},
+		{
+			name: "Append to array",
+			program: "es/#( \"array\" :( `\"Last\"` ): )#/",
+			input: miscInput,
+			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values","Last"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
+		},
 	}
 
 	for _, test := range tests {