67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package handlers
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"veza-backend-api/internal/models"
|
|
"veza-backend-api/internal/services"
|
|
)
|
|
|
|
func setupFeatureFlagTestDB(t *testing.T) *gorm.DB {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, db.AutoMigrate(&models.FeatureFlag{}))
|
|
// Seed with one flag for toggle test
|
|
assert.NoError(t, db.Create(&models.FeatureFlag{
|
|
Name: "TEST_FLAG",
|
|
Enabled: false,
|
|
Description: "Test",
|
|
}).Error)
|
|
return db
|
|
}
|
|
|
|
func TestFeatureFlagHandler_List(t *testing.T) {
|
|
db := setupFeatureFlagTestDB(t)
|
|
logger := zap.NewNop()
|
|
svc := services.NewFeatureFlagService(db, logger)
|
|
handler := NewFeatureFlagHandler(svc)
|
|
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
router.GET("/admin/feature-flags", handler.List)
|
|
|
|
req := httptest.NewRequest("GET", "/admin/feature-flags", nil)
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
}
|
|
|
|
func TestFeatureFlagHandler_Toggle_NotFound(t *testing.T) {
|
|
db := setupFeatureFlagTestDB(t)
|
|
logger := zap.NewNop()
|
|
svc := services.NewFeatureFlagService(db, logger)
|
|
handler := NewFeatureFlagHandler(svc)
|
|
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
router.PUT("/admin/feature-flags/:name", handler.Toggle)
|
|
|
|
body, _ := json.Marshal(map[string]bool{"enabled": true})
|
|
req := httptest.NewRequest("PUT", "/admin/feature-flags/NONEXISTENT", bytes.NewReader(body))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
w := httptest.NewRecorder()
|
|
router.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, http.StatusInternalServerError, w.Code)
|
|
}
|