packfiles_test.go

package gitbase

import (
	"os"
	"path/filepath"
	"testing"

	fixtures "github.com/src-d/go-git-fixtures"
	"github.com/stretchr/testify/require"
	"gopkg.in/src-d/go-git.v4/plumbing"
	"gopkg.in/src-d/go-git.v4/plumbing/cache"
)

var (
	testSivaFilePath = filepath.Join("_testdata", "fff7062de8474d10a67d417ccea87ba6f58ca81d.siva")
	testSivaRepoID   = "015dcc49-9049-b00c-ba72-b6f5fa98cbe7"
)

func TestRepositoryPackfiles(t *testing.T) {
	require := require.New(t)

	lib, pool, err := newMultiPool()
	require.NoError(err)

	cwd, err := os.Getwd()
	require.NoError(err)
	p := filepath.Join(cwd, testSivaFilePath)

	err = lib.AddSiva(p, nil)
	require.NoError(err)

	repo, err := pool.GetRepo(testSivaRepoID)
	require.NoError(err)

	f, err := repo.FS()
	require.NoError(err)

	fs, packfiles, err := repositoryPackfiles(f)

	require.NoError(err)
	require.Equal([]plumbing.Hash{
		plumbing.NewHash("433e5205f6e26099e7d34ba5e5306f69e4cef12b"),
		plumbing.NewHash("5d2ce6a45cb07803f9b0c8040e730f5715fc7144"),
	}, packfiles)
	require.NotNil(fs)
}

func TestRepositoryPackfilesNoBare(t *testing.T) {
	require := require.New(t)

	fs := fixtures.ByTag("worktree").One().Worktree()

	dotgit, packfiles, err := repositoryPackfiles(fs)
	require.NoError(err)
	require.Equal([]plumbing.Hash{
		plumbing.NewHash("323a4b6b5de684f9966953a043bc800154e5dbfa"),
	}, packfiles)

	require.NoError(dotgit.Close())
}

func TestGetUnpackedObject(t *testing.T) {
	require := require.New(t)

	fs := fixtures.ByURL("https://github.com/git-fixtures/submodule.git").One().Worktree()
	path := fs.Root()

	lib, err := newMultiLibrary()
	require.NoError(err)
	pool := NewRepositoryPool(cache.NewObjectLRUDefault(), lib)
	require.NoError(lib.AddPlain(path, path, nil))

	r, err := pool.GetRepo(path)
	require.NoError(err)

	obj, err := getUnpackedObject(r, plumbing.NewHash("3bf5d30ad4f23cf517676fee232e3bcb8537c1d0"))
	require.NoError(err)
	require.NotNil(obj)
	require.NoError(r.Close())
}

func TestRepositoryIndex(t *testing.T) {
	lib, pool, err := newMultiPool()
	require.NoError(t, err)

	cwd, err := os.Getwd()
	require.NoError(t, err)
	p := filepath.Join(cwd, testSivaFilePath)

	err = lib.AddSiva(p, nil)
	require.NoError(t, err)

	repo, err := pool.GetRepo(testSivaRepoID)
	require.NoError(t, err)

	idx, err := newRepositoryIndex(repo)
	require.NoError(t, err)

	testCases := []struct {
		hash     string
		offset   int64
		packfile string
	}{
		{
			"52c853392c25d3a670446641f4b44b22770b3bbe",
			3046713,
			"5d2ce6a45cb07803f9b0c8040e730f5715fc7144",
		},
		{
			"aa7ef7dafd292737ed493b7d74c0abfa761344f4",
			3046902,
			"5d2ce6a45cb07803f9b0c8040e730f5715fc7144",
		},
	}

	for _, tt := range testCases {
		t.Run(tt.hash, func(t *testing.T) {
			offset, packfile, err := idx.find(plumbing.NewHash(tt.hash))
			require.NoError(t, err)
			require.Equal(t, tt.offset, offset)
			require.Equal(t, tt.packfile, packfile.String())
		})
	}
}