Test running git-sizer from a subdirectory of the repository

Author: mhagger

git_sizer_test.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"os"
 	"os/exec"
+	"path/filepath"
 	"testing"
 	"time"
 
@@ -30,6 +31,24 @@ func gitCommand(t *testing.T, repo *git.Repository, args ...string) *exec.Cmd {
 	return cmd
 }
 
+func addFile(t *testing.T, repoPath string, repo *git.Repository, relativePath, contents string) {
+	dirPath := filepath.Dir(relativePath)
+	if dirPath != "." {
+		require.NoError(t, os.MkdirAll(filepath.Join(repoPath, dirPath), 0777), "creating subdir")
+	}
+
+	filename := filepath.Join(repoPath, relativePath)
+	f, err := os.Create(filename)
+	require.NoErrorf(t, err, "creating file %q", filename)
+	_, err = f.WriteString(contents)
+	require.NoErrorf(t, err, "writing to file %q", filename)
+	require.NoErrorf(t, f.Close(), "closing file %q", filename)
+
+	cmd := gitCommand(t, repo, "add", relativePath)
+	cmd.Dir = repoPath
+	require.NoErrorf(t, cmd.Run(), "adding file %q", relativePath)
+}
+
 func addAuthorInfo(cmd *exec.Cmd, timestamp *time.Time) {
 	cmd.Env = append(cmd.Env,
 		"GIT_AUTHOR_NAME=Arthur",
@@ -217,3 +236,33 @@ func TestTaggedTags(t *testing.T) {
 	require.NoError(t, err, "scanning repository")
 	assert.Equal(t, counts.Count32(3), h.MaxTagDepth, "tag depth")
 }
+
+func TestFromSubdir(t *testing.T) {
+	t.Parallel()
+	path, err := ioutil.TempDir("", "subdir")
+	require.NoError(t, err, "creating temporary directory")
+
+	defer func() {
+		os.RemoveAll(path)
+	}()
+
+	cmd := exec.Command("git", "init", path)
+	require.NoError(t, cmd.Run(), "initializing repo")
+	repo, err := git.NewRepository(path)
+	require.NoError(t, err, "initializing Repository object")
+
+	timestamp := time.Unix(1112911993, 0)
+
+	addFile(t, path, repo, "subdir/file.txt", "Hello, world!\n")
+
+	cmd = gitCommand(t, repo, "commit", "-m", "initial")
+	addAuthorInfo(cmd, &timestamp)
+	require.NoError(t, cmd.Run(), "creating commit")
+
+	repo2, err := git.NewRepository(filepath.Join(path, "subdir"))
+	require.NoError(t, err, "creating Repository object in subdirectory")
+	_, err = sizes.ScanRepositoryUsingGraph(
+		repo2, git.AllReferencesFilter, sizes.NameStyleNone, false,
+	)
+	require.NoError(t, err, "scanning repository")
+}