1 files changed, 199 insertions, 0 deletions
diff --git a/formats/packp/pktline/pktline_test.go b/formats/packp/pktline/pktline_test.go
new file mode 100644
index 0000000..3c18f53
--- /dev/null
+++ b/formats/packp/pktline/pktline_test.go
@@ -0,0 +1,199 @@
+package pktline_test
+
+import (
+	"io"
+	"io/ioutil"
+	"os"
+	"strings"
+	"testing"
+
+	"gopkg.in/src-d/go-git.v4/formats/packp/pktline"
+
+	. "gopkg.in/check.v1"
+)
+
+func Test(t *testing.T) { TestingT(t) }
+
+type SuitePktLine struct {
+}
+
+var _ = Suite(&SuitePktLine{})
+
+func (s *SuitePktLine) TestNew(c *C) {
+	for i, test := range [...]struct {
+		input    [][]byte
+		expected []byte
+	}{
+		{
+			input:    [][]byte{},
+			expected: []byte{},
+		}, {
+			input: [][]byte{
+				[]byte(nil),
+			},
+			expected: []byte("0000"),
+		}, {
+			input: [][]byte{
+				[]byte{},
+			},
+			expected: []byte("0000"),
+		}, {
+			input: [][]byte{
+				[]byte(""),
+			},
+			expected: []byte("0000"),
+		}, {
+			input: [][]byte{
+				[]byte("hello\n"),
+			},
+			expected: []byte("000ahello\n"),
+		}, {
+			input: [][]byte{
+				[]byte("hello\n"),
+				[]byte("world!\n"),
+				[]byte(""),
+				[]byte("foo"),
+				[]byte(""),
+			},
+			expected: []byte("000ahello\n000bworld!\n00000007foo0000"),
+		}, {
+			input: [][]byte{
+				[]byte(strings.Repeat("a", pktline.MaxPayloadSize)),
+			},
+			expected: []byte("fff0" + strings.Repeat("a", pktline.MaxPayloadSize)),
+		},
+	} {
+		r, err := pktline.New(test.input...)
+		c.Assert(err, IsNil, Commentf("input %d = %v", i, test.input))
+
+		obtained, err := ioutil.ReadAll(r)
+		c.Assert(err, IsNil, Commentf("input %d = %v", i, test.input))
+
+		c.Assert(obtained, DeepEquals, test.expected,
+			Commentf("input %d = %v", i, test.input))
+	}
+}
+
+func (s *SuitePktLine) TestNewErrPayloadTooLong(c *C) {
+	for _, input := range [...][][]byte{
+		[][]byte{
+			[]byte(strings.Repeat("a", pktline.MaxPayloadSize+1)),
+		},
+		[][]byte{
+			[]byte("hello world!"),
+			[]byte(""),
+			[]byte(strings.Repeat("a", pktline.MaxPayloadSize+1)),
+		},
+		[][]byte{
+			[]byte("hello world!"),
+			[]byte(strings.Repeat("a", pktline.MaxPayloadSize+1)),
+			[]byte("foo"),
+		},
+	} {
+		_, err := pktline.New(input...)
+
+		c.Assert(err, Equals, pktline.ErrPayloadTooLong,
+			Commentf("%v\n", input))
+	}
+}
+
+func (s *SuitePktLine) TestNewFromStrings(c *C) {
+	for _, test := range [...]struct {
+		input    []string
+		expected []byte
+	}{
+		{
+			input:    []string(nil),
+			expected: []byte{},
+		}, {
+			input:    []string{},
+			expected: []byte{},
+		}, {
+			input:    []string{""},
+			expected: []byte("0000"),
+		}, {
+			input:    []string{"hello\n"},
+			expected: []byte("000ahello\n"),
+		}, {
+			input:    []string{"hello\n", "world!\n", "", "foo", ""},
+			expected: []byte("000ahello\n000bworld!\n00000007foo0000"),
+		}, {
+			input: []string{
+				strings.Repeat("a", pktline.MaxPayloadSize),
+			},
+			expected: []byte("fff0" + strings.Repeat("a", pktline.MaxPayloadSize)),
+		},
+	} {
+		r, err := pktline.NewFromStrings(test.input...)
+		c.Assert(err, IsNil)
+
+		obtained, err := ioutil.ReadAll(r)
+		c.Assert(err, IsNil)
+
+		c.Assert(obtained, DeepEquals, test.expected,
+			Commentf("input = %v\n", test.input))
+	}
+}
+
+func (s *SuitePktLine) TestNewFromStringsErrPayloadTooLong(c *C) {
+	for _, input := range [...][]string{
+		[]string{
+			strings.Repeat("a", pktline.MaxPayloadSize+1),
+		},
+		[]string{
+			"hello world!",
+			"",
+			strings.Repeat("a", pktline.MaxPayloadSize+1),
+		},
+		[]string{
+			"hello world!",
+			strings.Repeat("a", pktline.MaxPayloadSize+1),
+			"foo",
+		},
+	} {
+		_, err := pktline.NewFromStrings(input...)
+
+		c.Assert(err, Equals, pktline.ErrPayloadTooLong,
+			Commentf("%v\n", input))
+	}
+}
+
+func ExampleNew() {
+	// These are the payloads we want to turn into pkt-lines,
+	// the empty slice at the end will generate a flush-pkt.
+	payloads := [][]byte{
+		[]byte{'h', 'e', 'l', 'l', 'o', '\n'},
+		[]byte{'w', 'o', 'r', 'l', 'd', '!', '\n'},
+		[]byte{},
+	}
+
+	// Create the pkt-lines, ignoring errors...
+	pktlines, _ := pktline.New(payloads...)
+
+	// Send the raw data to stdout, ignoring errors...
+	_, _ = io.Copy(os.Stdout, pktlines)
+
+	// Output: 000ahello
+	// 000bworld!
+	// 0000
+}
+
+func ExampleNewFromStrings() {
+	// These are the payloads we want to turn into pkt-lines,
+	// the empty string at the end will generate a flush-pkt.
+	payloads := []string{
+		"hello\n",
+		"world!\n",
+		"",
+	}
+
+	// Create the pkt-lines, ignoring errors...
+	pktlines, _ := pktline.NewFromStrings(payloads...)
+
+	// Send the raw data to stdout, ignoring errors...
+	_, _ = io.Copy(os.Stdout, pktlines)
+
+	// Output: 000ahello
+	// 000bworld!
+	// 0000
+}