forked from TrueCloudLab/restic
78 lines
2 KiB
Go
78 lines
2 KiB
Go
package khepri_test
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"hash"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/fd0/khepri"
|
|
)
|
|
|
|
// assert fails the test if the condition is false.
|
|
func assert(tb testing.TB, condition bool, msg string, v ...interface{}) {
|
|
if !condition {
|
|
_, file, line, _ := runtime.Caller(1)
|
|
fmt.Printf("\033[31m%s:%d: "+msg+"\033[39m\n\n", append([]interface{}{filepath.Base(file), line}, v...)...)
|
|
tb.FailNow()
|
|
}
|
|
}
|
|
|
|
// ok fails the test if an err is not nil.
|
|
func ok(tb testing.TB, err error) {
|
|
if err != nil {
|
|
_, file, line, _ := runtime.Caller(1)
|
|
fmt.Printf("\033[31m%s:%d: unexpected error: %s\033[39m\n\n", filepath.Base(file), line, err.Error())
|
|
tb.FailNow()
|
|
}
|
|
}
|
|
|
|
// equals fails the test if exp is not equal to act.
|
|
func equals(tb testing.TB, exp, act interface{}) {
|
|
if !reflect.DeepEqual(exp, act) {
|
|
_, file, line, _ := runtime.Caller(1)
|
|
fmt.Printf("\033[31m%s:%d:\n\n\texp: %#v\n\n\tgot: %#v\033[39m\n\n", filepath.Base(file), line, exp, act)
|
|
tb.FailNow()
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|