forked from TrueCloudLab/restic
138 lines
3 KiB
Go
138 lines
3 KiB
Go
package khepri_test
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/fd0/khepri"
|
|
)
|
|
|
|
var testCleanup = flag.Bool("test.cleanup", true, "clean up after running tests (remove repository directory with all content)")
|
|
|
|
var TestStrings = []struct {
|
|
id string
|
|
t khepri.Type
|
|
data string
|
|
}{
|
|
{"c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2", khepri.TypeBlob, "foobar"},
|
|
{"248d6a61d20638b8e5c026930c3e6039a33ce45964ff2167f6ecedd419db06c1", khepri.TypeBlob, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"},
|
|
{"cc5d46bdb4991c6eae3eb739c9c8a7a46fe9654fab79c47b4fe48383b5b25e1c", khepri.TypeRef, "foo/bar"},
|
|
{"4e54d2c721cbdb730f01b10b62dec622962b36966ec685880effa63d71c808f2", khepri.TypeBlob, "foo/../../baz"},
|
|
}
|
|
|
|
func setupRepo() (*khepri.DirRepository, error) {
|
|
tempdir, err := ioutil.TempDir("", "khepri-test-")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
repo, err := khepri.NewDirRepository(tempdir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return repo, nil
|
|
}
|
|
|
|
func teardownRepo(repo *khepri.DirRepository) error {
|
|
if !*testCleanup {
|
|
fmt.Fprintf(os.Stderr, "leaving repository at %s\n", repo.Path())
|
|
return nil
|
|
}
|
|
|
|
err := os.RemoveAll(repo.Path())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestRepository(t *testing.T) {
|
|
repo, err := setupRepo()
|
|
ok(t, err)
|
|
|
|
defer func() {
|
|
err = teardownRepo(repo)
|
|
ok(t, err)
|
|
}()
|
|
|
|
// detect non-existing files
|
|
for _, test := range TestStrings {
|
|
id, err := khepri.ParseID(test.id)
|
|
ok(t, err)
|
|
|
|
// try to get string out, should fail
|
|
ret, err := repo.Test(test.t, id)
|
|
ok(t, err)
|
|
assert(t, !ret, fmt.Sprintf("id %q was found (but should not have)", test.id))
|
|
}
|
|
|
|
// add files
|
|
for _, test := range TestStrings {
|
|
// store string in repository
|
|
id, err := repo.Put(test.t, strings.NewReader(test.data))
|
|
ok(t, err)
|
|
equals(t, test.id, id.String())
|
|
|
|
// try to get it out again
|
|
rd, err := repo.Get(test.t, id)
|
|
ok(t, err)
|
|
assert(t, rd != nil, "Get() returned nil reader")
|
|
|
|
// compare content
|
|
buf, err := ioutil.ReadAll(rd)
|
|
equals(t, test.data, string(buf))
|
|
}
|
|
|
|
// add buffer
|
|
for _, test := range TestStrings {
|
|
// store buf in repository
|
|
id, err := repo.PutRaw(test.t, []byte(test.data))
|
|
ok(t, err)
|
|
equals(t, test.id, id.String())
|
|
}
|
|
|
|
// list ids
|
|
for _, tpe := range []khepri.Type{khepri.TypeBlob, khepri.TypeRef} {
|
|
IDs := khepri.IDs{}
|
|
for _, test := range TestStrings {
|
|
if test.t == tpe {
|
|
id, err := khepri.ParseID(test.id)
|
|
ok(t, err)
|
|
IDs = append(IDs, id)
|
|
}
|
|
}
|
|
|
|
ids, err := repo.ListIDs(tpe)
|
|
ok(t, err)
|
|
|
|
sort.Sort(ids)
|
|
sort.Sort(IDs)
|
|
equals(t, IDs, ids)
|
|
}
|
|
|
|
// remove content if requested
|
|
if *testCleanup {
|
|
for _, test := range TestStrings {
|
|
id, err := khepri.ParseID(test.id)
|
|
ok(t, err)
|
|
|
|
found, err := repo.Test(test.t, id)
|
|
ok(t, err)
|
|
assert(t, found, fmt.Sprintf("id %q was not found before removal"))
|
|
|
|
err = repo.Remove(test.t, id)
|
|
ok(t, err)
|
|
|
|
found, err = repo.Test(test.t, id)
|
|
ok(t, err)
|
|
assert(t, !found, fmt.Sprintf("id %q was not found before removal"))
|
|
}
|
|
}
|
|
}
|