add TestGetBranchPos (#98)

JunNishimura · Jun 6, 2023 · c08e126 · c08e126
1 parent a063565
commit c08e126
Showing 1 changed file with 100 additions and 8 deletions.
diff --git a/internal/store/refs_test.go b/internal/store/refs_test.go
@@ -109,22 +109,22 @@ func TestNewRefs(t *testing.T) {
 		hash sha.SHA1
 	}
 	tests := []struct {
-		name       string
-		filedsList []*fields
-		want       *Refs
-		wantErr    error
+		name    string
+		fields  []*fields
+		want    *Refs
+		wantErr error
 	}{
 		{
-			name:       "success: no heads",
-			filedsList: nil,
+			name:   "success: no heads",
+			fields: nil,
 			want: &Refs{
 				Heads: make([]*branch, 0),
 			},
 			wantErr: nil,
 		},
 		{
 			name: "success: some heads",
-			filedsList: []*fields{
+			fields: []*fields{
 				{
 					name: "main",
 					hash: sha.SHA1([]byte("87f3c49bccf2597484ece08746d3ee5defaba335")),
@@ -166,7 +166,7 @@ func TestNewRefs(t *testing.T) {
 			if err := os.Mkdir(headsDir, os.ModePerm); err != nil {
 				t.Logf("%v: %s", err, headsDir)
 			}
-			for _, field := range tt.filedsList {
+			for _, field := range tt.fields {
 				// make main branch
 				branchPath := filepath.Join(headsDir, field.name)
 				f, err := os.Create(branchPath)
@@ -294,3 +294,95 @@ func TestAddBranch(t *testing.T) {
 		})
 	}
 }
+
+func TestGetBranchPos(t *testing.T) {
+	type args struct {
+		branchName string
+	}
+	type fields struct {
+		name string
+		hash sha.SHA1
+	}
+	tests := []struct {
+		name       string
+		args       args
+		fieldsList []*fields
+		want       int
+	}{
+		{
+			name: "found existing branch",
+			args: args{
+				branchName: "main",
+			},
+			fieldsList: []*fields{
+				{
+					name: "main",
+					hash: sha.SHA1([]byte("87f3c49bccf2597484ece08746d3ee5defaba335")),
+				},
+				{
+					name: "test",
+					hash: sha.SHA1([]byte("87f3c49bccf2597484ece08746d3ee5defaba335")),
+				},
+			},
+			want: 0,
+		},
+		{
+			name: "not found",
+			args: args{
+				branchName: "xxxx",
+			},
+			fieldsList: []*fields{
+				{
+					name: "main",
+					hash: sha.SHA1([]byte("87f3c49bccf2597484ece08746d3ee5defaba335")),
+				},
+				{
+					name: "test",
+					hash: sha.SHA1([]byte("87f3c49bccf2597484ece08746d3ee5defaba335")),
+				},
+			},
+			want: NewBranchFlag,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			tmpDir := t.TempDir()
+			// .goit initialization
+			goitDir := filepath.Join(tmpDir, ".goit")
+			if err := os.Mkdir(goitDir, os.ModePerm); err != nil {
+				t.Logf("%v: %s", err, goitDir)
+			}
+			// make .goit/refs directory
+			refsDir := filepath.Join(goitDir, "refs")
+			if err := os.Mkdir(refsDir, os.ModePerm); err != nil {
+				t.Logf("%v: %s", err, refsDir)
+			}
+			// make .goit/refs/heads directory
+			headsDir := filepath.Join(refsDir, "heads")
+			if err := os.Mkdir(headsDir, os.ModePerm); err != nil {
+				t.Logf("%v: %s", err, headsDir)
+			}
+			for _, field := range tt.fieldsList {
+				// make main branch
+				branchPath := filepath.Join(headsDir, field.name)
+				f, err := os.Create(branchPath)
+				if err != nil {
+					t.Logf("%v: %s", err, branchPath)
+				}
+				if _, err := f.WriteString(field.hash.String()); err != nil {
+					t.Log(err)
+				}
+				f.Close()
+			}
+
+			r, err := NewRefs(goitDir)
+			if err != nil {
+				t.Log(err)
+			}
+			got := r.getBranchPos(tt.args.branchName)
+			if got != tt.want {
+				t.Errorf("got = %d, want = %d", got, tt.want)
+			}
+		})
+	}
+}