use io reader for GetFormatPatch (#105)

This commit is contained in:
Lunny Xiao 2018-01-07 00:11:05 -06:00 коммит произвёл GitHub
Родитель 2c3dc95a96
Коммит f4a9105367
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 16 добавлений и 4 удалений

Просмотреть файл

@ -5,8 +5,10 @@
package git
import (
"bytes"
"container/list"
"fmt"
"io"
"strconv"
"strings"
"time"
@ -75,6 +77,13 @@ func (repo *Repository) GetPatch(base, head string) ([]byte, error) {
}
// GetFormatPatch generates and returns format-patch data between given revisions.
func (repo *Repository) GetFormatPatch(base, head string) ([]byte, error) {
return NewCommand("format-patch", "--binary", "--stdout", base + "..." + head).RunInDirBytes(repo.Path)
func (repo *Repository) GetFormatPatch(base, head string) (io.Reader, error) {
stdout := new(bytes.Buffer)
stderr := new(bytes.Buffer)
if err := NewCommand("format-patch", "--binary", "--stdout", base+"..."+head).
RunInDirPipeline(repo.Path, stdout, stderr); err != nil {
return nil, concatenateError(err, stderr.String())
}
return stdout, nil
}

Просмотреть файл

@ -5,15 +5,18 @@
package git
import (
"io/ioutil"
"testing"
"github.com/stretchr/testify/assert"
)
func TestGetFormatPatch(t *testing.T) {
repo, err := OpenRepository(".");
repo, err := OpenRepository(".")
assert.NoError(t, err)
patchb, err := repo.GetFormatPatch("cdb43f0e^", "cdb43f0e")
rd, err := repo.GetFormatPatch("cdb43f0e^", "cdb43f0e")
assert.NoError(t, err)
patchb, err := ioutil.ReadAll(rd)
assert.NoError(t, err)
patch := string(patchb)
assert.Regexp(t, "^From cdb43f0e", patch)