mirror of
https://github.com/steveiliop56/tinyauth.git
synced 2025-10-28 04:35:40 +00:00
117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
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)
|
|
}
|