repo_reference: add unit tests

Author: unknwon

repo_reference.go

@@ -58,7 +58,7 @@ func (r *Repository) ShowRefVerify(ref string, opts ...ShowRefVerifyOptions) (st
 	return strings.Split(string(stdout), " ")[0], nil
 }
 
-// HasBranch returns true if given branch exists in the repository.
+// HasReference returns true if given reference exists in the repository.
 func (r *Repository) HasReference(ref string, opts ...ShowRefVerifyOptions) bool {
 	_, err := r.ShowRefVerify(ref, opts...)
 	return err == nil
@@ -125,8 +125,8 @@ func (r *Repository) ShowRef(opts ...ShowRefOptions) ([]*Reference, error) {
 	if opt.Tags {
 		cmd.AddArgs("--tags")
 	}
+	cmd.AddArgs("--")
 	if len(opt.Patterns) > 0 {
-		cmd.AddArgs("--")
 		cmd.AddArgs(opt.Patterns...)
 	}
 

repo_reference_test.go

@@ -0,0 +1,179 @@
+package git
+
+import (
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRefShortName(t *testing.T) {
+	tests := []struct {
+		ref    string
+		expVal string
+	}{
+		{
+			ref:    "refs/heads/master",
+			expVal: "master",
+		},
+		{
+			ref:    "refs/tags/v1.0.0",
+			expVal: "v1.0.0",
+		},
+		{
+			ref:    "refs/pull/98",
+			expVal: "refs/pull/98",
+		},
+	}
+	for _, test := range tests {
+		t.Run("", func(t *testing.T) {
+			assert.Equal(t, test.expVal, RefShortName(test.ref))
+		})
+	}
+}
+
+func TestRepository_HasReference(t *testing.T) {
+	tests := []struct {
+		ref    string
+		opt    ShowRefVerifyOptions
+		expVal bool
+	}{
+		{
+			ref:    RefsHeads + "master",
+			expVal: true,
+		},
+		{
+			ref:    "master",
+			expVal: false,
+		},
+	}
+	for _, test := range tests {
+		t.Run("", func(t *testing.T) {
+			assert.Equal(t, test.expVal, testrepo.HasReference(test.ref, test.opt))
+		})
+	}
+}
+
+func TestRepository_SymbolicRef(t *testing.T) {
+	r, cleanup, err := setupTempRepo()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cleanup()
+
+	// Get HEAD
+	ref, err := r.SymbolicRef()
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.Equal(t, RefsHeads+"master", ref)
+
+	// Set a symbolic reference
+	_, err = r.SymbolicRef(SymbolicRefOptions{
+		Name: "TEST-REF",
+		Ref:  RefsHeads + "develop",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Get the symbolic reference we just set
+	ref, err = r.SymbolicRef(SymbolicRefOptions{
+		Name: "TEST-REF",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.Equal(t, RefsHeads+"develop", ref)
+}
+
+func TestRepository_ShowRef(t *testing.T) {
+	tests := []struct {
+		opt     ShowRefOptions
+		expRefs []*Reference
+	}{
+		{
+			opt: ShowRefOptions{
+				Heads:    true,
+				Patterns: []string{"release-1.0"},
+			},
+			expRefs: []*Reference{
+				{
+					ID:      "0eedd79eba4394bbef888c804e899731644367fe",
+					Refspec: "refs/heads/release-1.0",
+				},
+			},
+		},
+		{
+			opt: ShowRefOptions{
+				Tags:     true,
+				Patterns: []string{"v1.0.0"},
+			},
+			expRefs: []*Reference{
+				{
+					ID:      "0eedd79eba4394bbef888c804e899731644367fe",
+					Refspec: "refs/tags/v1.0.0",
+				},
+			},
+		},
+	}
+	for _, test := range tests {
+		t.Run("", func(t *testing.T) {
+			refs, err := testrepo.ShowRef(test.opt)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			assert.Equal(t, test.expRefs, refs)
+		})
+	}
+}
+
+func TestRepository_DeleteBranch(t *testing.T) {
+	r, cleanup, err := setupTempRepo()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer cleanup()
+
+	tests := []struct {
+		opt DeleteBranchOptions
+	}{
+		{
+			opt: DeleteBranchOptions{
+				Force: false,
+			},
+		},
+		{
+			opt: DeleteBranchOptions{
+				Force: true,
+			},
+		},
+	}
+	for _, test := range tests {
+		t.Run("", func(t *testing.T) {
+			branch := strconv.Itoa(int(time.Now().UnixNano()))
+			err := r.Checkout(branch, CheckoutOptions{
+				BaseBranch: "master",
+			})
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			assert.True(t, r.HasReference(RefsHeads+branch))
+
+			err = r.Checkout("master")
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			err = r.DeleteBranch(branch, test.opt)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			assert.False(t, r.HasReference(RefsHeads+branch))
+		})
+	}
+}