mirror of https://github.com/restic/restic.git
48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
package khepri_test
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"hash"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/fd0/khepri"
|
|
)
|
|
|
|
var static_tests = []struct {
|
|
hash func() hash.Hash
|
|
text string
|
|
digest string
|
|
}{
|
|
{md5.New, "foobar\n", "14758f1afd44c09b7992073ccf00b43d"},
|
|
// test data from http://www.nsrl.nist.gov/testdata/
|
|
{sha1.New, "abc", "a9993e364706816aba3e25717850c26c9cd0d89d"},
|
|
{sha1.New, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", "84983e441c3bd26ebaae4aa1f95129e5e54670f1"},
|
|
}
|
|
|
|
func TestReader(t *testing.T) {
|
|
for _, test := range static_tests {
|
|
r := khepri.NewHashingReader(bytes.NewBuffer([]byte(test.text)), test.hash)
|
|
buf, err := ioutil.ReadAll(r)
|
|
ok(t, err)
|
|
equals(t, test.text, string(buf))
|
|
|
|
equals(t, hex.EncodeToString(r.Hash()), test.digest)
|
|
}
|
|
}
|
|
|
|
func TestWriter(t *testing.T) {
|
|
for _, test := range static_tests {
|
|
var buf bytes.Buffer
|
|
w := khepri.NewHashingWriter(&buf, test.hash)
|
|
|
|
_, err := w.Write([]byte(test.text))
|
|
ok(t, err)
|
|
|
|
equals(t, hex.EncodeToString(w.Hash()), test.digest)
|
|
}
|
|
}
|