Firestore API (#34)

* Prototyping GCS and Firestore APIs

* Implemented Firestore and GCS APIs; Added unit and integration tests

* Accepting a context argument in Auth(), Storage() and Firestore()

* Removing ctx from internal.AuthConfig and internal.StorageConfig

* Flagging Firestore integration test

Author: hiranya911

firebase.go

@@ -18,6 +18,10 @@
 package firebase
 
 import (
+	"errors"
+
+	"cloud.google.com/go/firestore"
+
 	"firebase.google.com/go/auth"
 	"firebase.google.com/go/internal"
 	"firebase.google.com/go/storage"
@@ -31,6 +35,8 @@ import (
 )
 
 var firebaseScopes = []string{
+	"https://www.googleapis.com/auth/cloud-platform",
+	"https://www.googleapis.com/auth/datastore",
 	"https://www.googleapis.com/auth/devstorage.full_control",
 	"https://www.googleapis.com/auth/firebase",
 	"https://www.googleapis.com/auth/userinfo.email",
@@ -72,6 +78,14 @@ func (a *App) Storage(ctx context.Context) (*storage.Client, error) {
 	return storage.NewClient(ctx, conf)
 }
 
+// Firestore returns a new instance of firestore.Client from the cloud.google.com/go package.
+func (a *App) Firestore(ctx context.Context) (*firestore.Client, error) {
+	if a.projectID == "" {
+		return nil, errors.New("project id is required to access Firestore")
+	}
+	return firestore.NewClient(ctx, a.projectID, a.opts...)
+}
+
 // NewApp creates a new App from the provided config and client options.
 //
 // If the client options contain a valid credential (a service account file, a refresh token file or an

firebase_test.go

@@ -228,6 +228,59 @@ func TestStorage(t *testing.T) {
 	}
 }
 
+func TestFirestore(t *testing.T) {
+	ctx := context.Background()
+	app, err := NewApp(ctx, nil, option.WithCredentialsFile("testdata/service_account.json"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if c, err := app.Firestore(ctx); c == nil || err != nil {
+		t.Errorf("Firestore() = (%v, %v); want (auth, nil)", c, err)
+	}
+}
+
+func TestFirestoreWithProjectID(t *testing.T) {
+	varName := "GCLOUD_PROJECT"
+	current := os.Getenv(varName)
+
+	if err := os.Setenv(varName, ""); err != nil {
+		t.Fatal(err)
+	}
+	defer os.Setenv(varName, current)
+
+	ctx := context.Background()
+	config := &Config{ProjectID: "project-id"}
+	app, err := NewApp(ctx, config, option.WithCredentialsFile("testdata/refresh_token.json"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if c, err := app.Firestore(ctx); c == nil || err != nil {
+		t.Errorf("Firestore() = (%v, %v); want (auth, nil)", c, err)
+	}
+}
+
+func TestFirestoreWithNoProjectID(t *testing.T) {
+	varName := "GCLOUD_PROJECT"
+	current := os.Getenv(varName)
+
+	if err := os.Setenv(varName, ""); err != nil {
+		t.Fatal(err)
+	}
+	defer os.Setenv(varName, current)
+
+	ctx := context.Background()
+	app, err := NewApp(ctx, nil, option.WithCredentialsFile("testdata/refresh_token.json"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if c, err := app.Firestore(ctx); c != nil || err == nil {
+		t.Errorf("Firestore() = (%v, %v); want (nil, error)", c, err)
+	}
+}
+
 func TestCustomTokenSource(t *testing.T) {
 	ctx := context.Background()
 	ts := &testTokenSource{AccessToken: "mock-token-from-custom"}

integration/auth/auth_test.go

@@ -26,10 +26,10 @@ import (
 	"os"
 	"testing"
 
+	"golang.org/x/net/context"
+
 	"firebase.google.com/go/auth"
 	"firebase.google.com/go/integration/internal"
-
-	"golang.org/x/net/context"
 )
 
 const idToolKitURL = "https://www.googleapis.com/identitytoolkit/v3/relyingparty/verifyCustomToken?key=%s"

integration/firestore/firestore_test.go

@@ -0,0 +1,62 @@
+// Copyright 2017 Google Inc. All Rights Reserved.
+//
+// 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 (
+	"context"
+	"log"
+	"reflect"
+	"testing"
+
+	"firebase.google.com/go/integration/internal"
+)
+
+func TestFirestore(t *testing.T) {
+	if testing.Short() {
+		log.Println("skipping Firestore integration tests in short mode.")
+		return
+	}
+	ctx := context.Background()
+	app, err := internal.NewTestApp(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	client, err := app.Firestore(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	doc := client.Collection("cities").Doc("Mountain View")
+	data := map[string]interface{}{
+		"name":       "Mountain View",
+		"country":    "USA",
+		"population": int64(77846),
+		"capital":    false,
+	}
+	if _, err := doc.Set(ctx, data); err != nil {
+		t.Fatal(err)
+	}
+	snap, err := doc.Get(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !reflect.DeepEqual(snap.Data(), data) {
+		t.Errorf("Get() = %v; want %v", snap.Data(), data)
+	}
+	if _, err := doc.Delete(ctx); err != nil {
+		t.Fatal(err)
+	}
+}

internal/internal.go

@@ -16,9 +16,8 @@
 package internal
 
 import (
-	"google.golang.org/api/option"
-
 	"golang.org/x/oauth2/google"
+	"google.golang.org/api/option"
 )
 
 // AuthConfig represents the configuration of Firebase Auth service.