add GDrive provider support (#118)
* GDrive provider support * More reliable basedir ownership * Fix mimetype
This commit is contained in:
Andrea Spacca
committed by
Remco Verhoef
Remco Verhoef
parent
d0c7241b31
commit
82493d6dcb
239
vendor/cloud.google.com/go/firestore/document_test.go
generated
vendored
Normal file
239
vendor/cloud.google.com/go/firestore/document_test.go
generated
vendored
Normal file
@@ -0,0 +1,239 @@
|
||||
// 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"
|
||||
"sort"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
pb "google.golang.org/genproto/googleapis/firestore/v1beta1"
|
||||
|
||||
tspb "github.com/golang/protobuf/ptypes/timestamp"
|
||||
)
|
||||
|
||||
func TestToProtoDocument(t *testing.T) {
|
||||
type s struct{ I int }
|
||||
|
||||
for _, test := range []struct {
|
||||
in interface{}
|
||||
want *pb.Document
|
||||
wantErr bool
|
||||
}{
|
||||
{nil, nil, true},
|
||||
{[]int{1}, nil, true},
|
||||
{map[string]int{"a": 1},
|
||||
&pb.Document{Fields: map[string]*pb.Value{"a": intval(1)}},
|
||||
false},
|
||||
{s{2}, &pb.Document{Fields: map[string]*pb.Value{"I": intval(2)}}, false},
|
||||
{&s{3}, &pb.Document{Fields: map[string]*pb.Value{"I": intval(3)}}, false},
|
||||
} {
|
||||
got, _, gotErr := toProtoDocument(test.in)
|
||||
if (gotErr != nil) != test.wantErr {
|
||||
t.Errorf("%v: got error %v, want %t", test.in, gotErr, test.wantErr)
|
||||
}
|
||||
if gotErr != nil {
|
||||
continue
|
||||
}
|
||||
if !testEqual(got, test.want) {
|
||||
t.Errorf("%v: got %v, want %v", test.in, got, test.want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewDocumentSnapshot(t *testing.T) {
|
||||
c := &Client{
|
||||
projectID: "projID",
|
||||
databaseID: "(database)",
|
||||
}
|
||||
docRef := c.Doc("C/a")
|
||||
in := &pb.Document{
|
||||
CreateTime: &tspb.Timestamp{Seconds: 10},
|
||||
UpdateTime: &tspb.Timestamp{Seconds: 20},
|
||||
Fields: map[string]*pb.Value{"a": intval(1)},
|
||||
}
|
||||
want := &DocumentSnapshot{
|
||||
Ref: docRef,
|
||||
CreateTime: time.Unix(10, 0).UTC(),
|
||||
UpdateTime: time.Unix(20, 0).UTC(),
|
||||
ReadTime: aTime,
|
||||
proto: in,
|
||||
c: c,
|
||||
}
|
||||
got, err := newDocumentSnapshot(docRef, in, c, aTimestamp)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !testEqual(got, want) {
|
||||
t.Errorf("got %+v\nwant %+v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestData(t *testing.T) {
|
||||
doc := &DocumentSnapshot{
|
||||
proto: &pb.Document{
|
||||
Fields: map[string]*pb.Value{"a": intval(1), "b": strval("x")},
|
||||
},
|
||||
}
|
||||
got := doc.Data()
|
||||
want := map[string]interface{}{"a": int64(1), "b": "x"}
|
||||
if !testEqual(got, want) {
|
||||
t.Errorf("got %#v\nwant %#v", got, want)
|
||||
}
|
||||
var got2 map[string]interface{}
|
||||
if err := doc.DataTo(&got2); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !testEqual(got2, want) {
|
||||
t.Errorf("got %#v\nwant %#v", got2, want)
|
||||
}
|
||||
|
||||
type s struct {
|
||||
A int
|
||||
B string
|
||||
}
|
||||
var got3 s
|
||||
if err := doc.DataTo(&got3); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
want2 := s{A: 1, B: "x"}
|
||||
if !testEqual(got3, want2) {
|
||||
t.Errorf("got %#v\nwant %#v", got3, want2)
|
||||
}
|
||||
}
|
||||
|
||||
var testDoc = &DocumentSnapshot{
|
||||
proto: &pb.Document{
|
||||
Fields: map[string]*pb.Value{
|
||||
"a": intval(1),
|
||||
"b": mapval(map[string]*pb.Value{
|
||||
"`": intval(2),
|
||||
"~": mapval(map[string]*pb.Value{
|
||||
"x": intval(3),
|
||||
}),
|
||||
}),
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
func TestDataAt(t *testing.T) {
|
||||
for _, test := range []struct {
|
||||
fieldPath string
|
||||
want interface{}
|
||||
}{
|
||||
{"a", int64(1)},
|
||||
{"b.`", int64(2)},
|
||||
} {
|
||||
got, err := testDoc.DataAt(test.fieldPath)
|
||||
if err != nil {
|
||||
t.Errorf("%q: %v", test.fieldPath, err)
|
||||
continue
|
||||
}
|
||||
if !testEqual(got, test.want) {
|
||||
t.Errorf("%q: got %v, want %v", test.fieldPath, got, test.want)
|
||||
}
|
||||
}
|
||||
|
||||
for _, bad := range []string{
|
||||
"c.~.x", // bad field path
|
||||
"a.b", // "a" isn't a map
|
||||
"z.b", // bad non-final key
|
||||
"b.z", // bad final key
|
||||
} {
|
||||
_, err := testDoc.DataAt(bad)
|
||||
if err == nil {
|
||||
t.Errorf("%q: got nil, want error", bad)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDataAtPath(t *testing.T) {
|
||||
for _, test := range []struct {
|
||||
fieldPath FieldPath
|
||||
want interface{}
|
||||
}{
|
||||
{[]string{"a"}, int64(1)},
|
||||
{[]string{"b", "`"}, int64(2)},
|
||||
{[]string{"b", "~"}, map[string]interface{}{"x": int64(3)}},
|
||||
{[]string{"b", "~", "x"}, int64(3)},
|
||||
} {
|
||||
got, err := testDoc.DataAtPath(test.fieldPath)
|
||||
if err != nil {
|
||||
t.Errorf("%v: %v", test.fieldPath, err)
|
||||
continue
|
||||
}
|
||||
if !testEqual(got, test.want) {
|
||||
t.Errorf("%v: got %v, want %v", test.fieldPath, got, test.want)
|
||||
}
|
||||
}
|
||||
|
||||
for _, bad := range []FieldPath{
|
||||
[]string{"c", "", "x"}, // bad field path
|
||||
[]string{"a", "b"}, // "a" isn't a map
|
||||
[]string{"z", "~"}, // bad non-final key
|
||||
[]string{"b", "z"}, // bad final key
|
||||
} {
|
||||
_, err := testDoc.DataAtPath(bad)
|
||||
if err == nil {
|
||||
t.Errorf("%v: got nil, want error", bad)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestExtractTransformPaths(t *testing.T) {
|
||||
type S struct {
|
||||
A time.Time `firestore:",serverTimestamp"`
|
||||
B time.Time `firestore:",serverTimestamp"`
|
||||
C *time.Time `firestore:",serverTimestamp"`
|
||||
D *time.Time `firestore:"d.d,serverTimestamp"`
|
||||
E *time.Time `firestore:",serverTimestamp"`
|
||||
F time.Time
|
||||
G int
|
||||
}
|
||||
|
||||
m := map[string]interface{}{
|
||||
"x": 1,
|
||||
"y": &S{
|
||||
// A is a zero time: included
|
||||
B: aTime, // not a zero time: excluded
|
||||
// C is nil: included
|
||||
D: &time.Time{}, // pointer to a zero time: included
|
||||
E: &aTime, // pointer to a non-zero time: excluded
|
||||
// F is a zero time, but does not have the right tag: excluded
|
||||
G: 15, // not a time.Time
|
||||
},
|
||||
"z": map[string]interface{}{"w": ServerTimestamp},
|
||||
}
|
||||
got, err := extractTransformPaths(reflect.ValueOf(m), nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
sort.Sort(byPath(got))
|
||||
want := []FieldPath{{"y", "A"}, {"y", "C"}, {"y", "d.d"}, {"z", "w"}}
|
||||
if !testEqual(got, want) {
|
||||
t.Errorf("got %#v, want %#v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestExtractTransformPathsErrors(t *testing.T) {
|
||||
type S struct {
|
||||
A int `firestore:",serverTimestamp"`
|
||||
}
|
||||
_, err := extractTransformPaths(reflect.ValueOf(S{}), nil)
|
||||
if err == nil {
|
||||
t.Error("got nil, want error")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user