Merge pull request #19 from chemidy/dev

Dev : add tests for fcm

Author: acoshift

app_test.go

@@ -3,16 +3,19 @@ package admin_test
 import (
 	"context"
 	"io/ioutil"
+	"testing"
 
 	admin "github.com/acoshift/go-firebase-admin"
+	"github.com/stretchr/testify/assert"
 	"gopkg.in/yaml.v2"
 )
 
 type config struct {
-	ProjectID      string `yaml:"projectId"`
-	ServiceAccount []byte `yaml:"serviceAccount"`
-	DatabaseURL    string `yaml:"databaseURL"`
-	APIKey         string `yaml:"apiKey"`
+	ProjectID                    string      `yaml:"projectId"`
+	ServiceAccount               []byte      `yaml:"serviceAccount"`
+	DatabaseURL                  string      `yaml:"databaseURL"`
+	DatabaseAuthVariableOverride interface{} `yaml:"DatabaseAuthVariableOverride"`
+	APIKey                       string      `yaml:"apiKey"`
 }
 
 func initApp() *admin.App {
@@ -21,6 +24,39 @@ func initApp() *admin.App {
 	var c config
 	yaml.Unmarshal(bs, &c)
 
+	// if service account is in separate file service_account.json
+	if len(c.ServiceAccount) <= 0 {
+		serviceAccount, _ := ioutil.ReadFile("private/service_account.json")
+		c.ServiceAccount = serviceAccount
+	}
+
 	app, _ := admin.InitializeApp(context.Background(), admin.AppOptions(c))
 	return app
 }
+
+func TestAuth(t *testing.T) {
+
+	app := initApp()
+	firAuth := app.Auth()
+
+	assert.NotNil(t, app)
+	assert.NotNil(t, firAuth)
+}
+
+func TestDatabase(t *testing.T) {
+
+	app := initApp()
+	firDatabase := app.Database()
+
+	assert.NotNil(t, app)
+	assert.NotNil(t, firDatabase)
+}
+
+func TestFCM(t *testing.T) {
+
+	app := initApp()
+	firFCM := app.FCM()
+
+	assert.NotNil(t, app)
+	assert.NotNil(t, firFCM)
+}

fcm.go

@@ -17,7 +17,7 @@ type FCM struct {
 	client *http.Client
 }
 
-const (
+var (
 	fcmSendEndpoint        = "https://fcm.googleapis.com/fcm/send"
 	fcmTopicAddEndpoint    = "https://iid.googleapis.com/iid/v1:batchAdd"
 	fcmTopicRemoveEndpoint = "https://iid.googleapis.com/iid/v1:batchRemove"
@@ -30,6 +30,21 @@ func newFCM(app *App) *FCM {
 	}
 }
 
+// NewFcmSendEndpoint set fcmSendEndpoint URL
+func (fcm *FCM) NewFcmSendEndpoint(endpoint string) {
+	fcmSendEndpoint = endpoint
+}
+
+// NewFcmTopicAddEndpoint set fcmTopicAddEndpoint URL
+func (fcm *FCM) NewFcmTopicAddEndpoint(endpoint string) {
+	fcmTopicAddEndpoint = endpoint
+}
+
+// NewFcmTopicRemoveEndpoint set fcmTopicRemoveEndpoint URL
+func (fcm *FCM) NewFcmTopicRemoveEndpoint(endpoint string) {
+	fcmTopicRemoveEndpoint = endpoint
+}
+
 // SendToDevice Send Message to individual device
 // see https://firebase.google.com/docs/cloud-messaging/admin/send-messages#send_to_individual_devices
 func (fcm *FCM) SendToDevice(ctx context.Context, registrationToken string, payload Message) (*Response, error) {
@@ -235,5 +250,5 @@ func normalizeTopicName(topic string) (result string) {
 	if strings.HasPrefix(topic, "/topics/") {
 		return topic
 	}
-	return fmt.Sprint("/topic/", topic)
+	return fmt.Sprint("/topics/", topic)
 }

fcm_test.go

@@ -0,0 +1,73 @@
+package admin_test
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	admin "github.com/acoshift/go-firebase-admin"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSendToDevices(t *testing.T) {
+
+	t.Run("send=success", func(t *testing.T) {
+		// generate Response
+		srv := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
+
+			rw.WriteHeader(http.StatusOK)
+			rw.Header().Set("Content-Type", "application/json")
+			fmt.Fprint(rw, `{"success": 1,"failure": 0,"results": [{"message_id":"118218","registration_id": "abcd118218","error": ""}]}`)
+		}))
+		defer srv.Close()
+
+		app := initApp()
+		firFCM := app.FCM()
+		firFCM.NewFcmSendEndpoint(srv.URL)
+
+		assert.NotNil(t, app)
+		assert.NotNil(t, firFCM)
+
+		response, err := firFCM.SendToDevice(context.Background(), "mydevicetoken",
+			admin.Message{Notification: admin.Notification{
+				Title: "Hello go firebase admin",
+				Body:  "My little Big Notification",
+				Color: "#ffcc33"},
+			})
+
+		assert.Nil(t, err)
+		assert.NotNil(t, response)
+		assert.Equal(t, 1, response.Success)
+		assert.Equal(t, 0, response.Failure)
+	})
+
+	t.Run("send=failure", func(t *testing.T) {
+		// generate Response
+		srv := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
+
+			rw.WriteHeader(http.StatusBadRequest)
+		}))
+		defer srv.Close()
+
+		app := initApp()
+		firFCM := app.FCM()
+		firFCM.NewFcmSendEndpoint(srv.URL)
+
+		assert.NotNil(t, app)
+		assert.NotNil(t, firFCM)
+
+		response, err := firFCM.SendToDevice(context.Background(), "mydevicetoken",
+			admin.Message{Notification: admin.Notification{
+				Title: "Hello go firebase admin",
+				Body:  "My little Big Notification",
+				Color: "#ffcc33"},
+			})
+
+		assert.NotNil(t, err)
+		assert.Nil(t, response)
+
+	})
+
+}