go/git/repograph/local_test.go

package repograph_test

import (
	"context"
	"os"
	"testing"

	"github.com/stretchr/testify/require"
	"go.skia.org/infra/go/git/repograph"
	"go.skia.org/infra/go/git/repograph/shared_tests"
	git_testutils "go.skia.org/infra/go/git/testutils"
	"go.skia.org/infra/go/testutils"
	"go.skia.org/infra/go/vcsinfo"
)

// localRepoRefresher is a RepoImplRefresher backed by a local git repo.
type localRepoRefresher struct{}

// No-op, since the repo held by Graph is updated by the Graph during Update.
func (u *localRepoRefresher) Refresh(...*vcsinfo.LongCommit) {}

// setupRepo performs common setup for git.Repo based Graphs.
func setupRepo(t *testing.T) (context.Context, *git_testutils.GitBuilder, *repograph.Graph, shared_tests.RepoImplRefresher, func()) {
	ctx, g, cleanup := shared_tests.CommonSetup(t)

	tmp, err := os.MkdirTemp("", "")
	require.NoError(t, err)

	repo, err := repograph.NewLocalGraph(ctx, g.Dir(), tmp)
	require.NoError(t, err)

	return ctx, g, repo, &localRepoRefresher{}, func() {
		testutils.RemoveAll(t, tmp)
		cleanup()
	}
}

func TestGraphWellFormedRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestGraphWellFormed(t, ctx, g, repo, ud)
}

func TestRecurseRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestRecurse(t, ctx, g, repo, ud)
}

func TestRecurseAllBranchesRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestRecurseAllBranches(t, ctx, g, repo, ud)
}

func TestLogLinearRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestLogLinear(t, ctx, g, repo, ud)
}

func TestUpdateHistoryChangedRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestUpdateHistoryChanged(t, ctx, g, repo, ud)
}

func TestUpdateAndReturnCommitDiffsRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestUpdateAndReturnCommitDiffs(t, ctx, g, repo, ud)
}

func TestRevListRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestRevList(t, ctx, g, repo, ud)
}

func TestBranchMembershipRepo(t *testing.T) {
	ctx, g, repo, ud, cleanup := setupRepo(t)
	defer cleanup()
	shared_tests.TestBranchMembership(t, ctx, g, repo, ud)
}