testingUtils.go

/* Copyright (c) 2021, VRAI Labs and/or its affiliates. All rights reserved.
 *
 * This software is licensed under the Apache License, Version 2.0 (the
 * "License") as published by the Apache Software Foundation.
 *
 * 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 thirdparty

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/supertokens/supertokens-golang/recipe/emailverification"
	"github.com/supertokens/supertokens-golang/recipe/session"
	"github.com/supertokens/supertokens-golang/recipe/thirdparty/tpmodels"
	"github.com/supertokens/supertokens-golang/supertokens"
	"github.com/supertokens/supertokens-golang/test/unittesting"
)

func resetAll() {
	supertokens.ResetForTest()
	ResetForTest()
	emailverification.ResetForTest()
	session.ResetForTest()
}

func BeforeEach() {
	unittesting.KillAllST()
	resetAll()
	unittesting.SetUpST()
}

func AfterEach() {
	unittesting.KillAllST()
	resetAll()
	unittesting.CleanST()
}

type PostDataForCustomProvider struct {
	ThirdPartyId     string            `json:"thirdPartyId"`
	AuthCodeResponse map[string]string `json:"authCodeResponse"`
	RedirectUri      string            `json:"redirectURI"`
}

func supertokensInitForTest(t *testing.T, recipes ...supertokens.Recipe) *httptest.Server {
	config := supertokens.TypeInput{
		Supertokens: &supertokens.ConnectionInfo{
			ConnectionURI: "http://localhost:8080",
		},
		AppInfo: supertokens.AppInfo{
			APIDomain:     "api.supertokens.io",
			AppName:       "SuperTokens",
			WebsiteDomain: "supertokens.io",
		},
		RecipeList: recipes,
	}

	err := supertokens.Init(config)
	assert.NoError(t, err)

	mux := http.NewServeMux()
	testServer := httptest.NewServer(supertokens.Middleware(mux))
	return testServer
}

var customProviderForEmailVerification = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				if authCodeResponse.(map[string]interface{})["access_token"] == nil {
					return tpmodels.UserInfo{}, nil
				}
				return tpmodels.UserInfo{
					ID: "user",
					Email: &tpmodels.EmailStruct{
						ID:         "test@example.com",
						IsVerified: false,
					},
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}

var customProvider6 = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				if authCodeResponse.(map[string]interface{})["access_token"] == nil {
					return tpmodels.UserInfo{}, nil
				}
				return tpmodels.UserInfo{
					ID: "user",
					Email: &tpmodels.EmailStruct{
						ID:         "email@test.com",
						IsVerified: true,
					},
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}

var customProvider1 = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				return tpmodels.UserInfo{
					ID: "user",
					Email: &tpmodels.EmailStruct{
						ID:         "email@test.com",
						IsVerified: true,
					},
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}

var customProvider2 = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				return tpmodels.UserInfo{
					ID: authCodeResponse.(map[string]interface{})["id"].(string),
					Email: &tpmodels.EmailStruct{
						ID:         authCodeResponse.(map[string]interface{})["email"].(string),
						IsVerified: true,
					},
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}

var customProvider5 = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				return tpmodels.UserInfo{
					ID: "user",
					Email: &tpmodels.EmailStruct{
						ID:         "email@test.com",
						IsVerified: false,
					},
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}

var customProvider3 = tpmodels.TypeProvider{
	ID: "custom",
	Get: func(redirectURI, authCodeFromRequest *string, userContext *map[string]interface{}) tpmodels.TypeProviderGetResponse {
		return tpmodels.TypeProviderGetResponse{
			AccessTokenAPI: tpmodels.AccessTokenAPI{
				URL: "https://test.com/oauth/token",
			},
			AuthorisationRedirect: tpmodels.AuthorisationRedirect{
				URL: "https://test.com/oauth/auth",
			},
			GetProfileInfo: func(authCodeResponse interface{}, userContext *map[string]interface{}) (tpmodels.UserInfo, error) {
				return tpmodels.UserInfo{
					ID: "user",
				}, nil
			},
			GetClientId: func(userContext *map[string]interface{}) string {
				return "supertokens"
			},
		}
	},
}