/
suggestion_test.go
101 lines (81 loc) · 3.47 KB
/
suggestion_test.go
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
92
93
94
95
96
97
98
99
100
101
package jid
import (
"bytes"
"io"
"testing"
simplejson "github.com/bitly/go-simplejson"
"github.com/stretchr/testify/assert"
)
func TestNewSuggestion(t *testing.T) {
var assert = assert.New(t)
assert.Equal(NewSuggestion(), &Suggestion{})
}
func TestSuggestionGet(t *testing.T) {
var assert = assert.New(t)
j := createJson(`{"name":"simeji-github"}`)
s := NewSuggestion()
j = createJson(`{"name":"simeji-github", "naming":"simeji", "nickname":"simejisimeji"}`)
assert.Equal([]string{"m", "nam"}, s.Get(j, "na"))
j = createJson(`{"abcde":"simeji-github", "abcdef":"simeji", "ab":"simejisimeji"}`)
assert.Equal([]string{"", ""}, s.Get(j, ""))
assert.Equal([]string{"b", "ab"}, s.Get(j, "a"))
assert.Equal([]string{"de", "abcde"}, s.Get(j, "abc"))
assert.Equal([]string{"", "abcde"}, s.Get(j, "abcde"))
j = createJson(`["zero"]`)
assert.Equal([]string{"[0]", "[0]"}, s.Get(j, ""))
assert.Equal([]string{"0]", "[0]"}, s.Get(j, "["))
assert.Equal([]string{"]", "[0]"}, s.Get(j, "[0"))
j = createJson(`["zero", "one"]`)
assert.Equal([]string{"[", "["}, s.Get(j, ""))
assert.Equal([]string{"", "["}, s.Get(j, "["))
assert.Equal([]string{"]", "[0]"}, s.Get(j, "[0"))
j = createJson(`{"Abcabc":"simeji-github", "Abcdef":"simeji"}`)
assert.Equal([]string{"bc", "Abc"}, s.Get(j, "a"))
assert.Equal([]string{"c", "Abc"}, s.Get(j, "ab"))
j = createJson(`{"RootDeviceNames":"simeji-github", "RootDeviceType":"simeji"}`)
assert.Equal([]string{"ootDevice", "RootDevice"}, s.Get(j, "r"))
assert.Equal([]string{"ootDevice", "RootDevice"}, s.Get(j, "R"))
}
func TestSuggestionGetCurrentType(t *testing.T) {
var assert = assert.New(t)
s := NewSuggestion()
j := createJson(`[1,2,3]`)
assert.Equal(ARRAY, s.GetCurrentType(j))
j = createJson(`{"name":[1,2,3], "naming":{"account":"simeji"}, "test":"simeji", "testing":"ok"}`)
assert.Equal(MAP, s.GetCurrentType(j))
j = createJson(`"name"`)
assert.Equal(STRING, s.GetCurrentType(j))
j = createJson("1")
assert.Equal(UNKNOWN, s.GetCurrentType(j))
}
func TestSuggestionGetCandidateKeys(t *testing.T) {
var assert = assert.New(t)
j := createJson(`{"naming":"simeji", "nickname":"simejisimeji", "city":"tokyo", "name":"simeji-github" }`)
s := NewSuggestion()
assert.Equal([]string{"city", "name", "naming", "nickname"}, s.GetCandidateKeys(j, ""))
assert.Equal([]string{"name", "naming", "nickname"}, s.GetCandidateKeys(j, "n"))
assert.Equal([]string{"name", "naming"}, s.GetCandidateKeys(j, "na"))
assert.Equal([]string{}, s.GetCandidateKeys(j, "nana"))
j = createJson(`{"abcde":"simeji-github", "abcdef":"simeji", "ab":"simejisimeji"}`)
assert.Equal([]string{"abcde", "abcdef"}, s.GetCandidateKeys(j, "abcde"))
j = createJson(`{"name":"simeji-github"}`)
assert.Equal([]string{"name"}, s.GetCandidateKeys(j, ""))
j = createJson(`{"n":"simeji-github"}`)
assert.Equal([]string{"n"}, s.GetCandidateKeys(j, ""))
j = createJson(`[1,2,"aa"]`)
s = NewSuggestion()
assert.Equal([]string{}, s.GetCandidateKeys(j, "["))
}
func TestSuggestionGetCandidateKeysWithDots(t *testing.T) {
var assert = assert.New(t)
j := createJson(`{"nam.ing":"simeji", "nickname":"simejisimeji", "city":"tokyo", "name":"simeji-github" }`)
s := NewSuggestion()
assert.Equal([]string{"city", `\"nam.ing\"`, "name", "nickname"}, s.GetCandidateKeys(j, ""))
assert.Equal([]string{`\"nam.ing\"`, "name", "nickname"}, s.GetCandidateKeys(j, "n"))
}
func createJson(s string) *simplejson.Json {
r := bytes.NewBufferString(s)
buf, _ := io.ReadAll(r)
j, _ := simplejson.NewJson(buf)
return j
}