mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-08-15 01:14:00 +08:00
0e6ed66ef5
* feat: show upload progress for file tasks * feat: show upload progress for batch file tasks * style: distinguish download and upload phases * style: mark successful task completion * fix: preserve storage save error context * fix: serialize single-file progress updates * fix: stabilize batch upload progress reporting Correct batch upload totals, completion state, actual file sizes, and synchronized progress snapshots. Add regression coverage for concurrent updates and phase transitions. * feat: show per-file transfer progress Format single-file and batch download and upload states with Telegram entities, blockquotes, speeds, transferred sizes, progress bars, concise counters, and accurate confirmation handling. Cover upload retries plus final, error, and cancellation messages with regression tests. * chore: remove transfer progress tests * test: restore critical transfer progress coverage * test: cover interleaved batch transfers * fix: declare progress styles in locale templates
81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
package telegram
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gotd/td/telegram/uploader"
|
|
"github.com/krau/SaveAny-Bot/pkg/storagetypes"
|
|
)
|
|
|
|
func TestUploadProgressAggregatesUploaderParts(t *testing.T) {
|
|
type update struct {
|
|
uploaded int64
|
|
total int64
|
|
}
|
|
var updates []update
|
|
progress := newUploadProgress(100, func(uploaded, total int64) {
|
|
updates = append(updates, update{uploaded: uploaded, total: total})
|
|
})
|
|
|
|
states := []uploader.ProgressState{
|
|
{ID: 1, Uploaded: 20, Total: 60},
|
|
{ID: 1, Uploaded: 20, Total: 60},
|
|
{ID: 1, Uploaded: 60, Total: 60},
|
|
{ID: 2, Uploaded: 10, Total: 40},
|
|
{ID: 2, Uploaded: 40, Total: 40},
|
|
}
|
|
for _, state := range states {
|
|
if err := progress.Chunk(context.Background(), state); err != nil {
|
|
t.Fatalf("Chunk() failed: %v", err)
|
|
}
|
|
}
|
|
|
|
want := []update{{20, 100}, {60, 100}, {70, 100}, {100, 100}}
|
|
if len(updates) != len(want) {
|
|
t.Fatalf("got %d updates, want %d", len(updates), len(want))
|
|
}
|
|
for i := range want {
|
|
if updates[i] != want[i] {
|
|
t.Fatalf("update %d = %+v, want %+v", i, updates[i], want[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestUploadProgressResetForSplitFiles(t *testing.T) {
|
|
var uploaded, total int64
|
|
progress := newUploadProgress(100, func(current, size int64) {
|
|
uploaded, total = current, size
|
|
})
|
|
if err := progress.Chunk(context.Background(), uploader.ProgressState{ID: 1, Uploaded: 100, Total: 100}); err != nil {
|
|
t.Fatalf("Chunk() failed: %v", err)
|
|
}
|
|
|
|
progress.reset(120)
|
|
if err := progress.Chunk(context.Background(), uploader.ProgressState{ID: 2, Uploaded: 30, Total: 60}); err != nil {
|
|
t.Fatalf("Chunk() after reset failed: %v", err)
|
|
}
|
|
if uploaded != 30 || total != 120 {
|
|
t.Fatalf("progress after reset = %d/%d, want 30/120", uploaded, total)
|
|
}
|
|
}
|
|
|
|
func TestBatchItemUploadProgressPreservesItemIndex(t *testing.T) {
|
|
var gotIndex int
|
|
var gotUploaded, gotTotal int64
|
|
progress := batchItemUploadProgress(batchMediaItem{
|
|
index: 4,
|
|
item: storagetypes.BatchItem{Size: 100},
|
|
}, func(index int, uploaded, total int64) {
|
|
gotIndex = index
|
|
gotUploaded = uploaded
|
|
gotTotal = total
|
|
})
|
|
if err := progress.Chunk(context.Background(), uploader.ProgressState{ID: 1, Uploaded: 25, Total: 100}); err != nil {
|
|
t.Fatalf("Chunk() failed: %v", err)
|
|
}
|
|
if gotIndex != 4 || gotUploaded != 25 || gotTotal != 100 {
|
|
t.Fatalf("batch progress = index %d, %d/%d; want index 4, 25/100", gotIndex, gotUploaded, gotTotal)
|
|
}
|
|
}
|