Avoid fission installation failure due to analytics connection error (#2457)

pkg/tracker/tracker_test.go

@@ -0,0 +1,127 @@
+package tracker
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTracker(t *testing.T) {
+	t.Run("NewTracker", func(test *testing.T) {
+		for _, test := range []struct {
+			name     string
+			gaAPIURl string
+			expected error
+		}{
+			{
+				name:     "GA Tracking ID should not be empty",
+				gaAPIURl: "",
+				expected: errors.New("tracker.NewTracker: GA_TRACKING_ID env not set"),
+			},
+			{
+				name:     "Tracker should initialize properly",
+				gaAPIURl: "/",
+				expected: nil,
+			},
+		} {
+			t.Run(test.name, func(testing *testing.T) {
+				if test.expected == nil {
+					os.Setenv(GA_TRACKING_ID, "UA-000000-2")
+					t, err := NewTracker()
+					assert.Nil(testing, err)
+					assert.NotNil(testing, t)
+				} else {
+					t, err := NewTracker()
+					assert.Nil(testing, t)
+					assert.NotNil(testing, err)
+					assert.Equal(testing, err.Error(), test.expected.Error())
+				}
+			})
+		}
+	})
+
+	t.Run("SendEvent", func(test *testing.T) {
+
+		os.Setenv(GA_TRACKING_ID, "UA-000000-2")
+		tr, err := NewTracker()
+		if err != nil {
+			log.Fatal(err)
+		}
+		for _, test := range []struct {
+			name     string
+			request  *Event
+			expected error
+			status   int
+		}{
+			{
+				name: "category and action should not be empty",
+				request: &Event{
+					Category: "",
+					Action:   "",
+					Label:    "play",
+					Value:    "value",
+				},
+				expected: errors.New("tracker.SendEvent: category and action are required"),
+				status:   http.StatusInternalServerError,
+			},
+			{
+				name: "Google Analytics response should not be OK",
+				request: &Event{
+					Category: "UI_action",
+					Action:   "button_press",
+					Label:    "play",
+					Value:    "value",
+				},
+
+				expected: errors.New("tracker.SendEvent: analytics response status not ok"),
+				status:   http.StatusInternalServerError,
+			},
+			{
+				name: "Google Analytics response should be OK",
+				request: &Event{
+					Category: "UI_action",
+					Action:   "button_press",
+					Label:    "play",
+					Value:    "value",
+				},
+
+				expected: nil,
+				status:   http.StatusOK,
+			},
+		} {
+			t.Run(test.name, func(testing *testing.T) {
+				server := MockHTTPServer(test.status, "")
+				defer server.Close()
+				tr.gaAPIURL = server.URL
+
+				t := tr.SendEvent(context.Background(), *test.request)
+				if test.status == http.StatusOK {
+					assert.Nil(testing, t, test.expected)
+				} else {
+					assert.NotNil(testing, t)
+					assert.Equal(testing, t.Error(), test.expected.Error())
+				}
+
+			})
+		}
+	})
+}
+
+func MockHTTPServer(status int, encodeValue interface{}) *httptest.Server {
+	f := func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(status)
+		w.Header().Set("Content-Type", "application/json")
+		err := json.NewEncoder(w).Encode(encodeValue)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+	return httptest.NewServer(http.HandlerFunc(f))
+}