From c15bf1dff332873644290db0e186b8f5ad9b8fb2 Mon Sep 17 00:00:00 2001
From: Máximo Cuadros <mcuadros@gmail.com>
Date: Thu, 1 Dec 2016 09:59:19 +0100
Subject: capabilities: full integration (#151)

* format/pktline: fix readPayloadLen err handling

* protocol/pakp: UploadReq validation and creation of capabilities

* protocol/pakp: AdvRef tests

* protocol/pakp: capability.List.Delete

* protocol: filter unsupported capabilities

* remote capability negociation

* transport: UploadRequest validation

* requested changes
---
 plumbing/protocol/packp/ulreq_test.go | 101 ++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)

(limited to 'plumbing/protocol/packp/ulreq_test.go')

diff --git a/plumbing/protocol/packp/ulreq_test.go b/plumbing/protocol/packp/ulreq_test.go
index 5e9e978..53626ee 100644
--- a/plumbing/protocol/packp/ulreq_test.go
+++ b/plumbing/protocol/packp/ulreq_test.go
@@ -9,8 +9,109 @@ import (
 	"gopkg.in/src-d/go-git.v4/plumbing"
 	"gopkg.in/src-d/go-git.v4/plumbing/format/pktline"
 	"gopkg.in/src-d/go-git.v4/plumbing/protocol/packp/capability"
+
+	. "gopkg.in/check.v1"
 )
 
+type UlReqSuite struct{}
+
+var _ = Suite(&UlReqSuite{})
+
+func (s *UlReqSuite) TestNewUploadRequestFromCapabilities(c *C) {
+	cap := capability.NewList()
+	cap.Set(capability.Sideband)
+	cap.Set(capability.Sideband64k)
+	cap.Set(capability.MultiACK)
+	cap.Set(capability.MultiACKDetailed)
+	cap.Set(capability.ThinPack)
+	cap.Set(capability.OFSDelta)
+	cap.Set(capability.Agent, "foo")
+
+	r := NewUploadRequestFromCapabilities(cap)
+	c.Assert(r.Capabilities.String(), Equals,
+		"multi_ack_detailed thin-pack ofs-delta agent=go-git/4.x",
+	)
+}
+
+func (s *UlReqSuite) TestValidateWants(c *C) {
+	r := NewUploadRequest()
+	err := r.Validate()
+	c.Assert(err, NotNil)
+
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	err = r.Validate()
+	c.Assert(err, IsNil)
+}
+
+func (s *UlReqSuite) TestValidateShallows(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Shallows = append(r.Shallows, plumbing.NewHash("2222222222222222222222222222222222222222"))
+	err := r.Validate()
+	c.Assert(err, NotNil)
+
+	r.Capabilities.Set(capability.Shallow)
+	err = r.Validate()
+	c.Assert(err, IsNil)
+}
+
+func (s *UlReqSuite) TestValidateDepthCommits(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Depth = DepthCommits(42)
+
+	err := r.Validate()
+	c.Assert(err, NotNil)
+
+	r.Capabilities.Set(capability.Shallow)
+	err = r.Validate()
+	c.Assert(err, IsNil)
+}
+
+func (s *UlReqSuite) TestValidateDepthReference(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Depth = DepthReference("1111111111111111111111111111111111111111")
+
+	err := r.Validate()
+	c.Assert(err, NotNil)
+
+	r.Capabilities.Set(capability.DeepenNot)
+	err = r.Validate()
+	c.Assert(err, IsNil)
+}
+
+func (s *UlReqSuite) TestValidateDepthSince(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Depth = DepthSince(time.Now())
+
+	err := r.Validate()
+	c.Assert(err, NotNil)
+
+	r.Capabilities.Set(capability.DeepenSince)
+	err = r.Validate()
+	c.Assert(err, IsNil)
+}
+
+func (s *UlReqSuite) TestValidateConflictSideband(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Capabilities.Set(capability.Sideband)
+	r.Capabilities.Set(capability.Sideband64k)
+	err := r.Validate()
+	c.Assert(err, NotNil)
+}
+
+func (s *UlReqSuite) TestValidateConflictMultiACK(c *C) {
+	r := NewUploadRequest()
+	r.Wants = append(r.Wants, plumbing.NewHash("1111111111111111111111111111111111111111"))
+	r.Capabilities.Set(capability.MultiACK)
+	r.Capabilities.Set(capability.MultiACKDetailed)
+	err := r.Validate()
+	c.Assert(err, NotNil)
+}
+
 func ExampleUlReqEncoder_Encode() {
 	// Create an empty UlReq with the contents you want...
 	ur := NewUploadRequest()
-- 
cgit