mirror of https://github.com/octoleo/restic.git
82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package restic_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/restic/restic/internal/checker"
|
|
"github.com/restic/restic/internal/repository"
|
|
"github.com/restic/restic/internal/restic"
|
|
)
|
|
|
|
var testSnapshotTime = time.Unix(1460289341, 207401672)
|
|
|
|
const (
|
|
testCreateSnapshots = 3
|
|
testDepth = 2
|
|
)
|
|
|
|
// LoadAllSnapshots returns a list of all snapshots in the repo.
|
|
// If a snapshot ID is in excludeIDs, it will not be included in the result.
|
|
func loadAllSnapshots(ctx context.Context, repo restic.Repository, excludeIDs restic.IDSet) (snapshots restic.Snapshots, err error) {
|
|
err = restic.ForAllSnapshots(ctx, repo.Backend(), repo, excludeIDs, func(id restic.ID, sn *restic.Snapshot, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
snapshots = append(snapshots, sn)
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return snapshots, nil
|
|
}
|
|
|
|
func TestCreateSnapshot(t *testing.T) {
|
|
repo, cleanup := repository.TestRepository(t)
|
|
defer cleanup()
|
|
|
|
for i := 0; i < testCreateSnapshots; i++ {
|
|
restic.TestCreateSnapshot(t, repo, testSnapshotTime.Add(time.Duration(i)*time.Second), testDepth, 0)
|
|
}
|
|
|
|
snapshots, err := loadAllSnapshots(context.TODO(), repo, restic.NewIDSet())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(snapshots) != testCreateSnapshots {
|
|
t.Fatalf("got %d snapshots, expected %d", len(snapshots), 1)
|
|
}
|
|
|
|
sn := snapshots[0]
|
|
if sn.Time.Before(testSnapshotTime) || sn.Time.After(testSnapshotTime.Add(testCreateSnapshots*time.Second)) {
|
|
t.Fatalf("timestamp %v is outside of the allowed time range", sn.Time)
|
|
}
|
|
|
|
if sn.Tree == nil {
|
|
t.Fatalf("tree id is nil")
|
|
}
|
|
|
|
if sn.Tree.IsNull() {
|
|
t.Fatalf("snapshot has zero tree ID")
|
|
}
|
|
|
|
checker.TestCheckRepo(t, repo)
|
|
}
|
|
|
|
func BenchmarkTestCreateSnapshot(t *testing.B) {
|
|
repo, cleanup := repository.TestRepository(t)
|
|
defer cleanup()
|
|
|
|
t.ResetTimer()
|
|
|
|
for i := 0; i < t.N; i++ {
|
|
restic.TestCreateSnapshot(t, repo, testSnapshotTime.Add(time.Duration(i)*time.Second), testDepth, 0)
|
|
}
|
|
}
|