package queryapi_test import ( "encoding/json" "net/http" "testing" queryapi "queryorchestration/api/queryAPI" "queryorchestration/internal/database/repository" "queryorchestration/internal/test" "github.com/google/uuid" "github.com/labstack/echo/v4" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestCreateUISetting(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) body := queryapi.UISettingCreate{ Namespace: "demo-dashboard:user1", Key: "theme", Value: map[string]interface{}{"dark": true}, } ctx, rec := createContextWithJSONBody(t, body) err := cons.CreateUISetting(ctx) require.NoError(t, err) assert.Equal(t, http.StatusCreated, rec.Code) var resp queryapi.UISetting require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &resp)) assert.Equal(t, "demo-dashboard:user1", resp.Namespace) assert.Equal(t, "theme", resp.Key) assert.False(t, resp.IsDeleted) assert.NotEqual(t, uuid.Nil, uuid.UUID(resp.Id)) } func TestListUISettings(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) _, err := cfg.GetDBQueries().CreateUISetting(t.Context(), &repository.CreateUISettingParams{ Namespace: "list-ns", Key: "k1", Value: []byte(`{"a":1}`), }) require.NoError(t, err) ctx, rec := createContext(t) params := queryapi.ListUISettingsParams{Namespace: strPtr("list-ns")} err = cons.ListUISettings(ctx, params) require.NoError(t, err) assert.Equal(t, http.StatusOK, rec.Code) var resp queryapi.UISettingListResponse require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &resp)) require.Len(t, resp.Settings, 1) assert.Equal(t, "list-ns", resp.Settings[0].Namespace) assert.Equal(t, int32(1), resp.TotalCount) } func TestGetUISetting(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) row, err := cfg.GetDBQueries().CreateUISetting(t.Context(), &repository.CreateUISettingParams{ Namespace: "get-ns", Key: "key", Value: []byte(`{}`), }) require.NoError(t, err) ctx, _ := createContext(t) err = cons.GetUISetting(ctx, queryapi.UISettingID(row.ID)) require.NoError(t, err) } func TestGetUISetting_NotFound(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) ctx, rec := createContext(t) err := cons.GetUISetting(ctx, queryapi.UISettingID(uuid.New())) require.Error(t, err) httperr, ok := err.(*echo.HTTPError) require.True(t, ok) assert.Equal(t, http.StatusNotFound, httperr.Code) _ = rec } func TestUpdateUISetting(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) row, err := cfg.GetDBQueries().CreateUISetting(t.Context(), &repository.CreateUISettingParams{ Namespace: "upd-ns", Key: "k", Value: []byte(`{"old":true}`), }) require.NoError(t, err) body := queryapi.UISettingUpdate{ Value: map[string]interface{}{"new": true}, } ctx, rec := createContextWithJSONBody(t, body) err = cons.UpdateUISetting(ctx, queryapi.UISettingID(row.ID)) require.NoError(t, err) assert.Equal(t, http.StatusOK, rec.Code) var resp queryapi.UISetting require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &resp)) assert.Equal(t, "upd-ns", resp.Namespace) } func TestDeleteUISetting(t *testing.T) { cfg := &ControllerConfig{} test.CreateDB(t, cfg) initializeTestConfig(t, cfg) svc := createControllerServices(cfg) cons := queryapi.NewControllers(svc, cfg) row, err := cfg.GetDBQueries().CreateUISetting(t.Context(), &repository.CreateUISettingParams{ Namespace: "del-ns", Key: "k", Value: []byte(`{}`), }) require.NoError(t, err) ctx, rec := createContext(t) err = cons.DeleteUISetting(ctx, queryapi.UISettingID(row.ID)) require.NoError(t, err) assert.Equal(t, http.StatusNoContent, rec.Code) _, err = cfg.GetDBQueries().GetUISetting(t.Context(), row.ID) require.Error(t, err) // not found or no rows (soft-deleted) } func strPtr(s string) *string { return &s }