package packp
import (
"bytes"
"fmt"
"io"
"strings"
"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 AdvRefSuite struct{}
var _ = Suite(&AdvRefSuite{})
func (s *AdvRefSuite) TestAddReferenceSymbolic(c *C) {
ref := plumbing.NewSymbolicReference("foo", "bar")
a := NewAdvRefs()
err := a.AddReference(ref)
c.Assert(err, IsNil)
values := a.Capabilities.Get(capability.SymRef)
c.Assert(values, HasLen, 1)
c.Assert(values[0], Equals, "foo:bar")
}
func (s *AdvRefSuite) TestAddReferenceHash(c *C) {
ref := plumbing.NewHashReference("foo", plumbing.NewHash("5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c"))
a := NewAdvRefs()
err := a.AddReference(ref)
c.Assert(err, IsNil)
c.Assert(a.References, HasLen, 1)
c.Assert(a.References["foo"].String(), Equals, "5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c")
}
func (s *AdvRefSuite) TestAllReferences(c *C) {
hash := plumbing.NewHash("5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c")
a := NewAdvRefs()
err := a.AddReference(plumbing.NewSymbolicReference("foo", "bar"))
c.Assert(err, IsNil)
err = a.AddReference(plumbing.NewHashReference("bar", hash))
c.Assert(err, IsNil)
refs, err := a.AllReferences()
c.Assert(err, IsNil)
iter, err := refs.IterReferences()
c.Assert(err, IsNil)
var count int
iter.ForEach(func(ref *plumbing.Reference) error {
count++
switch ref.Name() {
case "bar":
c.Assert(ref.Hash(), Equals, hash)
case "foo":
c.Assert(ref.Target().String(), Equals, "bar")
}
return nil
})
c.Assert(count, Equals, 2)
}
func (s *AdvRefSuite) TestAllReferencesBadSymref(c *C) {
a := NewAdvRefs()
err := a.Capabilities.Set(capability.SymRef, "foo")
c.Assert(err, IsNil)
_, err = a.AllReferences()
c.Assert(err, NotNil)
}
type AdvRefsDecodeEncodeSuite struct{}
var _ = Suite(&AdvRefsDecodeEncodeSuite{})
func (s *AdvRefsDecodeEncodeSuite) test(c *C, in []string, exp []string) {
var err error
var input io.Reader
{
var buf bytes.Buffer
p := pktline.NewEncoder(&buf)
err = p.EncodeString(in...)
c.Assert(err, IsNil)
input = &buf
}
var expected []byte
{
var buf bytes.Buffer
p := pktline.NewEncoder(&buf)
err = p.EncodeString(exp...)
c.Assert(err, IsNil)
expected = buf.Bytes()
}
var obtained []byte
{
ar := NewAdvRefs()
c.Assert(ar.Decode(input), IsNil)
var buf bytes.Buffer
c.Assert(ar.Encode(&buf), IsNil)
obtained = buf.Bytes()
}
c.Assert(string(obtained), DeepEquals, string(expected))
}
func (s *AdvRefsDecodeEncodeSuite) TestNoHead(c *C) {
input := []string{
"0000000000000000000000000000000000000000 capabilities^{}\x00",
pktline.FlushString,
}
expected := []string{
"0000000000000000000000000000000000000000 capabilities^{}\x00\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestNoHeadSmart(c *C) {
input := []string{
"# service=git-upload-pack\n",
"0000000000000000000000000000000000000000 capabilities^{}\x00",
pktline.FlushString,
}
expected := []string{
"# service=git-upload-pack\n",
"0000000000000000000000000000000000000000 capabilities^{}\x00\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestNoHeadSmartBug(c *C) {
input := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"0000000000000000000000000000000000000000 capabilities^{}\x00\n",
pktline.FlushString,
}
expected := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"0000000000000000000000000000000000000000 capabilities^{}\x00\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestRefs(c *C) {
input := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree",
pktline.FlushString,
}
expected := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestPeeled(c *C) {
input := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
pktline.FlushString,
}
expected := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestAll(c *C) {
input := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}",
"shallow 1111111111111111111111111111111111111111",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
expected := []string{
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
"shallow 1111111111111111111111111111111111111111\n",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestAllSmart(c *C) {
input := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
"shallow 1111111111111111111111111111111111111111\n",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
expected := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
"shallow 1111111111111111111111111111111111111111\n",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func (s *AdvRefsDecodeEncodeSuite) TestAllSmartBug(c *C) {
input := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
"shallow 1111111111111111111111111111111111111111\n",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
expected := []string{
"# service=git-upload-pack\n",
pktline.FlushString,
"6ecf0ef2c2dffb796033e5a02219af86ec6584e5 HEAD\x00symref=HEAD:/refs/heads/master ofs-delta multi_ack\n",
"a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n",
"5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c refs/tags/v2.6.11-tree\n",
"c39ae07f393806ccf406ef966e9a15afc43cc36a refs/tags/v2.6.11-tree^{}\n",
"7777777777777777777777777777777777777777 refs/tags/v2.6.12-tree\n",
"8888888888888888888888888888888888888888 refs/tags/v2.6.12-tree^{}\n",
"shallow 1111111111111111111111111111111111111111\n",
"shallow 2222222222222222222222222222222222222222\n",
pktline.FlushString,
}
s.test(c, input, expected)
}
func ExampleAdvRefs_Decode() {
// Here is a raw advertised-ref message.
raw := "" +
"0065a6930aaee06755d1bdcfd943fbf614e4d92bb0c7 HEAD\x00multi_ack ofs-delta symref=HEAD:/refs/heads/master\n" +
"003fa6930aaee06755d1bdcfd943fbf614e4d92bb0c7 refs/heads/master\n" +
"00441111111111111111111111111111111111111111 refs/tags/v2.6.11-tree\n" +
"00475555555555555555555555555555555555555555 refs/tags/v2.6.11-tree^{}\n" +
"0035shallow 5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c\n" +
"0000"
// Use the raw message as our input.
input := strings.NewReader(raw)
// Decode the input into a newly allocated AdvRefs value.
ar := NewAdvRefs()
_ = ar.Decode(input) // error check ignored for brevity
// Do something interesting with the AdvRefs, e.g. print its contents.
fmt.Println("head =", ar.Head)
fmt.Println("capabilities =", ar.Capabilities.String())
fmt.Println("...")
fmt.Println("shallows =", ar.Shallows)
// Output: head = a6930aaee06755d1bdcfd943fbf614e4d92bb0c7
// capabilities = multi_ack ofs-delta symref=HEAD:/refs/heads/master
// ...
// shallows = [5dc01c595e6c6ec9ccda4f6f69c131c0dd945f8c]
}
func ExampleAdvRefs_Encode() {
// Create an AdvRefs with the contents you want...
ar := NewAdvRefs()
// ...add a hash for the HEAD...
head := plumbing.NewHash("1111111111111111111111111111111111111111")
ar.Head = &head
// ...add some server capabilities...
ar.Capabilities.Add(capability.MultiACK)
ar.Capabilities.Add(capability.OFSDelta)
ar.Capabilities.Add(capability.SymRef, "HEAD:/refs/heads/master")
// ...add a couple of references...
ar.References["refs/heads/master"] = plumbing.NewHash("2222222222222222222222222222222222222222")
ar.References["refs/tags/v1"] = plumbing.NewHash("3333333333333333333333333333333333333333")
// ...including a peeled ref...
ar.Peeled["refs/tags/v1"] = plumbing.NewHash("4444444444444444444444444444444444444444")
// ...and finally add a shallow
ar.Shallows = append(ar.Shallows, plumbing.NewHash("5555555555555555555555555555555555555555"))
// Encode the packpContents to a bytes.Buffer.
// You can encode into stdout too, but you will not be able
// see the '\x00' after "HEAD".
var buf bytes.Buffer
_ = ar.Encode(&buf) // error checks ignored for brevity
// Print the contents of the buffer as a quoted string.
// Printing is as a non-quoted string will be prettier but you
// will miss the '\x00' after "HEAD".
fmt.Printf("%q", buf.String())
// Output:
// "00651111111111111111111111111111111111111111 HEAD\x00multi_ack ofs-delta symref=HEAD:/refs/heads/master\n003f2222222222222222222222222222222222222222 refs/heads/master\n003a3333333333333333333333333333333333333333 refs/tags/v1\n003d4444444444444444444444444444444444444444 refs/tags/v1^{}\n0035shallow 5555555555555555555555555555555555555555\n0000"
}
