tinyauth/internal/controller/context_controller_test.go

package controller_test

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

	"github.com/steveiliop56/tinyauth/internal/config"
	"github.com/steveiliop56/tinyauth/internal/controller"
	"github.com/steveiliop56/tinyauth/internal/utils/tlog"

	"github.com/gin-gonic/gin"
	"gotest.tools/v3/assert"
)

var contextControllerCfg = controller.ContextControllerConfig{
	Providers: []controller.Provider{
		{
			Name:  "Local",
			ID:    "local",
			OAuth: false,
		},
		{
			Name:  "Google",
			ID:    "google",
			OAuth: true,
		},
	},
	Title:                 "Test App",
	AppURL:                "http://localhost:8080",
	CookieDomain:          "localhost",
	ForgotPasswordMessage: "Contact admin to reset your password.",
	BackgroundImage:       "/assets/bg.jpg",
	OAuthAutoRedirect:     "google",
	DisableUIWarnings:     false,
}

var contextCtrlTestContext = config.UserContext{
	Username:    "testuser",
	Name:        "testuser",
	Email:       "test@example.com",
	IsLoggedIn:  true,
	IsBasicAuth: false,
	OAuth:       false,
	Provider:    "local",
	TotpPending: false,
	OAuthGroups: "",
	TotpEnabled: false,
	OAuthSub:    "",
}

func setupContextController(middlewares *[]gin.HandlerFunc) (*gin.Engine, *httptest.ResponseRecorder) {
	tlog.NewSimpleLogger().Init()

	// Setup
	gin.SetMode(gin.TestMode)
	router := gin.Default()
	recorder := httptest.NewRecorder()

	if middlewares != nil {
		for _, m := range *middlewares {
			router.Use(m)
		}
	}

	group := router.Group("/api")

	ctrl := controller.NewContextController(contextControllerCfg, group)
	ctrl.SetupRoutes()

	return router, recorder
}

func TestAppContextHandler(t *testing.T) {
	expectedRes := controller.AppContextResponse{
		Status:                200,
		Message:               "Success",
		Providers:             contextControllerCfg.Providers,
		Title:                 contextControllerCfg.Title,
		AppURL:                contextControllerCfg.AppURL,
		CookieDomain:          contextControllerCfg.CookieDomain,
		ForgotPasswordMessage: contextControllerCfg.ForgotPasswordMessage,
		BackgroundImage:       contextControllerCfg.BackgroundImage,
		OAuthAutoRedirect:     contextControllerCfg.OAuthAutoRedirect,
		DisableUIWarnings:     contextControllerCfg.DisableUIWarnings,
	}

	router, recorder := setupContextController(nil)
	req := httptest.NewRequest("GET", "/api/context/app", nil)
	router.ServeHTTP(recorder, req)

	assert.Equal(t, 200, recorder.Code)

	var ctrlRes controller.AppContextResponse

	err := json.Unmarshal(recorder.Body.Bytes(), &ctrlRes)

	assert.NilError(t, err)
	assert.DeepEqual(t, expectedRes, ctrlRes)
}

func TestUserContextHandler(t *testing.T) {
	expectedRes := controller.UserContextResponse{
		Status:      200,
		Message:     "Success",
		IsLoggedIn:  contextCtrlTestContext.IsLoggedIn,
		Username:    contextCtrlTestContext.Username,
		Name:        contextCtrlTestContext.Name,
		Email:       contextCtrlTestContext.Email,
		Provider:    contextCtrlTestContext.Provider,
		OAuth:       contextCtrlTestContext.OAuth,
		TotpPending: contextCtrlTestContext.TotpPending,
		OAuthName:   contextCtrlTestContext.OAuthName,
	}

	// Test with context
	router, recorder := setupContextController(&[]gin.HandlerFunc{
		func(c *gin.Context) {
			c.Set("context", &contextCtrlTestContext)
			c.Next()
		},
	})

	req := httptest.NewRequest("GET", "/api/context/user", nil)
	router.ServeHTTP(recorder, req)

	assert.Equal(t, 200, recorder.Code)

	var ctrlRes controller.UserContextResponse

	err := json.Unmarshal(recorder.Body.Bytes(), &ctrlRes)

	assert.NilError(t, err)
	assert.DeepEqual(t, expectedRes, ctrlRes)

	// Test no context
	expectedRes = controller.UserContextResponse{
		Status:     401,
		Message:    "Unauthorized",
		IsLoggedIn: false,
	}

	router, recorder = setupContextController(nil)
	req = httptest.NewRequest("GET", "/api/context/user", nil)
	router.ServeHTTP(recorder, req)

	assert.Equal(t, 200, recorder.Code)

	err = json.Unmarshal(recorder.Body.Bytes(), &ctrlRes)

	assert.NilError(t, err)
	assert.DeepEqual(t, expectedRes, ctrlRes)
}