- Implemented SystemPermissionsHandler to check and repair file permissions. - Added endpoints for retrieving and repairing permissions. - Introduced utility functions for permission checks and error mapping. - Created tests for the new handler and utility functions. - Updated routes to include the new permissions endpoints. - Enhanced configuration to support new logging and plugin directories.
152 lines
5.2 KiB
Go
152 lines
5.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/Wikid82/charon/backend/internal/models"
|
|
"github.com/Wikid82/charon/backend/internal/services"
|
|
)
|
|
|
|
func TestNotificationTemplateHandler_CRUDAndPreview(t *testing.T) {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?mode=memory&cache=shared"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&models.NotificationTemplate{}, &models.Notification{}, &models.NotificationProvider{}))
|
|
|
|
svc := services.NewNotificationService(db)
|
|
h := NewNotificationTemplateHandler(svc)
|
|
|
|
r := gin.New()
|
|
r.Use(func(c *gin.Context) {
|
|
c.Set("role", "admin")
|
|
c.Set("userID", uint(1))
|
|
c.Next()
|
|
})
|
|
api := r.Group("/api/v1")
|
|
api.GET("/notifications/templates", h.List)
|
|
api.POST("/notifications/templates", h.Create)
|
|
api.PUT("/notifications/templates/:id", h.Update)
|
|
api.DELETE("/notifications/templates/:id", h.Delete)
|
|
api.POST("/notifications/templates/preview", h.Preview)
|
|
|
|
// Create
|
|
payload := `{"name":"test","config":"{\"hello\":\"world\"}"}`
|
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/notifications/templates", strings.NewReader(payload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusCreated, w.Code)
|
|
var created models.NotificationTemplate
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &created))
|
|
require.NotEmpty(t, created.ID)
|
|
|
|
// List
|
|
req = httptest.NewRequest(http.MethodGet, "/api/v1/notifications/templates", http.NoBody)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
var list []models.NotificationTemplate
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &list))
|
|
require.True(t, len(list) >= 1)
|
|
|
|
// Update
|
|
updatedPayload := `{"name":"updated","config":"{\"hello\":\"updated\"}"}`
|
|
req = httptest.NewRequest(http.MethodPut, "/api/v1/notifications/templates/"+created.ID, strings.NewReader(updatedPayload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
var up models.NotificationTemplate
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &up))
|
|
require.Equal(t, "updated", up.Name)
|
|
|
|
// Preview by id
|
|
previewPayload := `{"template_id":"` + created.ID + `", "data": {}}`
|
|
req = httptest.NewRequest(http.MethodPost, "/api/v1/notifications/templates/preview", strings.NewReader(previewPayload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
var previewResp map[string]any
|
|
require.NoError(t, json.Unmarshal(w.Body.Bytes(), &previewResp))
|
|
require.NotEmpty(t, previewResp["rendered"])
|
|
|
|
// Delete
|
|
req = httptest.NewRequest(http.MethodDelete, "/api/v1/notifications/templates/"+created.ID, http.NoBody)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusOK, w.Code)
|
|
}
|
|
|
|
func TestNotificationTemplateHandler_Create_InvalidJSON(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?mode=memory&cache=shared"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&models.NotificationTemplate{}))
|
|
svc := services.NewNotificationService(db)
|
|
h := NewNotificationTemplateHandler(svc)
|
|
r := gin.New()
|
|
r.Use(func(c *gin.Context) {
|
|
c.Set("role", "admin")
|
|
c.Set("userID", uint(1))
|
|
c.Next()
|
|
})
|
|
r.POST("/api/templates", h.Create)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/api/templates", strings.NewReader(`{invalid}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func TestNotificationTemplateHandler_Update_InvalidJSON(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?mode=memory&cache=shared"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&models.NotificationTemplate{}))
|
|
svc := services.NewNotificationService(db)
|
|
h := NewNotificationTemplateHandler(svc)
|
|
r := gin.New()
|
|
r.Use(func(c *gin.Context) {
|
|
c.Set("role", "admin")
|
|
c.Set("userID", uint(1))
|
|
c.Next()
|
|
})
|
|
r.PUT("/api/templates/:id", h.Update)
|
|
|
|
req := httptest.NewRequest(http.MethodPut, "/api/templates/test-id", strings.NewReader(`{invalid}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusBadRequest, w.Code)
|
|
}
|
|
|
|
func TestNotificationTemplateHandler_Preview_InvalidJSON(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?mode=memory&cache=shared"), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.AutoMigrate(&models.NotificationTemplate{}))
|
|
svc := services.NewNotificationService(db)
|
|
h := NewNotificationTemplateHandler(svc)
|
|
r := gin.New()
|
|
r.Use(func(c *gin.Context) {
|
|
c.Set("role", "admin")
|
|
c.Set("userID", uint(1))
|
|
c.Next()
|
|
})
|
|
r.POST("/api/templates/preview", h.Preview)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, "/api/templates/preview", strings.NewReader(`{invalid}`))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
require.Equal(t, http.StatusBadRequest, w.Code)
|
|
}
|