blob: 2c8072fdcc2abb18dca5728b53d1b4fccbe5e431 [file] [log] [blame]
// Copyright 2017 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package firestore
import (
"reflect"
"strings"
"testing"
"cloud.google.com/go/internal/testutil"
)
func TestFieldPathValidate(t *testing.T) {
for _, in := range [][]string{nil, {}, {"a", "", "b"}} {
if err := FieldPath(in).validate(); err == nil {
t.Errorf("%v: want error, got nil", in)
}
}
}
func TestFieldPathLess(t *testing.T) {
for _, test := range []struct {
in1, in2 string
want bool
}{
{"a b", "a b", false},
{"a", "b", true},
{"b", "a", false},
{"a", "a b", true},
{"a b", "a", false},
{"a b c", "a b d", true},
{"a b d", "a b c", false},
} {
fp1 := FieldPath(strings.Fields(test.in1))
fp2 := FieldPath(strings.Fields(test.in2))
got := fp1.less(fp2)
if got != test.want {
t.Errorf("%q.less(%q): got %t, want %t", test.in1, test.in2, got, test.want)
}
}
}
func TestCheckForPrefix(t *testing.T) {
for _, test := range []struct {
in []string // field paths as space-separated strings
wantErr bool
}{
{in: []string{"a", "b", "c"}, wantErr: false},
{in: []string{"a b", "b", "c d"}, wantErr: false},
{in: []string{"a b", "a c", "a d"}, wantErr: false},
{in: []string{"a b", "b", "b d"}, wantErr: true},
{in: []string{"a b", "b", "b d"}, wantErr: true},
{in: []string{"b c d", "c d", "b c"}, wantErr: true},
} {
var fps []FieldPath
for _, s := range test.in {
fps = append(fps, strings.Fields(s))
}
err := checkNoDupOrPrefix(fps)
if got, want := (err != nil), test.wantErr; got != want {
t.Errorf("%#v: got '%v', want %t", test.in, err, want)
}
}
}
func TestGetAtPath(t *testing.T) {
type S struct {
X int
Y int `firestore:"y"`
M map[string]interface{}
Next *S
}
const fail = "ERROR" // value for expected error
for _, test := range []struct {
val interface{}
fp FieldPath
want interface{}
}{
{
val: map[string]int(nil),
fp: nil,
want: map[string]int(nil),
},
{
val: 1,
fp: nil,
want: 1,
},
{
val: 1,
fp: []string{"a"},
want: fail,
},
{
val: map[string]int{"a": 2},
fp: []string{"a"},
want: 2,
},
{
val: map[string]int{"a": 2},
fp: []string{"b"},
want: fail,
},
{
val: map[string]interface{}{"a": map[string]int{"b": 3}},
fp: []string{"a", "b"},
want: 3,
},
{
val: map[string]interface{}{"a": map[string]int{"b": 3}},
fp: []string{"a", "b", "c"},
want: fail,
},
{
val: S{X: 1, Y: 2},
fp: nil,
want: S{X: 1, Y: 2},
},
{
val: S{X: 1, Y: 2},
fp: []string{"X"},
want: 1,
},
{
val: S{X: 1, Y: 2},
fp: []string{"Y"},
want: fail, // because Y is tagged with name "y"
},
{
val: S{X: 1, Y: 2},
fp: []string{"y"},
want: 2,
},
{
val: &S{X: 1},
fp: []string{"X"},
want: 1,
},
{
val: &S{X: 1, Next: nil},
fp: []string{"Next"},
want: (*S)(nil),
},
{
val: &S{X: 1, Next: nil},
fp: []string{"Next", "Next"},
want: fail,
},
{
val: map[string]S{"a": {X: 1, Y: 2}},
fp: []string{"a", "y"},
want: 2,
},
{
val: map[string]S{"a": {X: 1, Y: 2}},
fp: []string{"a", "z"},
want: fail,
},
{
val: map[string]*S{
"a": {
M: map[string]interface{}{
"b": S{
Next: &S{
X: 17,
},
},
},
},
},
fp: []string{"a", "M", "b", "Next", "X"},
want: 17,
},
} {
got, err := getAtPath(reflect.ValueOf(test.val), test.fp)
if err != nil && test.want != fail {
t.Errorf("%+v: got error <%v>, want nil", test, err)
}
if err == nil && !testutil.Equal(got, test.want) {
t.Errorf("%+v: got %v, want %v, want nil", test, got, test.want)
}
}
}
func TestToServiceFieldPath(t *testing.T) {
for _, test := range []struct {
in FieldPath
want string
}{
{[]string{"a"}, "a"},
{[]string{"a", "b"}, "a.b"},
{[]string{"a.", "[b*", "c2"}, "`a.`.`[b*`.c2"},
{[]string{"`a", `b\`}, "`\\`a`.`b\\\\`"},
} {
got := test.in.toServiceFieldPath()
if got != test.want {
t.Errorf("%v: got %s, want %s", test.in, got, test.want)
}
}
}
func TestToServiceFieldPathComponent(t *testing.T) {
for _, test := range []struct {
in, want string
}{
{"", "``"},
{"clam_chowder23", "clam_chowder23"},
{"23skidoo", "`23skidoo`"},
{"bak`tik", "`bak\\`tik`"},
{"a\\b", "`a\\\\b`"},
{"dots.are.confusing", "`dots.are.confusing`"},
} {
got := toServiceFieldPathComponent(test.in)
if got != test.want {
t.Errorf("%q: got %q, want %q", test.in, got, test.want)
}
}
}