query-orchestration/api/queryAPI/uisettings_test.go

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 }