-
Notifications
You must be signed in to change notification settings - Fork 0
/
checksummer_test.go
81 lines (69 loc) · 2.58 KB
/
checksummer_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
// Copyright: Jostein Stuhaug
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package fletcher4
import (
"bytes"
"fmt"
"testing"
)
type hexRes [4]string
func compare(t *testing.T, leadTxt string, exp hexRes, got [4]uint64) {
gotHex := fmt.Sprintf("%x/%x/%x/%x", got[0], got[1], got[2], got[3])
expHex := fmt.Sprintf("%v/%v/%v/%v", exp[0], exp[1], exp[2], exp[3])
if expHex != gotHex {
t.Errorf("%v:\nexpected\t%v,\ngot\t\t%v", leadTxt, expHex, gotHex)
}
}
// Test that writing 4 bytes fills all checksum result uints with the same 4 bytes
func TestChecksummer1(t *testing.T) {
inp1 := []byte{1, 2, 3, 4}
exp1 := hexRes{"4030201", "4030201", "4030201", "4030201"}
checksummer := New()
if _, err := checksummer.Write(inp1); err != nil {
t.Fatal(err)
}
res1 := checksummer.Sum64x4()
compare(t, "Checksum test 1, 4 bytes failed", exp1, res1)
}
// Test that writing 8 bytes and then 4 more gives correct result.
// Also test that Sum return correct result
func TestChecksummer2(t *testing.T) {
inp1 := []byte{1, 2, 3, 4, 5, 6, 7, 8}
exp1 := hexRes{"c0a0806", "100d0a07", "14100c08", "18130e09"}
checksummer := New()
if _, err := checksummer.Write(inp1); err != nil {
t.Fatal(err)
}
res1 := checksummer.Sum64x4()
compare(t, "Checksum test 2, 8 bytes written failed", exp1, res1)
inp2 := []byte{2, 4, 6, 8}
exp2 := hexRes{"14100c08", "241d160f", "382d2217", "50403020"}
if _, err := checksummer.Write(inp2); err != nil {
t.Fatal(err)
}
res2 := checksummer.Sum64x4()
compare(t, "Checksum test 2, 12 bytes written failed", exp2, res2)
var sum []byte
sum = checksummer.Sum(sum)
expSum := []byte{8, 12, 16, 20, 0, 0, 0, 0, 15, 22, 29, 36, 0, 0, 0, 0, 23, 34, 45, 56, 0, 0, 0, 0, 32, 48, 64, 80, 0, 0, 0, 0}
if !bytes.Equal(sum, expSum) {
t.Errorf("Checksum Sum method call 1 returned wrong result.\nExpected %x,\ngot: %x)", sum, expSum)
}
// Test that checksummer.Sum appends correctly
sum = checksummer.Sum(sum)
expSum2 := append(expSum, expSum...)
if !bytes.Equal(sum, expSum2) {
t.Errorf("Checksum Sum method call 2 returned wrong result.\nExpected %x,\ngot: %x)", sum, expSum2)
}
}