Scriberr/internal/api/profile_settings_test.go

package api

import (
	"net/http"
	"strings"
	"testing"
	"time"

	"scriberr/internal/database"
	"scriberr/internal/models"

	"github.com/stretchr/testify/require"
)

func TestProfileCRUDAndDefaultSelection(t *testing.T) {
	s := newAuthTestServer(t)
	token := registerForFileTests(t, s)

	resp, body := s.request(t, http.MethodPost, "/api/v1/profiles", map[string]any{
		"name":        "Fast local",
		"description": "Fast local transcription",
		"is_default":  true,
		"options": map[string]any{
			"model":       "base",
			"language":    "en",
			"diarization": false,
			"device":      "auto",
		},
	}, token, "")
	require.Equal(t, http.StatusCreated, resp.Code)
	firstID := body["id"].(string)
	require.True(t, strings.HasPrefix(firstID, "profile_"))
	require.Equal(t, true, body["is_default"])
	require.Equal(t, "Fast local", body["name"])
	require.Equal(t, "base", body["options"].(map[string]any)["model"])

	resp, body = s.request(t, http.MethodGet, "/api/v1/settings", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, firstID, body["default_profile_id"])

	resp, body = s.request(t, http.MethodPost, "/api/v1/profiles", map[string]any{
		"name":       "Accurate",
		"is_default": true,
		"options": map[string]any{
			"model":       "large-v3",
			"language":    "en",
			"diarization": true,
			"device":      "cpu",
		},
	}, token, "")
	require.Equal(t, http.StatusCreated, resp.Code)
	secondID := body["id"].(string)
	require.Equal(t, true, body["is_default"])

	resp, body = s.request(t, http.MethodGet, "/api/v1/profiles", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	items := body["items"].([]any)
	require.Len(t, items, 2)
	defaultCount := 0
	for _, raw := range items {
		item := raw.(map[string]any)
		if item["is_default"].(bool) {
			defaultCount++
			require.Equal(t, secondID, item["id"])
		}
	}
	require.Equal(t, 1, defaultCount)

	resp, body = s.request(t, http.MethodPost, "/api/v1/profiles/"+firstID+":set-default", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, firstID, body["id"])
	require.Equal(t, true, body["is_default"])

	resp, body = s.request(t, http.MethodPatch, "/api/v1/profiles/"+firstID, map[string]any{
		"name":        "Fast local renamed",
		"description": "Updated",
		"options": map[string]any{
			"model":       "small",
			"language":    "fr",
			"diarization": true,
			"device":      "auto",
		},
	}, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, "Fast local renamed", body["name"])
	require.Equal(t, "small", body["options"].(map[string]any)["model"])

	resp, body = s.request(t, http.MethodGet, "/api/v1/profiles/"+firstID, nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, true, body["is_default"])

	resp, _ = s.request(t, http.MethodDelete, "/api/v1/profiles/"+secondID, nil, token, "")
	require.Equal(t, http.StatusNoContent, resp.Code)

	resp, _ = s.request(t, http.MethodGet, "/api/v1/profiles/"+secondID, nil, token, "")
	require.Equal(t, http.StatusNotFound, resp.Code)
}

func TestProfileValidationAndAuth(t *testing.T) {
	s := newAuthTestServer(t)
	token := registerForFileTests(t, s)

	resp, _ := s.request(t, http.MethodGet, "/api/v1/profiles", nil, "", "")
	require.Equal(t, http.StatusUnauthorized, resp.Code)

	resp, body := s.request(t, http.MethodPost, "/api/v1/profiles", map[string]any{
		"name": "Invalid",
		"options": map[string]any{
			"language": "english",
		},
	}, token, "")
	require.Equal(t, http.StatusUnprocessableEntity, resp.Code)
	errBody := body["error"].(map[string]any)
	require.Equal(t, "options.language", errBody["field"])

	resp, _ = s.request(t, http.MethodGet, "/api/v1/profiles/profile_missing", nil, token, "")
	require.Equal(t, http.StatusNotFound, resp.Code)
}

func TestGetProfileDoesNotPublishUpdateEvent(t *testing.T) {
	s := newAuthTestServer(t)
	token := registerForFileTests(t, s)

	resp, body := s.request(t, http.MethodPost, "/api/v1/profiles", map[string]any{
		"name": "Read only profile",
		"options": map[string]any{
			"model": "base",
		},
	}, token, "")
	require.Equal(t, http.StatusCreated, resp.Code)
	profileID := body["id"].(string)

	sub, unsubscribe := s.handler.events.subscribe("")
	defer unsubscribe()

	resp, _ = s.request(t, http.MethodGet, "/api/v1/profiles/"+profileID, nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)

	select {
	case event := <-sub.ch:
		t.Fatalf("GET profile unexpectedly published %s", event.Name)
	case <-time.After(25 * time.Millisecond):
	}
}

func TestSettingsPartialUpdateAndValidation(t *testing.T) {
	s := newAuthTestServer(t)
	token := registerForFileTests(t, s)

	resp, body := s.request(t, http.MethodGet, "/api/v1/settings", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, false, body["auto_transcription_enabled"])
	require.Equal(t, true, body["local_only"])
	require.Greater(t, body["max_upload_size_mb"], float64(0))

	resp, body = s.request(t, http.MethodPost, "/api/v1/profiles", map[string]any{
		"name":       "Default",
		"is_default": true,
		"options": map[string]any{
			"model": "base",
		},
	}, token, "")
	require.Equal(t, http.StatusCreated, resp.Code)
	profileID := body["id"].(string)

	resp, body = s.request(t, http.MethodPatch, "/api/v1/settings", map[string]any{
		"auto_transcription_enabled": true,
		"default_profile_id":         profileID,
	}, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, true, body["auto_transcription_enabled"])
	require.Equal(t, profileID, body["default_profile_id"])
	require.Equal(t, true, body["local_only"])

	var user models.User
	require.NoError(t, database.DB.First(&user).Error)
	require.NotNil(t, user.DefaultProfileID)
	require.Equal(t, strings.TrimPrefix(profileID, "profile_"), *user.DefaultProfileID)

	resp, body = s.request(t, http.MethodPatch, "/api/v1/settings", map[string]any{
		"default_profile_id": "profile_missing",
	}, token, "")
	require.Equal(t, http.StatusUnprocessableEntity, resp.Code)
	errBody := body["error"].(map[string]any)
	require.Equal(t, "default_profile_id", errBody["field"])
}

func TestCapabilitiesQueueAndEvents(t *testing.T) {
	s := newAuthTestServer(t)
	token := registerForFileTests(t, s)
	fileID, _ := createUploadedFileForTranscription(t, s, token)

	resp, body := s.request(t, http.MethodPost, "/api/v1/transcriptions", map[string]any{
		"file_id": fileID,
		"title":   "Queued",
	}, token, "")
	require.Equal(t, http.StatusAccepted, resp.Code)
	transcriptionID := body["id"].(string)

	resp, body = s.request(t, http.MethodGet, "/api/v1/models/transcription", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	items := body["items"].([]any)
	require.NotEmpty(t, items)
	model := items[0].(map[string]any)
	require.Equal(t, "local", model["provider"])
	require.Contains(t, model["capabilities"].([]any), "transcription")

	resp, body = s.request(t, http.MethodGet, "/api/v1/admin/queue", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.Equal(t, float64(1), body["queued"])
	require.Equal(t, float64(0), body["processing"])
	require.Equal(t, float64(0), body["failed"])

	resp, rawLogs := s.rawRequest(t, http.MethodGet, "/api/v1/transcriptions/"+transcriptionID+"/logs", nil, token, "")
	require.Equal(t, http.StatusOK, resp.Code)
	require.NotContains(t, rawLogs, "/")
}