github.com/mhristof/go-stacks@v0.1.3/git/branches_test.go

package git

import (
	"io/ioutil"
	"os"
	"strings"
	"testing"

	"github.com/MakeNowJust/heredoc"
	"github.com/mhristof/go-stacks/bash"
	"github.com/stretchr/testify/assert"
)

func mkgit(t *testing.T, commands []string) string {
	dir, err := ioutil.TempDir("", "git")
	if err != nil {
		panic(err)
	}

	err = os.Chdir(dir)
	if err != nil {
		panic(err)
	}

	for _, command := range commands {
		_, _, err = bash.Run(command)
		if err != nil {
			t.Fatal(err)
		}

	}
	return dir
}

func TestBranches(t *testing.T) {
	var cases = []struct {
		name     string
		branches []string
		path     string
	}{
		{
			name: "non git folder",
			path: mkgit(t, []string{}),
		},
		{
			name: "git folder with main",
			path: mkgit(t, []string{
				"git init",
				"git commit --allow-empty -m 'empty.commit'",
			}),
			branches: []string{
				"main",
			},
		},
		{
			name: "git folder with a couple of branches",
			path: mkgit(t, []string{
				"git init",
				"git commit --allow-empty -m 'empty.commit'",
				"git checkout -b foobar",
				"git checkout -b foobar1",
			}),
			branches: []string{
				"foobar", "foobar1", "main",
			},
		},
	}

	for _, test := range cases {
		assert.Equal(t, test.branches, Branches(test.path), test.name) //nolint
		defer os.Remove(test.path)
	}
}

func TestRebase(t *testing.T) {
	var cases = []struct {
		name   string
		path   string
		branch string
		gitLog string
	}{
		{
			name:   "main ahead of feat1",
			branch: ".*",
			path: mkgit(t, []string{
				"git init",
				"git commit --allow-empty -m 'empty.commit'",
				"git checkout -b feat1",
				"git commit --allow-empty -m 'feat1.commit'",
				"git checkout main",
				"git commit --allow-empty -m 'main.commit'",
			}),
			gitLog: heredoc.Doc(`
				*  (HEAD -> feat1).'feat1.commit'
				*  (main).'main.commit'
				* .'empty.commit'`),
		},
		{
			name:   "feat1 ahead of feat1.1",
			branch: ".*",
			path: mkgit(t, []string{
				"git init",
				"git commit --allow-empty -m 'empty.commit'",
				"git commit --allow-empty -m 'empty.commit1'",
				"git checkout -b feat1",
				"git commit --allow-empty -m 'feat1.commit'",
				"git checkout -b feat1.1",
				"git commit --allow-empty -m 'feat1.1.commit'",
				"git checkout feat1",
				"git commit --allow-empty -m 'feat1.commit2'",
			}),
			gitLog: heredoc.Doc(`
				*  (HEAD -> feat1.1).'feat1.1.commit'
				*  (feat1).'feat1.commit2'
				* .'feat1.commit'
				* .'empty.commit1'
				*  (main).'empty.commit1'
				* .'empty.commit'`),
		},
		{
			name:   "no changes",
			branch: ".*",
			path: mkgit(t, []string{
				"git init",
				"git commit --allow-empty -m 'empty.commit'",
				"git commit --allow-empty -m 'empty.commit1'",
				"git checkout -b feat1",
				"git commit --allow-empty -m 'feat1.commit'",
				"git checkout -b feat1.1",
				"git commit --allow-empty -m 'feat1.1.commit'",
			}),
			gitLog: heredoc.Doc(`
				*  (HEAD -> feat1.1).'feat1.1.commit'
				*  (feat1).'feat1.commit'
				* .'empty.commit1'
				*  (main).'empty.commit1'
				* .'empty.commit'`),
		},
		{
			name:   "limit branch name",
			branch: "feat1",
			path: mkgit(t, strings.Split(heredoc.Doc(`
				git init
				git commit --allow-empty -m 'empty.commit'
				git commit --allow-empty -m 'empty.commit1'
				git checkout -b feat1
				git commit --allow-empty -m 'feat1.commit'
				git checkout -b feat1.1
				git commit --allow-empty -m 'feat1.1.commit'
				git checkout feat1
				git commit --allow-empty -m 'feat1.commit1'
				git checkout main
				git checkout -b feat2
				git commit --allow-empty -m 'feat2.commit1'
				git checkout -b feat2.1
				git commit --allow-empty -m 'feat2.1.commit1'
				git checkout feat2
				git commit --allow-empty -m 'feat2.commit2'`),
				"\n")),
			gitLog: heredoc.Doc(`
				*  (HEAD -> feat1.1).'feat1.1.commit'
				*  (feat1).'feat1.commit1'
				* .'feat1.commit'
				* .'empty.commit1'
				| *  (feat2).'feat2.commit2'
				| | *  (feat2.1).'feat2.1.commit1'
				| |/
				| * .'feat2.commit1'
				|/
				*  (main).'empty.commit1'
				* .'empty.commit'`),
		},
	}

	for _, test := range cases {
		err := os.Chdir(test.path)
		if err != nil {
			panic(err)
		}

		commands, err := Rebase(test.path, test.branch) //nolint
		if err != nil {
			panic(err)
		}

		for _, command := range commands {
			_, _, err := bash.Run(command)
			if err != nil {
				t.Fatal(err)
			}

		}
		stdout, _, err := bash.Run(`git log --graph --pretty=format:%d.%s --all`)
		if err != nil {
			t.Fatal(err)
		}

		assert.Equal(t, test.gitLog, strings.ReplaceAll(stdout, "/  ", "/"), test.name)
		defer os.Remove(test.path)
	}
}