github.com/qri-io/qri@v0.10.1-0.20220104210721-c771715036cb/repo/test/spec/test_refstore.go

package spec

import (
	"context"
	"fmt"
	"testing"

	"github.com/qri-io/qfs"
	"github.com/qri-io/qri/profile"
	"github.com/qri-io/qri/repo"
	reporef "github.com/qri-io/qri/repo/ref"
)

func testRefstoreInvalidRefs(t *testing.T, rmf RepoMakerFunc) {
	r, cleanup := rmf(t)
	defer cleanup()

	err := r.PutRef(reporef.DatasetRef{Name: "a", Path: "/path/to/a/thing"})
	if err != repo.ErrPeerIDRequired {
		t.Errorf("attempting to put empty peerID in refstore should return repo.ErrPeerIDRequired, got: %s", err)
		return
	}

	err = r.PutRef(reporef.DatasetRef{ProfileID: profile.IDRawByteString("badProfileID"), Peername: "peer", Path: "/path/to/a/thing"})
	if err != repo.ErrNameRequired {
		t.Errorf("attempting to put empty name in refstore should return repo.ErrNameRequired, got: %s", err)
		return
	}

	err = r.PutRef(reporef.DatasetRef{ProfileID: profile.IDRawByteString("badProfileID"), Peername: "peer", Name: "a", Path: ""})
	if err != repo.ErrPathRequired {
		t.Errorf("attempting to put empty path in refstore should return repo.ErrPathRequired, got: %s", err)
		return
	}

	return
}

func testRefstoreRefs(t *testing.T, rmf RepoMakerFunc) {
	ctx := context.Background()
	r, cleanup := rmf(t)
	defer cleanup()

	path, err := r.Filesystem().DefaultWriteFS().Put(ctx, qfs.NewMemfileBytes("test", []byte(`{ "title": "test data" }`)))
	if err != nil {
		t.Errorf("error putting test file in datastore: %s", err.Error())
		return
	}

	ref := reporef.DatasetRef{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Name: "test", Path: path, Peername: "peer"}

	if err := r.PutRef(ref); err != nil {
		t.Errorf("repo.PutName: %s", err.Error())
		return
	}

	res, err := r.GetRef(reporef.DatasetRef{ProfileID: ref.ProfileID, Name: ref.Name})
	if err != nil {
		t.Errorf("repo.GetRef with peerID/name: %s, ref: %s", err.Error(), reporef.DatasetRef{ProfileID: ref.ProfileID, Name: ref.Name})
		return
	}
	if !ref.Equal(res) {
		t.Errorf("repo.GetRef with peerID/name response mistmatch. expected: %s, got: %s", ref, res)
		return
	}

	res, err = r.GetRef(reporef.DatasetRef{Path: ref.Path})
	if err != nil {
		t.Errorf("repo.GetRef with path: %s", err.Error())
		return
	}
	if !ref.Equal(res) {
		t.Errorf("repo.GetRef with path response mismatch. expected: %s, got: %s", ref, res)
		return
	}

	if err := r.DeleteRef(ref); err != nil {
		t.Errorf("repo.DeleteName: %s", err.Error())
		return
	}

	_, err = r.GetRef(ref)
	if err != repo.ErrNotFound {
		t.Errorf("repo.GetRef where ref is deleted should return ErrNotFound")
		return
	}
	err = nil

	if err := r.Filesystem().DefaultWriteFS().Delete(ctx, ref.Path); err != nil {
		t.Errorf("error removing file from store")
		return
	}
	return
}

func testRefstoreMain(t *testing.T, rmf RepoMakerFunc) {
	ctx := context.Background()
	r, cleanup := rmf(t)
	defer cleanup()

	refs := []reporef.DatasetRef{
		{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Peername: "peer", Name: "test_namespace_a", Published: true},
		{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Peername: "peer", Name: "test_namespace_b"},
		{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Peername: "peer", Name: "test_namespace_c"},
		{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Peername: "peer", Name: "test_namespace_d"},
		{ProfileID: profile.IDB58MustDecode("QmZePf5LeXow3RW5U1AgEiNbW46YnRGhZ7HPvm1UmPFPwt"), Peername: "peer", Name: "test_namespace_e", Published: true},
	}
	for i, ref := range refs {
		path, err := r.Filesystem().DefaultWriteFS().Put(ctx, qfs.NewMemfileBytes("test", []byte(fmt.Sprintf(`{ "title": "test_dataset_%s" }`, ref.Name))))
		if err != nil {
			t.Errorf("error putting test file in cafs: %s", err.Error())
			return
		}

		ref.Path = path
		// set path on input refs for later comparison
		refs[i].Path = ref.Path

		if err := r.PutRef(ref); err != nil {
			t.Errorf("error putting name in repo for namespace test: %s", err.Error())
			return
		}
	}

	count, err := r.RefCount()
	if err != nil {
		t.Errorf("repo.NameCount: %s", err.Error())
		return
	}
	if count != len(refs) {
		t.Errorf("repo.NameCount should have returned %d results", len(refs))
		return
	}

	names := []reporef.DatasetRef{}
	pages := count
	pageSize := count / pages
	for i := 0; i <= pages; i++ {
		res, err := r.References(i*pageSize, pageSize)
		if err != nil {
			t.Errorf("repo.References(%d,%d): %s", i*pageSize, pageSize, err.Error())
			return
		}
		names = append(names, res...)
	}
	if len(names) != count {
		t.Errorf("failed to read all paginated names. expected %d results, got %d", count, len(names))
		return
	}

	idxs := map[string]int{}
	for i, ref := range names {
		idxs[ref.Name] = i
		if err := repo.CompareDatasetRef(refs[i], names[i]); err != nil {
			t.Errorf("ref %d error: %s", i, err)
		}
	}
	for i, ref := range refs {
		if i > 0 {
			if idxs[ref.Name] < idxs[refs[i-1].Name] {
				t.Errorf("expected results to be returned in lexographical order. %s:%d, %s:%d", ref.Name, idxs[ref.Name], refs[i-1].Name, idxs[refs[i-1].Name])
				return
			}
		}
	}

	refs[0].Published = false
	if err := r.PutRef(refs[0]); err != nil {
		t.Errorf("updating existing ref err: %s", err)
	}

	unpublished, err := r.GetRef(refs[0])
	if err != nil {
		t.Error(err)
	}
	if unpublished.Published {
		t.Error("expected setting published value to be retained")
	}

	for _, ref := range refs {
		if err := r.Filesystem().DefaultWriteFS().Delete(ctx, ref.Path); err != nil {
			t.Errorf("error removing path from repo store: %s", err.Error())
			return
		}
		if err := r.DeleteRef(ref); err != nil {
			t.Errorf("error removing test name from namespace: %s", err.Error())
			return
		}
	}
	return
}