Merge pull request #470 from lhchavez/fix-odbreadstream-read
Return io.EOF on OdbReadStream.Read()
(cherry picked from commit 7e9128bd58
)
This commit is contained in:
parent
13d27a4f62
commit
c74ce46055
4
odb.go
4
odb.go
|
@ -8,6 +8,7 @@ extern void _go_git_odb_backend_free(git_odb_backend *backend);
|
||||||
*/
|
*/
|
||||||
import "C"
|
import "C"
|
||||||
import (
|
import (
|
||||||
|
"io"
|
||||||
"reflect"
|
"reflect"
|
||||||
"runtime"
|
"runtime"
|
||||||
"unsafe"
|
"unsafe"
|
||||||
|
@ -287,6 +288,9 @@ func (stream *OdbReadStream) Read(data []byte) (int, error) {
|
||||||
if ret < 0 {
|
if ret < 0 {
|
||||||
return 0, MakeGitError(ret)
|
return 0, MakeGitError(ret)
|
||||||
}
|
}
|
||||||
|
if ret == 0 {
|
||||||
|
return 0, io.EOF
|
||||||
|
}
|
||||||
|
|
||||||
header.Len = int(ret)
|
header.Len = int(ret)
|
||||||
|
|
||||||
|
|
16
odb_test.go
16
odb_test.go
|
@ -3,6 +3,7 @@ package git
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -47,22 +48,29 @@ func TestOdbStream(t *testing.T) {
|
||||||
|
|
||||||
str := "hello, world!"
|
str := "hello, world!"
|
||||||
|
|
||||||
stream, error := odb.NewWriteStream(int64(len(str)), ObjectBlob)
|
writeStream, error := odb.NewWriteStream(int64(len(str)), ObjectBlob)
|
||||||
checkFatal(t, error)
|
checkFatal(t, error)
|
||||||
n, error := io.WriteString(stream, str)
|
n, error := io.WriteString(writeStream, str)
|
||||||
checkFatal(t, error)
|
checkFatal(t, error)
|
||||||
if n != len(str) {
|
if n != len(str) {
|
||||||
t.Fatalf("Bad write length %v != %v", n, len(str))
|
t.Fatalf("Bad write length %v != %v", n, len(str))
|
||||||
}
|
}
|
||||||
|
|
||||||
error = stream.Close()
|
error = writeStream.Close()
|
||||||
checkFatal(t, error)
|
checkFatal(t, error)
|
||||||
|
|
||||||
expectedId, error := NewOid("30f51a3fba5274d53522d0f19748456974647b4f")
|
expectedId, error := NewOid("30f51a3fba5274d53522d0f19748456974647b4f")
|
||||||
checkFatal(t, error)
|
checkFatal(t, error)
|
||||||
if stream.Id.Cmp(expectedId) != 0 {
|
if writeStream.Id.Cmp(expectedId) != 0 {
|
||||||
t.Fatal("Wrong data written")
|
t.Fatal("Wrong data written")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
readStream, error := odb.NewReadStream(&writeStream.Id)
|
||||||
|
checkFatal(t, error)
|
||||||
|
data, error := ioutil.ReadAll(readStream)
|
||||||
|
if str != string(data) {
|
||||||
|
t.Fatalf("Wrong data read %v != %v", str, string(data))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestOdbHash(t *testing.T) {
|
func TestOdbHash(t *testing.T) {
|
||||||
|
|
Loading…
Reference in New Issue