mirror of
https://github.com/kopia/kopia.git
synced 2026-01-05 21:17:52 -05:00
98f3473b67
* refactor(snapshots): extracted snapshotfs.Verifier component * refactor(repository): added tests for snapshotfs.Verifier, misc cleanups * fixed data race * fixed atomic alignment * nit
90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package server_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/kopia/kopia/internal/apiclient"
|
|
"github.com/kopia/kopia/internal/clock"
|
|
"github.com/kopia/kopia/internal/serverapi"
|
|
"github.com/kopia/kopia/internal/testlogging"
|
|
"github.com/kopia/kopia/internal/uitask"
|
|
"github.com/kopia/kopia/snapshot"
|
|
"github.com/kopia/kopia/snapshot/policy"
|
|
)
|
|
|
|
func mustCreateSource(t *testing.T, cli *apiclient.KopiaAPIClient, path string, pol *policy.Policy) {
|
|
t.Helper()
|
|
|
|
_, err := serverapi.CreateSnapshotSource(testlogging.Context(t), cli, &serverapi.CreateSnapshotSourceRequest{
|
|
Path: path,
|
|
Policy: pol,
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func mustSetPolicy(t *testing.T, cli *apiclient.KopiaAPIClient, si snapshot.SourceInfo, pol *policy.Policy) {
|
|
t.Helper()
|
|
|
|
require.NoError(t, serverapi.SetPolicy(testlogging.Context(t), cli, si, pol))
|
|
}
|
|
|
|
func mustListSources(t *testing.T, cli *apiclient.KopiaAPIClient, match *snapshot.SourceInfo) []*serverapi.SourceStatus {
|
|
t.Helper()
|
|
|
|
resp, err := serverapi.ListSources(testlogging.Context(t), cli, match)
|
|
require.NoError(t, err)
|
|
|
|
return resp.Sources
|
|
}
|
|
|
|
func mustGetTask(t *testing.T, cli *apiclient.KopiaAPIClient, taskID string) uitask.Info {
|
|
t.Helper()
|
|
|
|
resp, err := serverapi.GetTask(testlogging.Context(t), cli, taskID)
|
|
require.NoError(t, err)
|
|
|
|
return *resp
|
|
}
|
|
|
|
func mustListTasks(t *testing.T, cli *apiclient.KopiaAPIClient) []uitask.Info {
|
|
t.Helper()
|
|
|
|
resp, err := serverapi.ListTasks(testlogging.Context(t), cli)
|
|
require.NoError(t, err)
|
|
|
|
return resp.Tasks
|
|
}
|
|
|
|
func mustGetLatestTask(t *testing.T, cli *apiclient.KopiaAPIClient) uitask.Info {
|
|
t.Helper()
|
|
|
|
tl := mustListTasks(t, cli)
|
|
require.NotEmpty(t, tl)
|
|
|
|
return tl[0]
|
|
}
|
|
|
|
func waitForTask(t *testing.T, cli *apiclient.KopiaAPIClient, taskID string, timeout time.Duration) uitask.Info {
|
|
t.Helper()
|
|
|
|
var lastInfo uitask.Info
|
|
|
|
deadline := clock.Now().Add(timeout)
|
|
for clock.Now().Before(deadline) {
|
|
lastInfo = mustGetTask(t, cli, taskID)
|
|
|
|
if lastInfo.Status.IsFinished() {
|
|
return lastInfo
|
|
}
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
|
|
t.Fatalf("task %v did not complete in %v, last: %v", taskID, timeout, lastInfo)
|
|
|
|
return lastInfo
|
|
}
|