Initial cloud-services repo - gateway service + pkg modules

pkg/superset/login_test.go

@@ -0,0 +1,104 @@
+package superset_test
+
+import (
+	"bytes"
+	"io/ioutil"
+	"net/http"
+	"testing"
+
+	"fiskerinc.com/modules/httpclient"
+	"fiskerinc.com/modules/httpclient/mock"
+	"fiskerinc.com/modules/redis"
+	"fiskerinc.com/modules/redis/tester"
+	"fiskerinc.com/modules/superset"
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/assert"
+)
+
+var someErr = errors.New("some err")
+
+func TestGetAccessToken(t *testing.T) {
+	const validToken = "valid_token"
+
+	redisMock := tester.NewRedisMock()
+	type test struct {
+		redisGetResults  interface{}
+		httpClientDoFunc func(req *http.Request) (*http.Response, error)
+		redisErr         error
+		expErr           error
+		expResp          string
+		expSetValues     map[string]tester.ExpiringCache
+	}
+
+	tests := map[string]test{
+		"success_cached": {
+			redisGetResults: []byte(validToken),
+			expErr:          nil,
+			expResp:         validToken,
+			expSetValues:    map[string]tester.ExpiringCache{},
+		},
+		"success": {
+			httpClientDoFunc: successAccHttpDo,
+			expErr:           nil,
+			expResp:          validToken,
+			expSetValues: map[string]tester.ExpiringCache{
+				redis.SupersetAccTokenKey: {
+					Value: validToken,
+				},
+			},
+		},
+		"failed_cache_func": {
+			redisErr: someErr,
+			expErr:   someErr,
+		},
+		"invalid_token": {
+			redisGetResults: 12,
+			expErr:          errors.New("invalid superset access token type (expected []byte]); access token: 12"),
+		},
+		"error_request": {
+			httpClientDoFunc: errorHttpDo,
+			expErr:           someErr,
+		},
+		"forbidden_request": {
+			httpClientDoFunc: forbiddenAccHttpDo,
+			expErr:           errors.New("superset login answered with status: 403 Forbidden"),
+		},
+	}
+
+	for name, tt := range tests {
+		t.Run(name, func(t *testing.T) {
+			httpclient.Client = &mock.Client{DoFunc: tt.httpClientDoFunc}
+			redisMock.Reset()
+			redisMock.GetResults = tt.redisGetResults
+			redisMock.Error = tt.redisErr
+			got, err := superset.GetAccessToken(redisMock)
+			if err != nil && tt.expErr != nil {
+				assert.Equal(t, tt.expErr.Error(), err.Error())
+				return
+			}
+
+			assert.Equal(t, tt.expErr, err)
+			assert.Equal(t, tt.expResp, got)
+			assert.Equal(t, tt.expSetValues, redisMock.SetValues)
+		})
+	}
+}
+
+func successAccHttpDo(req *http.Request) (*http.Response, error) {
+	return &http.Response{
+		StatusCode: http.StatusOK,
+		Body:       ioutil.NopCloser(bytes.NewReader([]byte(`{"access_token":"valid_token","refresh_token":""}`))),
+	}, nil
+}
+
+func errorHttpDo(req *http.Request) (*http.Response, error) {
+	return nil, someErr
+}
+
+func forbiddenAccHttpDo(req *http.Request) (*http.Response, error) {
+	return &http.Response{
+		StatusCode: http.StatusForbidden,
+		Status:     "403 Forbidden",
+		Body:       ioutil.NopCloser(bytes.NewReader([]byte(``))),
+	}, nil
+}