package gitobj
import (
"bytes"
"compress/zlib"
"crypto/sha1"
"crypto/sha256"
"encoding/hex"
"errors"
"hash"
"io"
"io/ioutil"
"sync/atomic"
"testing"
"github.com/stretchr/testify/assert"
)
func TestObjectWriterWritesHeaders(t *testing.T) {
var buf bytes.Buffer
w := NewObjectWriter(&buf, sha1.New())
n, err := w.WriteHeader(BlobObjectType, 1)
assert.Equal(t, 7, n)
assert.Nil(t, err)
assert.Nil(t, w.Close())
r, err := zlib.NewReader(&buf)
assert.Nil(t, err)
all, err := ioutil.ReadAll(r)
assert.Nil(t, err)
assert.Equal(t, []byte("blob 1\x00"), all)
assert.Nil(t, r.Close())
}
func TestObjectWriterWritesData(t *testing.T) {
testCases := []struct {
h hash.Hash
sha string
}{
{
sha1.New(), "56a6051ca2b02b04ef92d5150c9ef600403cb1de",
},
{
sha256.New(), "36456d9b87f21fc54ed5babf1222a9ab0fbbd0c4ad239a7933522d5e4447049c",
},
}
for _, test := range testCases {
var buf bytes.Buffer
w := NewObjectWriter(&buf, test.h)
w.WriteHeader(BlobObjectType, 1)
n, err := w.Write([]byte{0x31})
assert.Equal(t, 1, n)
assert.Nil(t, err)
assert.Nil(t, w.Close())
r, err := zlib.NewReader(&buf)
assert.Nil(t, err)
all, err := ioutil.ReadAll(r)
assert.Nil(t, err)
assert.Equal(t, []byte("blob 1\x001"), all)
assert.Nil(t, r.Close())
assert.Equal(t, test.sha, hex.EncodeToString(w.Sha()))
}
}
func TestObjectWriterPanicsOnWritesWithoutHeader(t *testing.T) {
defer func() {
err := recover()
assert.NotNil(t, err)
assert.Equal(t, "gitobj: cannot write data without header", err)
}()
w := NewObjectWriter(new(bytes.Buffer), sha1.New())
w.Write(nil)
}
func TestObjectWriterPanicsOnMultipleHeaderWrites(t *testing.T) {
defer func() {
err := recover()
assert.NotNil(t, err)
assert.Equal(t, "gitobj: cannot write headers more than once", err)
}()
w := NewObjectWriter(new(bytes.Buffer), sha1.New())
w.WriteHeader(BlobObjectType, 1)
w.WriteHeader(TreeObjectType, 2)
}
func TestObjectWriterKeepsTrackOfHash(t *testing.T) {
w := NewObjectWriter(new(bytes.Buffer), sha1.New())
n, err := w.WriteHeader(BlobObjectType, 1)
assert.Nil(t, err)
assert.Equal(t, 7, n)
assert.Equal(t, "bb6ca78b66403a67c6281df142de5ef472186283", hex.EncodeToString(w.Sha()))
w = NewObjectWriter(new(bytes.Buffer), sha256.New())
n, err = w.WriteHeader(BlobObjectType, 1)
assert.Nil(t, err)
assert.Equal(t, 7, n)
assert.Equal(t, "3a68c454a6eb75cc55bda147a53756f0f581497eb80b9b67156fb8a8d3931cd7", hex.EncodeToString(w.Sha()))
}
type WriteCloserFn struct {
io.Writer
closeFn func() error
}
func (r *WriteCloserFn) Close() error { return r.closeFn() }
func TestObjectWriterCallsClose(t *testing.T) {
var calls uint32
expected := errors.New("close error")
w := NewObjectWriteCloser(&WriteCloserFn{
Writer: new(bytes.Buffer),
closeFn: func() error {
atomic.AddUint32(&calls, 1)
return expected
},
}, sha1.New())
got := w.Close()
assert.EqualValues(t, 1, calls)
assert.Equal(t, expected, got)
}