1 files changed, 44 insertions, 0 deletions
diff --git a/plumbing/protocol/packp/uppackresp_test.go b/plumbing/protocol/packp/uppackresp_test.go
index c81bb76..c27fdda 100644
--- a/plumbing/protocol/packp/uppackresp_test.go
+++ b/plumbing/protocol/packp/uppackresp_test.go
@@ -7,6 +7,7 @@ import (
 	"gopkg.in/src-d/go-git.v4/plumbing/protocol/packp/capability"
 
 	. "gopkg.in/check.v1"
+	"gopkg.in/src-d/go-git.v4/plumbing"
 )
 
 type UploadPackResponseSuite struct{}
@@ -80,3 +81,46 @@ func (s *UploadPackResponseSuite) TestReadNoDecode(c *C) {
 	c.Assert(err, Equals, ErrUploadPackResponseNotDecoded)
 	c.Assert(n, Equals, 0)
 }
+
+func (s *UploadPackResponseSuite) TestEncodeNAK(c *C) {
+	pf := ioutil.NopCloser(bytes.NewBuffer([]byte("[PACK]")))
+	req := NewUploadPackRequest()
+	res := NewUploadPackResponseWithPackfile(req, pf)
+	defer func() { c.Assert(res.Close(), IsNil) }()
+
+	b := bytes.NewBuffer(nil)
+	c.Assert(res.Encode(b), IsNil)
+
+	expected := "0008NAK\n[PACK]"
+	c.Assert(string(b.Bytes()), Equals, expected)
+}
+
+func (s *UploadPackResponseSuite) TestEncodeDepth(c *C) {
+	pf := ioutil.NopCloser(bytes.NewBuffer([]byte("[PACK]")))
+	req := NewUploadPackRequest()
+	req.Depth = DepthCommits(1)
+
+	res := NewUploadPackResponseWithPackfile(req, pf)
+	defer func() { c.Assert(res.Close(), IsNil) }()
+
+	b := bytes.NewBuffer(nil)
+	c.Assert(res.Encode(b), IsNil)
+
+	expected := "00000008NAK\n[PACK]"
+	c.Assert(string(b.Bytes()), Equals, expected)
+}
+
+func (s *UploadPackResponseSuite) TestEncodeMultiACK(c *C) {
+	pf := ioutil.NopCloser(bytes.NewBuffer([]byte("[PACK]")))
+	req := NewUploadPackRequest()
+
+	res := NewUploadPackResponseWithPackfile(req, pf)
+	defer func() { c.Assert(res.Close(), IsNil) }()
+	res.ACKs = []plumbing.Hash{
+		plumbing.NewHash("5dc01c595e6c6ec9ccda4f6f69c131c0dd945f81"),
+		plumbing.NewHash("5dc01c595e6c6ec9ccda4f6f69c131c0dd945f82"),
+	}
+
+	b := bytes.NewBuffer(nil)
+	c.Assert(res.Encode(b), NotNil)
+}