passage_test.go

package passage_test

import (
	"crypto/rand"
	"fmt"
	"os"
	"strings"
	"sync"
	"testing"

	"github.com/joho/godotenv"
	"github.com/passageidentity/passage-go"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

var (
	PassageAppID     string
	PassageApiKey    string
	PassageUserID    string
	PassageAuthToken string
	RandomEmail      = generateRandomEmail(14)
	CreatedUser      passage.PassageUser
)

func generateRandomEmail(prefixLength int) string {
	n := prefixLength
	randomChars := make([]byte, n)
	if _, err := rand.Read(randomChars); err != nil {
		panic(err)
	}
	email := fmt.Sprintf("%X@email.com", randomChars)
	return strings.ToLower(email)
}

func TestMain(m *testing.M) {
	_ = godotenv.Load(".env")

	PassageAppID = os.Getenv("PASSAGE_APP_ID")
	PassageApiKey = os.Getenv("PASSAGE_API_KEY")
	PassageUserID = os.Getenv("PASSAGE_USER_ID")
	PassageAuthToken = os.Getenv("PASSAGE_AUTH_TOKEN")

	exitVal := m.Run()
	os.Exit(exitVal)
}

func passageUserNotFoundAsserts(t *testing.T, err error) {
	splitError := strings.Split(err.Error(), ", ")
	assert.Len(t, splitError, 3)
	assert.Equal(t, "PassageError - message: User not found", splitError[0])
	assert.Equal(t, "errorCode: user_not_found", splitError[1])
	assert.Equal(t, "statusCode: 404", splitError[2])
}

func passageCouldNotFindUserByIdentifierAsserts(t *testing.T, err error) {
	splitError := strings.Split(err.Error(), ", ")
	assert.Len(t, splitError, 3)
	assert.Equal(t, "PassageError - message: Could not find user with that identifier.", splitError[0])
	assert.Equal(t, "errorCode: user_not_found", splitError[1])
	assert.Equal(t, "statusCode: 404", splitError[2])
}

func passageUnauthorizedAsserts(t *testing.T, err error) {
	splitError := strings.Split(err.Error(), ", ")
	assert.Len(t, splitError, 3)
	assert.Equal(t, "PassageError - message: Invalid access token", splitError[0])
	assert.Equal(t, "errorCode: invalid_access_token", splitError[1])
	assert.Equal(t, "statusCode: 401", splitError[2])
}

func passageBadRequestAsserts(t *testing.T, err error, message string) {
	splitError := strings.Split(err.Error(), ", ")
	assert.Len(t, splitError, 3)
	assert.Equal(t, "PassageError - message: "+message, splitError[0])
	assert.Equal(t, "errorCode: invalid_request", splitError[1])
	assert.Equal(t, "statusCode: 400", splitError[2])
}

// should be run with the -race flag, i.e. `go test -race -run TestAppJWKSCacheWriteConcurrency`
func TestAppJWKSCacheWriteConcurrency(t *testing.T) {
	goRoutineCount := 2

	var wg sync.WaitGroup
	wg.Add(goRoutineCount)

	for i := 0; i < goRoutineCount; i++ {
		go func() {
			defer wg.Done()

			_, err := passage.New(PassageAppID, PassageApiKey)
			require.Nil(t, err)
		}()
	}

	wg.Wait()
}