summaryrefslogtreecommitdiff
path: root/jsonpath/parser_test.go
blob: dd869444d8bdb474437c9688dd35f3825905c5ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package jsonpath

import (
	"strings"
	"testing"

	"src.userspace.com.au/query/json"
)

func TestParse(t *testing.T) {
	tests := []struct {
		path, src, expect string
	}{
		{
			path:   "$",
			src:    `{"test":"one"}`,
			expect: "one",
		},
		{
			path:   "$.test",
			src:    `{"test":"one"}`,
			expect: "one",
		},
		{
			path:   "$.a.b",
			src:    `{"a":{"b":"two"}}`,
			expect: "two",
		},
		{
			path:   "$.a.b.c",
			src:    `{"a":{"b":{"c":"two"}}}`,
			expect: "two",
		},
		{
			path:   "$.a.b",
			src:    `{"fail":{"a":"one"},"a":{"b":"three"}}`,
			expect: "three",
		},
		{
			path:   "$.test.test",
			src:    `{"fail":{"test1":"one"},"test1":{"test3":"two"}}`,
			expect: "",
		},
		{
			path:   "$..test",
			src:    `{"fail":{"test1":{"test":"two"}}}`,
			expect: "two",
		},
		{
			path:   "$.a.b.c.d",
			src:    `{"a":{"b":{"c":{"d":"blah"}}}}`,
			expect: "blah",
		},
		{
			path:   "$.test[2]",
			src:    `{"test":[1,"two","three"]}`,
			expect: "three",
		},
		{
			path:   "$.*",
			src:    `{"test":"one"}`,
			expect: "one",
		},
	}

	p := Parser{}

	for _, tt := range tests {
		doc, err := json.Parse(strings.NewReader(tt.src))
		if err != nil {
			t.Fatalf("json.Parse(%q) failed: %s", tt.src, err)
		}
		//doc.PrintTree(0)

		sel, err := p.Parse(tt.path)
		if err != nil {
			t.Fatalf("Parse(%q) failed: %s", tt.path, err)
		}
		actualText := ""
		actual := sel.MatchFirst(doc)
		if actual != nil {
			actualText = actual.InnerText()
		}

		if tt.expect == "" && actualText != "" {
			t.Fatalf("MatchFirst(%s, %s) => %s, expected nothing", tt.path, tt.src, actualText)
		} else if actualText != tt.expect {
			t.Fatalf("MatchFirst(%s, %s) => %s, expected %s", tt.path, tt.src, actualText, tt.expect)
		}
	}
}