Files
kopia/internal/repotesting/repotesting_test.go
Julio López 70df5f738c testing: Refactor faketime (#597)
* Allow auto-advance in faketime.TimeAdvance
* Leverage TimeAdvance in faketime.AutoAdvance
* Concurrent test for faketime.AdvanceTime
2020-09-10 00:52:14 -07:00

100 lines
2.6 KiB
Go

package repotesting
import (
"testing"
"time"
"github.com/kopia/kopia/internal/faketime"
"github.com/kopia/kopia/internal/mockfs"
"github.com/kopia/kopia/internal/testlogging"
"github.com/kopia/kopia/repo"
"github.com/kopia/kopia/snapshot"
"github.com/kopia/kopia/snapshot/policy"
"github.com/kopia/kopia/snapshot/snapshotfs"
)
func TestTimeFuncWiring(t *testing.T) {
var env Environment
ctx := testlogging.Context(t)
defer env.Setup(t).Close(ctx, t)
env.Repository.Close(ctx)
ft := faketime.NewTimeAdvance(time.Date(2018, time.February, 6, 0, 0, 0, 0, time.UTC), 0)
// Re open with injected time
rep, err := repo.Open(ctx, env.Repository.ConfigFile, masterPassword, &repo.Options{TimeNowFunc: ft.NowFunc()})
if err != nil {
t.Fatal("Failed to open repo:", err)
}
r := rep.(*repo.DirectRepository)
env.Repository = r
// verify wiring for the repo layer
if got, want := r.Time(), ft.NowFunc()(); !got.Equal(want) {
t.Errorf("times don't match, got %v, want %v", got, want)
}
if want, got := ft.Advance(10*time.Minute), r.Time(); !got.Equal(want) {
t.Errorf("times don't match, got %v, want %v", got, want)
}
// verify wiring for the content layer
nt := ft.Advance(20 * time.Second)
cid, err := r.Content.WriteContent(ctx, []byte("foo"), "")
if err != nil {
t.Fatal("failed to write content:", err)
}
info, err := r.Content.ContentInfo(ctx, cid)
if err != nil {
t.Fatal("failed to get content info for", cid, err)
}
if got, want := info.Timestamp(), nt; !got.Equal(want) {
t.Errorf("content time does not match, got %v, want %v", got, want)
}
// verify wiring for the manifest layer
nt = ft.Advance(3 * time.Minute)
labels := map[string]string{"l1": "v1", "l2": "v2", "type": "my-manifest"}
mid, err := r.Manifests.Put(ctx, labels, "manifest content")
if err != nil {
t.Fatal("failed to put manifest:", err)
}
meta, err := r.Manifests.GetMetadata(ctx, mid)
if err != nil {
t.Fatal("failed to get manifest metadata:", err)
}
if got, want := meta.ModTime, nt; !got.Equal(want) {
t.Errorf("manifest time does not match, got %v, want %v", got, want)
}
const defaultPermissions = 0o777
// verify wiring for the snapshot layer
sourceDir := mockfs.NewDirectory()
sourceDir.AddFile("f1", []byte{1, 2, 3}, defaultPermissions)
nt = ft.Advance(1 * time.Hour)
u := snapshotfs.NewUploader(r)
policyTree := policy.BuildTree(nil, policy.DefaultPolicy)
s1, err := u.Upload(ctx, sourceDir, policyTree, snapshot.SourceInfo{})
if err != nil {
t.Fatal("failed to create snapshot:", err)
}
if got, want := nt, s1.StartTime; !got.Equal(want) {
t.Fatalf("snapshot time does not match, got: %v, want: %v", got, want)
}
}