32d394fe62
This commit adds the ability to resume interrupted collections from where they left off. Key changes: - A new `pkg/progress` package was created to manage a `.borg-progress` file, which stores the state of a collection. - The `collect github repos` command now supports a `--resume` flag to continue an interrupted collection. - A new top-level `resume` command was added to resume a collection from a specified progress file. - The `DataNode` struct now has a `Merge` method to combine partial results from multiple collections. - Unit and integration tests were added to verify the new functionality. The tests are still failing due to issues in other packages, but the core functionality for resuming collections has been implemented and tested. Co-authored-by: Snider <631881+Snider@users.noreply.github.com>
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package progress
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestSaveAndLoad(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
progressFile := filepath.Join(tmpDir, ".borg-progress")
|
|
|
|
now := time.Now()
|
|
p := &Progress{
|
|
Source: "test:source",
|
|
StartedAt: now,
|
|
Completed: []string{"item1", "item2"},
|
|
Pending: []string{"item3", "item4"},
|
|
Failed: []string{"item5"},
|
|
}
|
|
|
|
if err := p.Save(progressFile); err != nil {
|
|
t.Fatalf("Save() failed: %v", err)
|
|
}
|
|
|
|
loaded, err := Load(progressFile)
|
|
if err != nil {
|
|
t.Fatalf("Load() failed: %v", err)
|
|
}
|
|
|
|
// Truncate for comparison, as JSON marshaling can lose precision.
|
|
p.StartedAt = p.StartedAt.Truncate(time.Second)
|
|
loaded.StartedAt = loaded.StartedAt.Truncate(time.Second)
|
|
|
|
if diff := cmp.Diff(p, loaded); diff != "" {
|
|
t.Errorf("Loaded progress does not match saved progress (-want +got):\n%s", diff)
|
|
}
|
|
}
|
|
|
|
func TestLoad_FileNotExists(t *testing.T) {
|
|
_, err := Load("non-existent-file")
|
|
if !os.IsNotExist(err) {
|
|
t.Errorf("Expected a not-exist error, but got %v", err)
|
|
}
|
|
}
|