From 4bf531be9cc774370cd92b2a59c7e69e8c753c26 Mon Sep 17 00:00:00 2001
From: Grzegorz Dlugoszewski <g.dlugoszewski@gmail.com>
Date: Tue, 19 May 2020 14:22:38 +0200
Subject: [PATCH] Convert branch tests to table driven

---
 branch_test.go | 50 +++++++++++++++++++++++---------------------------
 1 file changed, 23 insertions(+), 27 deletions(-)

diff --git a/branch_test.go b/branch_test.go
index 8861d70..75a3424 100644
--- a/branch_test.go
+++ b/branch_test.go
@@ -45,34 +45,30 @@ func TestClonedBranches(t *testing.T) {
 	branches, err := Branches(repo)
 	checkFatal(t, err)
 
-	master := branches["master"]
-	wantMaster := BranchStatus{
-		Name:        "master",
-		IsRemote:    false,
-		HasUpstream: true,
+	var tests = []struct {
+		got  BranchStatus
+		want BranchStatus
+	}{
+		{branches["master"], BranchStatus{
+			Name:        "master",
+			IsRemote:    false,
+			HasUpstream: true,
+		}},
+		{branches["origin/master"], BranchStatus{
+			Name:        "origin/master",
+			IsRemote:    true,
+			HasUpstream: false,
+		}},
+		{branches["branch"], BranchStatus{
+			Name:        "branch",
+			IsRemote:    false,
+			HasUpstream: false,
+		}},
 	}
 
-	originMaster := branches["origin/master"]
-	wantOriginMaster := BranchStatus{
-		Name:        "origin/master",
-		IsRemote:    true,
-		HasUpstream: false,
-	}
-
-	branch := branches["branch"]
-	wantBranch := BranchStatus{
-		Name:        "branch",
-		IsRemote:    false,
-		HasUpstream: false,
-	}
-
-	if !reflect.DeepEqual(master, wantMaster) {
-		t.Errorf("Wrong branch status, got %+v; want %+v", master, wantMaster)
-	}
-	if !reflect.DeepEqual(originMaster, wantOriginMaster) {
-		t.Errorf("Wrong branch status, got %+v; want %+v", originMaster, wantOriginMaster)
-	}
-	if !reflect.DeepEqual(branch, wantBranch) {
-		t.Errorf("Wrong branch status, got %+v; want %+v", branch, wantBranch)
+	for _, test := range tests {
+		if !reflect.DeepEqual(test.got, test.want) {
+			t.Errorf("Wrong branch status, got %+v; want %+v", test.got, test.want)
+		}
 	}
 }