Add tests for json path and improve implementation

jsonte/types/array.go

@@ -58,6 +58,9 @@ func (o JsonArray) Index(i JsonType) (JsonType, error) {
 			return Null, burrito.WrappedErrorf("Index out of bounds: %d", index)
 		}
 	}
+	if b, ok := i.(JsonPath); ok {
+		return b.Get(o)
+	}
 	return Null, burrito.WrappedErrorf("Index must be a number: %s", i.StringValue())
 }
 

jsonte/types/json_path.go

@@ -204,7 +204,7 @@ func AsJsonPath(obj interface{}) JsonPath {
 
 func ParseJsonPath(path string) (JsonPath, error) {
 	path = strings.TrimPrefix(path, "#")
-	if !strings.HasPrefix(path, "/") {
+	if !strings.HasPrefix(path, "/") && !strings.HasPrefix(path, "[") {
 		path = "/" + path
 	}
 	parts := make([]JsonType, 0)

test/eval_test.go

@@ -120,6 +120,20 @@ func assertSemver(t *testing.T, eval jsonte.Result, expected types.Semver) {
 	}
 }
 
+func assertJsonPath(t *testing.T, eval jsonte.Result, expected types.JsonPath) {
+	assertAction(t, eval, types.Value)
+	if eval.Value == nil {
+		t.Fatalf("Result is null")
+	}
+	if !types.IsJsonPath(eval.Value) {
+		t.Fatalf("Result is not a JSON Path (%s)", reflect.TypeOf(eval.Value).Name())
+	}
+	path := types.AsJsonPath(eval.Value)
+	if path.StringValue() != expected.StringValue() {
+		t.Fatalf("Result is not correct (expected %s, got %s)", expected.StringValue(), path.StringValue())
+	}
+}
+
 func assertBool(t *testing.T, eval jsonte.Result, expected bool) {
 	assertAction(t, eval, types.Value)
 	if eval.Value == nil {

test/jsonpath_test.go

@@ -0,0 +1,57 @@
+package test
+
+import (
+	"github.com/MCDevKit/jsonte/jsonte/types"
+	"testing"
+)
+
+func TestJsonPathParsing(t *testing.T) {
+	eval := evaluate(t, `jsonpath('#/test/path[1]')`)
+	assertJsonPath(t, eval, types.JsonPath{
+		Path: []types.JsonType{
+			types.JsonString{Value: "test"},
+			types.JsonString{Value: "path"},
+			types.AsNumber(1),
+		},
+	})
+}
+
+func TestJsonPathParsingWithoutHash(t *testing.T) {
+	eval := evaluate(t, `jsonpath('/test/path[1]')`)
+	assertJsonPath(t, eval, types.JsonPath{
+		Path: []types.JsonType{
+			types.JsonString{Value: "test"},
+			types.JsonString{Value: "path"},
+			types.AsNumber(1),
+		},
+	})
+}
+
+func TestJsonPathParent(t *testing.T) {
+	eval := evaluate(t, `jsonpath('test/path[1]').parent()`)
+	assertJsonPath(t, eval, types.JsonPath{
+		Path: []types.JsonType{
+			types.JsonString{Value: "test"},
+			types.JsonString{Value: "path"},
+		},
+	})
+}
+
+func TestJsonPathParsingNumber(t *testing.T) {
+	eval := evaluate(t, `jsonpath('#[1]')`)
+	assertJsonPath(t, eval, types.JsonPath{
+		Path: []types.JsonType{
+			types.AsNumber(1),
+		},
+	})
+}
+
+func TestObjectAccessByJsonPath(t *testing.T) {
+	eval := evaluate(t, `{"test": {"path": [1, 2, 3]}}[jsonpath('#/test/path[1]')]`)
+	assertNumber(t, eval, 2)
+}
+
+func TestArrayAccessByJsonPath(t *testing.T) {
+	eval := evaluate(t, `[1, 2, 3][jsonpath('#[1]')]`)
+	assertNumber(t, eval, 2)
+}