package packp import ( "fmt" "os" "strings" "time" "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() // Add a couple of wants ur.Wants = append(ur.Wants, plumbing.NewHash("3333333333333333333333333333333333333333")) ur.Wants = append(ur.Wants, plumbing.NewHash("1111111111111111111111111111111111111111")) ur.Wants = append(ur.Wants, plumbing.NewHash("2222222222222222222222222222222222222222")) // And some capabilities you will like the server to use ur.Capabilities.Add(capability.OFSDelta) ur.Capabilities.Add(capability.SymRef, "HEAD:/refs/heads/master") // Add a couple of shallows ur.Shallows = append(ur.Shallows, plumbing.NewHash("bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb")) ur.Shallows = append(ur.Shallows, plumbing.NewHash("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")) // And retrict the answer of the server to commits newer than "2015-01-02 03:04:05 UTC" since := time.Date(2015, time.January, 2, 3, 4, 5, 0, time.UTC) ur.Depth = DepthSince(since) // Create a new Encode for the stdout... e := newUlReqEncoder(os.Stdout) // ...and encode the upload-request to it. _ = e.Encode(ur) // ignoring errors for brevity // Output: // 005bwant 1111111111111111111111111111111111111111 ofs-delta symref=HEAD:/refs/heads/master // 0032want 2222222222222222222222222222222222222222 // 0032want 3333333333333333333333333333333333333333 // 0035shallow aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa // 0035shallow bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb // 001cdeepen-since 1420167845 // 0000 } func ExampleUlReqDecoder_Decode() { // Here is a raw advertised-ref message. raw := "" + "005bwant 1111111111111111111111111111111111111111 ofs-delta symref=HEAD:/refs/heads/master\n" + "0032want 2222222222222222222222222222222222222222\n" + "0032want 3333333333333333333333333333333333333333\n" + "0035shallow aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa\n" + "0035shallow bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb\n" + "001cdeepen-since 1420167845\n" + // 2015-01-02 03:04:05 +0000 UTC pktline.FlushString // Use the raw message as our input. input := strings.NewReader(raw) // Create the Decoder reading from our input. d := newUlReqDecoder(input) // Decode the input into a newly allocated UlReq value. ur := NewUploadRequest() _ = d.Decode(ur) // error check ignored for brevity // Do something interesting with the UlReq, e.g. print its contents. fmt.Println("capabilities =", ur.Capabilities.String()) fmt.Println("wants =", ur.Wants) fmt.Println("shallows =", ur.Shallows) switch depth := ur.Depth.(type) { case DepthCommits: fmt.Println("depth =", int(depth)) case DepthSince: fmt.Println("depth =", time.Time(depth)) case DepthReference: fmt.Println("depth =", string(depth)) } // Output: // capabilities = ofs-delta symref=HEAD:/refs/heads/master // wants = [1111111111111111111111111111111111111111 2222222222222222222222222222222222222222 3333333333333333333333333333333333333333] // shallows = [aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb] // depth = 2015-01-02 03:04:05 +0000 UTC }