chore: refresh gramsrv public release

internal/app/files/upload_parts_test.go

@@ -0,0 +1,149 @@
+package files
+
+import (
+	"context"
+	"errors"
+	"strconv"
+	"strings"
+	"testing"
+
+	"telesrv/internal/domain"
+)
+
+func TestSaveFilePartQuotaTreatsRetryAsOverwrite(t *testing.T) {
+	ctx := context.Background()
+	media := newFakeMediaStore()
+	svc, blobs := newUploadPartTestService(t, media, domain.UploadPartQuota{MaxBytes: 4, MaxParts: 1, MaxFiles: 1})
+
+	if _, err := svc.SaveFilePart(ctx, 10, 100, 0, []byte("1234")); err != nil {
+		t.Fatalf("save first part: %v", err)
+	}
+	firstParts, err := media.LoadFileParts(ctx, 10, 100)
+	if err != nil || len(firstParts) != 1 || firstParts[0].ObjectKey == "" {
+		t.Fatalf("load first part metadata: parts=%+v err=%v", firstParts, err)
+	}
+	firstKey := firstParts[0].ObjectKey
+	if _, err := svc.SaveFilePart(ctx, 10, 100, 0, []byte("1234")); err != nil {
+		t.Fatalf("retry same part should overwrite without extra quota: %v", err)
+	}
+	parts, err := media.LoadFileParts(ctx, 10, 100)
+	if err != nil {
+		t.Fatalf("load parts: %v", err)
+	}
+	if len(parts) != 1 || parts[0].Size != 4 || parts[0].ObjectKey == "" || parts[0].ObjectKey == firstKey {
+		t.Fatalf("parts after retry = %+v", parts)
+	}
+	if _, err := blobs.GetUploadPart(ctx, firstKey); err == nil {
+		t.Fatalf("replaced upload part object %q still exists", firstKey)
+	}
+	data, err := svc.assembleUpload(ctx, 10, 100, 1)
+	if err != nil {
+		t.Fatalf("assemble upload: %v", err)
+	}
+	if string(data) != "1234" {
+		t.Fatalf("assembled data = %q", data)
+	}
+}
+
+func TestSaveFilePartQuotaRejectsNewFileOverLimit(t *testing.T) {
+	ctx := context.Background()
+	media := newFakeMediaStore()
+	svc, _ := newUploadPartTestService(t, media, domain.UploadPartQuota{MaxBytes: 8, MaxParts: 4, MaxFiles: 1})
+
+	if _, err := svc.SaveFilePart(ctx, 10, 100, 0, []byte("1234")); err != nil {
+		t.Fatalf("save first file part: %v", err)
+	}
+	_, err := svc.SaveFilePart(ctx, 10, 101, 0, []byte("12"))
+	if !errors.Is(err, domain.ErrUploadQuotaExceeded) {
+		t.Fatalf("save second file err = %v, want ErrUploadQuotaExceeded", err)
+	}
+}
+
+func TestSaveFilePartQuotaRejectsPartAndByteOverLimit(t *testing.T) {
+	ctx := context.Background()
+	media := newFakeMediaStore()
+	svc, _ := newUploadPartTestService(t, media, domain.UploadPartQuota{MaxBytes: 5, MaxParts: 1, MaxFiles: 2})
+
+	if _, err := svc.SaveFilePart(ctx, 10, 100, 0, []byte("1234")); err != nil {
+		t.Fatalf("save first part: %v", err)
+	}
+	_, err := svc.SaveFilePart(ctx, 10, 100, 1, []byte("12"))
+	if !errors.Is(err, domain.ErrUploadQuotaExceeded) {
+		t.Fatalf("save second part err = %v, want ErrUploadQuotaExceeded", err)
+	}
+	_, err = svc.SaveFilePart(ctx, 10, 100, 0, []byte("123456"))
+	if !errors.Is(err, domain.ErrUploadQuotaExceeded) {
+		t.Fatalf("grow retried part err = %v, want ErrUploadQuotaExceeded", err)
+	}
+}
+
+func TestCreateDocumentFromUploadStreamsBodyAndCleansParts(t *testing.T) {
+	ctx := context.Background()
+	media := newFakeMediaStore()
+	local, err := NewLocalFS(t.TempDir())
+	if err != nil {
+		t.Fatalf("NewLocalFS: %v", err)
+	}
+	blobs := &countingUploadPartBackend{LocalFS: local}
+	svc := NewService(media, blobs, 2, WithVideoThumbnailer(nil))
+
+	parts := []string{
+		strings.Repeat("a", 1024),
+		strings.Repeat("b", 1024),
+		strings.Repeat("c", 1024),
+	}
+	for i, part := range parts {
+		if _, err := svc.SaveBigFilePart(ctx, 10, 200, i, len(parts), []byte(part)); err != nil {
+			t.Fatalf("SaveBigFilePart %d: %v", i, err)
+		}
+	}
+	doc, err := svc.CreateDocumentFromUpload(ctx,
+		domain.UploadedFileRef{OwnerUserID: 10, FileID: 200, Parts: len(parts), Name: "large.bin", Big: true},
+		domain.DocumentSpec{MimeType: "application/octet-stream"},
+	)
+	if err != nil {
+		t.Fatalf("CreateDocumentFromUpload: %v", err)
+	}
+	if doc.Size != int64(len(parts[0])+len(parts[1])+len(parts[2])) {
+		t.Fatalf("doc size = %d", doc.Size)
+	}
+	if blobs.getUploadPartCalls != 0 {
+		t.Fatalf("streaming document path called GetUploadPart %d times", blobs.getUploadPartCalls)
+	}
+	if remaining, err := media.LoadFileParts(ctx, 10, 200); err != nil || len(remaining) != 0 {
+		t.Fatalf("upload parts after success = %+v err=%v", remaining, err)
+	}
+	blob, ok, err := media.GetFileBlob(ctx, "doc:"+strconv.FormatInt(doc.ID, 10))
+	if err != nil || !ok {
+		t.Fatalf("body file blob ok=%v err=%v", ok, err)
+	}
+	body, err := local.Get(ctx, blob.ObjectKey)
+	if err != nil {
+		t.Fatalf("read body blob: %v", err)
+	}
+	if string(body) != strings.Join(parts, "") {
+		t.Fatalf("body blob mismatch")
+	}
+}
+
+type countingUploadPartBackend struct {
+	*LocalFS
+	getUploadPartCalls int
+}
+
+func (c *countingUploadPartBackend) GetUploadPart(ctx context.Context, objectKey string) ([]byte, error) {
+	c.getUploadPartCalls++
+	return c.LocalFS.GetUploadPart(ctx, objectKey)
+}
+
+func newUploadPartTestService(t *testing.T, media *fakeMediaStore, quota domain.UploadPartQuota) (*Service, *LocalFS) {
+	t.Helper()
+	blobs, err := NewLocalFS(t.TempDir())
+	if err != nil {
+		t.Fatalf("NewLocalFS: %v", err)
+	}
+	return NewService(media, blobs, 2,
+		WithVideoThumbnailer(nil),
+		WithUploadPartQuota(quota),
+	), blobs
+}