mirror of https://github.com/Xhofe/alist
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
)
|
|
|
|
type hashTest struct {
|
|
input []byte
|
|
output map[*HashType]string
|
|
}
|
|
|
|
var hashTestSet = []hashTest{
|
|
{
|
|
input: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14},
|
|
output: map[*HashType]string{
|
|
MD5: "bf13fc19e5151ac57d4252e0e0f87abe",
|
|
SHA1: "3ab6543c08a75f292a5ecedac87ec41642d12166",
|
|
SHA256: "c839e57675862af5c21bd0a15413c3ec579e0d5522dab600bc6c3489b05b8f54",
|
|
},
|
|
},
|
|
// Empty data set
|
|
{
|
|
input: []byte{},
|
|
output: map[*HashType]string{
|
|
MD5: "d41d8cd98f00b204e9800998ecf8427e",
|
|
SHA1: "da39a3ee5e6b4b0d3255bfef95601890afd80709",
|
|
SHA256: "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestMultiHasher(t *testing.T) {
|
|
for _, test := range hashTestSet {
|
|
mh := NewMultiHasher([]*HashType{MD5, SHA1, SHA256})
|
|
n, err := CopyWithBuffer(mh, bytes.NewBuffer(test.input))
|
|
require.NoError(t, err)
|
|
assert.Len(t, test.input, int(n))
|
|
hashInfo := mh.GetHashInfo()
|
|
for k, v := range hashInfo.h {
|
|
expect, ok := test.output[k]
|
|
require.True(t, ok, "test output for hash not found")
|
|
assert.Equal(t, expect, v)
|
|
}
|
|
// Test that all are present
|
|
for k, v := range test.output {
|
|
expect, ok := hashInfo.h[k]
|
|
require.True(t, ok, "test output for hash not found")
|
|
assert.Equal(t, expect, v)
|
|
}
|
|
for k, v := range test.output {
|
|
expect := hashInfo.GetHash(k)
|
|
require.True(t, len(expect) > 0, "test output for hash not found")
|
|
assert.Equal(t, expect, v)
|
|
}
|
|
expect := hashInfo.GetHash(nil)
|
|
require.True(t, len(expect) == 0, "unknown type should return empty string")
|
|
str := hashInfo.String()
|
|
Log.Info("str=" + str)
|
|
newHi := FromString(str)
|
|
assert.Equal(t, newHi.h, hashInfo.h)
|
|
|
|
}
|
|
}
|