format/pktline: fix readPayloadLen err handling (#148)

1 files changed, 11 insertions, 0 deletions

diff --git a/plumbing/format/pktline/scanner_test.go b/plumbing/format/pktline/scanner_test.go
index 9f440a4..b8a78ec 100644
--- a/plumbing/format/pktline/scanner_test.go
+++ b/plumbing/format/pktline/scanner_test.go
@@ -2,6 +2,7 @@ package pktline_test
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
 	"io"
 	"strings"
@@ -145,6 +146,16 @@ func (s *SuiteScanner) TestEOF(c *C) {
 	c.Assert(sc.Err(), IsNil)
 }
 
+type mockReader struct{}
+
+func (r *mockReader) Read([]byte) (int, error) { return 0, errors.New("foo") }
+
+func (s *SuiteScanner) TestInternalReadError(c *C) {
+	sc := pktline.NewScanner(&mockReader{})
+	c.Assert(sc.Scan(), Equals, false)
+	c.Assert(sc.Err(), ErrorMatches, "foo")
+}
+
 // A section are several non flush-pkt lines followed by a flush-pkt, which
 // how the git protocol sends long messages.
 func (s *SuiteScanner) TestReadSomeSections(c *C) {