nexarch/dms/internal/folderapi/handler_test.go

package folderapi

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"os"
	"testing"
	"time"

	"github.com/jackc/pgx/v5/pgxpool"

	"gitea.perlbach24.de/scripte/nexarch/dms/internal/folder"
)

func requireTestPool(t *testing.T) (*pgxpool.Pool, string) {
	t.Helper()
	dsn := os.Getenv("TEST_TENANT_DSN")
	if dsn == "" {
		t.Skip("TEST_TENANT_DSN nicht gesetzt, Integrationstest uebersprungen")
	}
	ctx := context.Background()
	pool, err := pgxpool.New(ctx, dsn)
	if err != nil {
		t.Fatalf("pool: %v", err)
	}
	t.Cleanup(func() { pool.Close() })

	if _, err := pool.Exec(ctx, `
		CREATE EXTENSION IF NOT EXISTS pgcrypto;
		CREATE TABLE IF NOT EXISTS users (
			id UUID PRIMARY KEY DEFAULT gen_random_uuid(), email TEXT NOT NULL UNIQUE, name TEXT NOT NULL,
			status TEXT NOT NULL DEFAULT 'active', created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
		);
		CREATE TABLE IF NOT EXISTS folders (
			id UUID PRIMARY KEY DEFAULT gen_random_uuid(), parent_folder_id UUID REFERENCES folders(id) ON DELETE CASCADE,
			name TEXT NOT NULL, path TEXT NOT NULL, created_by UUID NOT NULL REFERENCES users(id),
			trashed_at TIMESTAMPTZ, trashed_via_folder_id UUID,
			created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
		);
		CREATE INDEX IF NOT EXISTS idx_folders_path ON folders (path text_pattern_ops);
		CREATE TABLE IF NOT EXISTS documents (
			id UUID PRIMARY KEY DEFAULT gen_random_uuid(), folder_id UUID REFERENCES folders(id) ON DELETE SET NULL,
			title TEXT NOT NULL, current_revision_id UUID, created_by UUID NOT NULL REFERENCES users(id),
			trashed_at TIMESTAMPTZ, trashed_via_folder_id UUID,
			created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now(), deleted_at TIMESTAMPTZ
		);
		CREATE TABLE IF NOT EXISTS file_revisions (
			id UUID PRIMARY KEY DEFAULT gen_random_uuid(), document_id UUID NOT NULL REFERENCES documents(id) ON DELETE CASCADE,
			revision_number INT NOT NULL, storage_key TEXT NOT NULL, checksum_sha256 TEXT NOT NULL,
			size_bytes BIGINT NOT NULL, mime_type TEXT NOT NULL, created_by UUID NOT NULL REFERENCES users(id),
			created_at TIMESTAMPTZ NOT NULL DEFAULT now(), wrapped_dek BYTEA,
			UNIQUE (document_id, revision_number)
		);
		CREATE TABLE IF NOT EXISTS revision_scans (
			revision_id UUID PRIMARY KEY REFERENCES file_revisions(id) ON DELETE CASCADE,
			status TEXT NOT NULL CHECK (status IN ('pending', 'processing', 'clean', 'infected')),
			finding TEXT, created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
		);
	`); err != nil {
		t.Fatalf("schema: %v", err)
	}
	t.Cleanup(func() {
		_, _ = pool.Exec(context.Background(), `TRUNCATE revision_scans, file_revisions, documents, folders, users CASCADE`)
	})

	var userID string
	if err := pool.QueryRow(ctx, `
		INSERT INTO users (email, name) VALUES ($1, 'Test-Benutzer') RETURNING id
	`, fmt.Sprintf("folderapi-test-%d@example.test", time.Now().UnixNano())).Scan(&userID); err != nil {
		t.Fatalf("testbenutzer anlegen: %v", err)
	}
	return pool, userID
}

func createDocument(t *testing.T, pool *pgxpool.Pool, userID string, folderID *string, title string, sizeBytes int64) string {
	t.Helper()
	ctx := context.Background()
	var documentID string
	if err := pool.QueryRow(ctx, `
		INSERT INTO documents (folder_id, title, created_by) VALUES ($1, $2, $3) RETURNING id
	`, folderID, title, userID).Scan(&documentID); err != nil {
		t.Fatalf("dokument anlegen: %v", err)
	}
	var revisionID string
	if err := pool.QueryRow(ctx, `
		INSERT INTO file_revisions (document_id, revision_number, storage_key, checksum_sha256, size_bytes, mime_type, created_by)
		VALUES ($1, 1, 'k', 'x', $2, 'application/pdf', $3) RETURNING id
	`, documentID, sizeBytes, userID).Scan(&revisionID); err != nil {
		t.Fatalf("revision anlegen: %v", err)
	}
	if _, err := pool.Exec(ctx, `UPDATE documents SET current_revision_id = $2 WHERE id = $1`, documentID, revisionID); err != nil {
		t.Fatalf("aktuelle revision setzen: %v", err)
	}
	return documentID
}

// Akzeptanzkriterium 1: Baum und Liste bleiben synchron - beide fragen
// denselben Ordner-/Dokumentenbestand ab.
func TestHandleChildrenAndDocuments_ReflectSameFolderState(t *testing.T) {
	pool, userID := requireTestPool(t)
	root, err := folder.Create(context.Background(), pool, "wurzel", nil, userID)
	if err != nil {
		t.Fatalf("root: %v", err)
	}
	child, err := folder.Create(context.Background(), pool, "kind", &root.ID, userID)
	if err != nil {
		t.Fatalf("child: %v", err)
	}
	createDocument(t, pool, userID, &root.ID, "dok-a.pdf", 1000)
	createDocument(t, pool, userID, &child.ID, "dok-b.pdf", 2000)

	mux := http.NewServeMux()
	NewServer(pool).RegisterRoutes(mux)
	server := httptest.NewServer(mux)
	defer server.Close()

	resp, err := http.Get(server.URL + "/folders/" + root.ID + "/children")
	if err != nil {
		t.Fatalf("get children: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var children []folderDTO
	_ = json.NewDecoder(resp.Body).Decode(&children)
	if len(children) != 1 || children[0].ID != child.ID {
		t.Fatalf("erwartet genau 1 kind (%q), habe: %+v", child.ID, children)
	}

	respDocs, err := http.Get(server.URL + "/folders/" + root.ID + "/documents")
	if err != nil {
		t.Fatalf("get documents: %v", err)
	}
	defer func() { _ = respDocs.Body.Close() }()
	var docs []documentDTO
	_ = json.NewDecoder(respDocs.Body).Decode(&docs)
	if len(docs) != 1 || docs[0].Title != "dok-a.pdf" {
		t.Fatalf("erwartet genau 1 dokument in root (dok-a.pdf), habe: %+v", docs)
	}
}

// Brotkrumenpfad liefert die Kette von der Wurzel bis zum Ordner.
func TestHandleBreadcrumb_ReturnsFullTrail(t *testing.T) {
	pool, userID := requireTestPool(t)
	ctx := context.Background()
	root, err := folder.Create(ctx, pool, "wurzel", nil, userID)
	if err != nil {
		t.Fatalf("root: %v", err)
	}
	child, err := folder.Create(ctx, pool, "kind", &root.ID, userID)
	if err != nil {
		t.Fatalf("child: %v", err)
	}
	grandchild, err := folder.Create(ctx, pool, "enkel", &child.ID, userID)
	if err != nil {
		t.Fatalf("grandchild: %v", err)
	}

	mux := http.NewServeMux()
	NewServer(pool).RegisterRoutes(mux)
	server := httptest.NewServer(mux)
	defer server.Close()

	resp, err := http.Get(server.URL + "/folders/" + grandchild.ID + "/breadcrumb")
	if err != nil {
		t.Fatalf("get: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var trail []folderDTO
	_ = json.NewDecoder(resp.Body).Decode(&trail)
	if len(trail) != 3 {
		t.Fatalf("erwartet 3 brotkrumen-eintraege, habe %d: %+v", len(trail), trail)
	}
	if trail[0].Name != "wurzel" || trail[1].Name != "kind" || trail[2].Name != "enkel" {
		t.Fatalf("brotkrumen-reihenfolge falsch: %+v", trail)
	}
}

// Dokumentenliste unterstuetzt Sortierung.
func TestHandleDocuments_SortsByRequestedColumn(t *testing.T) {
	pool, userID := requireTestPool(t)
	root, err := folder.Create(context.Background(), pool, "wurzel", nil, userID)
	if err != nil {
		t.Fatalf("root: %v", err)
	}
	createDocument(t, pool, userID, &root.ID, "b.pdf", 500)
	createDocument(t, pool, userID, &root.ID, "a.pdf", 9000)

	mux := http.NewServeMux()
	NewServer(pool).RegisterRoutes(mux)
	server := httptest.NewServer(mux)
	defer server.Close()

	resp, err := http.Get(server.URL + "/folders/" + root.ID + "/documents?sort=title&order=asc")
	if err != nil {
		t.Fatalf("get: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var docs []documentDTO
	_ = json.NewDecoder(resp.Body).Decode(&docs)
	if len(docs) != 2 || docs[0].Title != "a.pdf" || docs[1].Title != "b.pdf" {
		t.Fatalf("erwartet nach titel aufsteigend sortiert, habe: %+v", docs)
	}
}