Kazier/backend/repository/resources_test.go

package repository

import (
	"errors"
	"testing"

	"github.com/vaultdrop/backend/dbtest"
)

const repositoryTestURL = "postgres://vaultdrop:vaultdrop@localhost:5432/vaultdrop_repository_test?sslmode=disable"

func newTestResources(t *testing.T) *Resources {
	t.Helper()
	conn := dbtest.OpenTestDatabase(t, repositoryTestURL)
	return &Resources{DB: conn}
}

func mustInsertUser(t *testing.T, repo *Resources, userID string) {
	t.Helper()
	users := &Users{DB: repo.DB}
	username := "user-" + userID[:8]
	if _, err := users.DB.Exec(
		`INSERT INTO users (id, username, username_normalized, password_hash, is_admin, created_at)
		 VALUES ($1, $2, $3, 'test-hash', false, NOW())`,
		userID, username, username,
	); err != nil {
		t.Fatalf("create user: %v", err)
	}
}

func TestCRUDScopedByOwner(t *testing.T) {
	repo := newTestResources(t)
	owner := NewID()
	other := NewID()
	mustInsertUser(t, repo, owner)
	mustInsertUser(t, repo, other)

	folderID := NewID()
	if err := repo.InsertFolder(owner, folderID, "Docs", ""); err != nil {
		t.Fatalf("insert folder: %v", err)
	}

	mime := "text/plain"
	ext := "txt"
	fileID := NewID()
	if err := repo.InsertFile(owner, fileID, "note.txt", folderID, 42, &mime, &ext); err != nil {
		t.Fatalf("insert file: %v", err)
	}

	got, err := repo.GetFile(owner, fileID)
	if err != nil {
		t.Fatalf("get file: %v", err)
	}
	if got.Name != "note.txt" || got.FolderID != folderID || got.Size != 42 {
		t.Errorf("file row inattendu: %+v", got)
	}

	files, total, err := repo.ListFiles(owner, folderID, 10, 0, "created_at", "desc")
	if err != nil {
		t.Fatalf("list files: %v", err)
	}
	if total != 1 || len(files) != 1 || files[0].ID != fileID {
		t.Errorf("list folder: total=%d files=%+v", total, files)
	}

	roots, err := repo.ListRootFolders(owner)
	if err != nil {
		t.Fatalf("list roots: %v", err)
	}
	if len(roots) != 1 || roots[0].ID != folderID {
		t.Errorf("roots: %+v", roots)
	}

	if _, err := repo.GetFile(other, fileID); err != ErrNotFound {
		t.Errorf("autre user doit voir NOT_FOUND, got %v", err)
	}

	deleted, err := repo.DeleteFile(owner, fileID)
	if err != nil || deleted != fileID {
		t.Fatalf("delete file: %v %v", deleted, err)
	}
	if _, err := repo.GetFile(owner, fileID); err != ErrNotFound {
		t.Errorf("après suppression: %v", err)
	}
}

func TestNameConflictAndUnknownFolder(t *testing.T) {
	repo := newTestResources(t)
	owner := NewID()
	mustInsertUser(t, repo, owner)

	folderID := NewID()
	if err := repo.InsertFolder(owner, folderID, "Docs", ""); err != nil {
		t.Fatalf("insert folder: %v", err)
	}

	ext := "txt"
	if err := repo.InsertFile(owner, NewID(), "note.txt", folderID, 1, nil, &ext); err != nil {
		t.Fatalf("insert first file: %v", err)
	}
	if err := repo.InsertFile(owner, NewID(), "note.txt", folderID, 1, nil, nil); err != ErrNameConflict {
		t.Errorf("même nom dans le même dossier doit être NAME_CONFLICT, got %v", err)
	}

	// Le même nom est autorisé dans un dossier différent (UNIQUE(parent_id, name)).
	otherFolder := NewID()
	if err := repo.InsertFolder(owner, otherFolder, "Other", ""); err != nil {
		t.Fatalf("insert folder 2: %v", err)
	}
	if err := repo.InsertFile(owner, NewID(), "note.txt", otherFolder, 1, nil, nil); err != nil {
		t.Errorf("même nom dans un autre dossier : %v", err)
	}

	unknown := NewID()
	err := repo.InsertFile(owner, NewID(), "x.txt", unknown, 1, nil, nil)
	if !errors.Is(err, ErrNotFound) {
		t.Errorf("folder inconnu doit être NOT_FOUND, got %v", err)
	}
}