update tests

Author: chemidy

app_test.go

@@ -3,8 +3,10 @@ package admin_test
 import (
 	"context"
 	"io/ioutil"
+	"testing"
 
 	admin "github.com/acoshift/go-firebase-admin"
+	"github.com/stretchr/testify/assert"
 	"gopkg.in/yaml.v2"
 )
 
@@ -31,3 +33,30 @@ func initApp() *admin.App {
 	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_test.go

@@ -13,38 +13,61 @@ import (
 
 func TestSendToDevices(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, 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, fmt.Sprintf("unexpected error: %v", err))
-	assert.Equal(t, 1, response.Success, fmt.Sprintf("expected success: %v", err))
-	assert.Equal(t, 0, response.Failure, fmt.Sprintf("Not expected failure: %v", err))
+	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)
+
+	})
 
 }