tests: add tests for context controller

internal/controller/context_controller_test.go

@@ -0,0 +1,116 @@
+package controller_test
+
+import (
+	"encoding/json"
+	"net/http/httptest"
+	"testing"
+	"tinyauth/internal/config"
+	"tinyauth/internal/controller"
+
+	"github.com/gin-gonic/gin"
+	"gotest.tools/v3/assert"
+)
+
+func TestAppContextHandler(t *testing.T) {
+	// Setup
+	controllerCfg := controller.ContextControllerConfig{
+		ConfiguredProviders:   []string{"github", "google", "generic"},
+		Title:                 "Test App",
+		GenericName:           "Generic",
+		AppURL:                "http://localhost:8080",
+		RootDomain:            "localhost",
+		ForgotPasswordMessage: "Contact admin to reset your password.",
+		BackgroundImage:       "/assets/bg.jpg",
+		OAuthAutoRedirect:     "google",
+	}
+
+	expectedRes := controller.AppContextResponse{
+		Status:                200,
+		Message:               "Success",
+		ConfiguredProviders:   controllerCfg.ConfiguredProviders,
+		Title:                 controllerCfg.Title,
+		GenericName:           controllerCfg.GenericName,
+		AppURL:                controllerCfg.AppURL,
+		RootDomain:            controllerCfg.RootDomain,
+		ForgotPasswordMessage: controllerCfg.ForgotPasswordMessage,
+		BackgroundImage:       controllerCfg.BackgroundImage,
+		OAuthAutoRedirect:     controllerCfg.OAuthAutoRedirect,
+	}
+
+	gin.SetMode(gin.TestMode)
+	router := gin.Default()
+	group := router.Group("/api")
+	recorder := httptest.NewRecorder()
+
+	// Test
+	ctrl := controller.NewContextController(controllerCfg, group)
+	ctrl.SetupRoutes()
+
+	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 TestUserContextController(t *testing.T) {
+	// Setup
+	controllerCfg := controller.ContextControllerConfig{}
+
+	userContext := config.UserContext{
+		Username:    "testuser",
+		Name:        "testuser",
+		Email:       "test@example.com",
+		IsLoggedIn:  true,
+		OAuth:       false,
+		Provider:    "username",
+		TotpPending: false,
+		OAuthGroups: "",
+		TotpEnabled: false,
+	}
+
+	expectedRes := controller.UserContextResponse{
+		Status:      200,
+		Message:     "Success",
+		IsLoggedIn:  userContext.IsLoggedIn,
+		Username:    userContext.Username,
+		Name:        userContext.Name,
+		Email:       userContext.Email,
+		Provider:    userContext.Provider,
+		OAuth:       userContext.OAuth,
+		TotpPending: userContext.TotpPending,
+	}
+
+	gin.SetMode(gin.TestMode)
+	router := gin.Default()
+	recorder := httptest.NewRecorder()
+
+	router.Use(func(c *gin.Context) {
+		c.Set("context", &userContext)
+		c.Next()
+	})
+
+	group := router.Group("/api")
+
+	// Test
+	ctrl := controller.NewContextController(controllerCfg, group)
+	ctrl.SetupRoutes()
+
+	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)
+}