浏览代码

refactor(vcs): implement GitBuilder interface

jguer 3 年之前
父节点
当前提交
444151de0d
共有 2 个文件被更改,包括 3 次插入3 次删除
  1. 2 2
      pkg/vcs/vcs.go
  2. 1 1
      pkg/vcs/vcs_test.go

+ 2 - 2
pkg/vcs/vcs.go

@@ -21,7 +21,7 @@ type InfoStore struct {
 	OriginsByPackage map[string]OriginInfoByURL
 	FilePath         string
 	Runner           exe.Runner
-	CmdBuilder       exe.ICmdBuilder
+	CmdBuilder       exe.GitCmdBuilder
 }
 
 // OriginInfoByURL stores the OriginInfo of each origin URL provided
@@ -42,7 +42,7 @@ type OriginInfo struct {
 	SHA       string   `json:"sha"`
 }
 
-func NewInfoStore(filePath string, runner exe.Runner, cmdBuilder exe.ICmdBuilder) *InfoStore {
+func NewInfoStore(filePath string, runner exe.Runner, cmdBuilder exe.GitCmdBuilder) *InfoStore {
 	infoStore := &InfoStore{
 		CmdBuilder:       cmdBuilder,
 		FilePath:         filePath,

+ 1 - 1
pkg/vcs/vcs_test.go

@@ -75,7 +75,7 @@ func TestNewInfoStore(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			got := NewInfoStore(tt.args.filePath, tt.args.runner, tt.args.cmdBuilder)
 			assert.NotNil(t, got)
-			assert.Equal(t, []string{"--a", "--b"}, got.CmdBuilder.GitFlags)
+			assert.Equal(t, []string{"--a", "--b"}, got.CmdBuilder.(*exe.CmdBuilder).GitFlags)
 			assert.Equal(t, tt.args.cmdBuilder, got.CmdBuilder)
 			assert.Equal(t, tt.args.runner, got.Runner)
 			assert.Equal(t, "/tmp/a.json", got.FilePath)