/
tea_test.go
executable file
·106 lines (92 loc) · 7.37 KB
/
tea_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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package dongle
import (
"strconv"
"testing"
"github.com/stretchr/testify/assert"
)
var teaTests = []struct {
input string
key string
rounds int
toHex string
toBase64 string
}{
{"", "", 64, "", ""},
{"", "0123456789abcdef", 64, "", ""},
{"hello go", "0123456789abcdef", 0, "3f9a9d3f2f58277f", "P5qdPy9YJ38="},
{"hello world", "0123456789abcdef", 32, "e63b448416433bda1610497542fa66a7", "5jtEhBZDO9oWEEl1Qvpmpw=="},
{"hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world, hello world", "0123456789abcdef", 64, "bfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3527b3b10a4d8297e3f580fe6b42b368aed5195ce34ea5edcd0f74478ba9369708023b1a013f199ac10e3070b4134003e563adc981e1f34dcbfa8d956247c0dce3d10237bc95b856f16a79755d41c8503c93450b273110d9d93722af35615315b71cd2258c94efdb3a831e8fcd2166d8e", "v6jZViR8Dc49ECN7yVuFbxanl1XUHIUDyTRQsnMRDZ2TcirzVhUxW3HNIljJTv2zUns7EKTYKX4/WA/mtCs2iu1Rlc406l7c0PdEeLqTaXCAI7GgE/GZrBDjBwtBNAA+VjrcmB4fNNy/qNlWJHwNzj0QI3vJW4VvFqeXVdQchQPJNFCycxENnZNyKvNWFTFbcc0iWMlO/bNSezsQpNgpfj9YD+a0KzaK7VGVzjTqXtzQ90R4upNpcIAjsaAT8ZmsEOMHC0E0AD5WOtyYHh803L+o2VYkfA3OPRAje8lbhW8Wp5dV1ByFA8k0ULJzEQ2dk3Iq81YVMVtxzSJYyU79s1J7OxCk2Cl+P1gP5rQrNortUZXONOpe3ND3RHi6k2lwgCOxoBPxmawQ4wcLQTQAPlY63JgeHzTcv6jZViR8Dc49ECN7yVuFbxanl1XUHIUDyTRQsnMRDZ2TcirzVhUxW3HNIljJTv2zUns7EKTYKX4/WA/mtCs2iu1Rlc406l7c0PdEeLqTaXCAI7GgE/GZrBDjBwtBNAA+VjrcmB4fNNy/qNlWJHwNzj0QI3vJW4VvFqeXVdQchQPJNFCycxENnZNyKvNWFTFbcc0iWMlO/bNSezsQpNgpfj9YD+a0KzaK7VGVzjTqXtzQ90R4upNpcIAjsaAT8ZmsEOMHC0E0AD5WOtyYHh803L+o2VYkfA3OPRAje8lbhW8Wp5dV1ByFA8k0ULJzEQ2dk3Iq81YVMVtxzSJYyU79s1J7OxCk2Cl+P1gP5rQrNortUZXONOpe3ND3RHi6k2lwgCOxoBPxmawQ4wcLQTQAPlY63JgeHzTcv6jZViR8Dc49ECN7yVuFbxanl1XUHIUDyTRQsnMRDZ2TcirzVhUxW3HNIljJTv2zUns7EKTYKX4/WA/mtCs2iu1Rlc406l7c0PdEeLqTaXCAI7GgE/GZrBDjBwtBNAA+VjrcmB4fNNy/qNlWJHwNzj0QI3vJW4VvFqeXVdQchQPJNFCycxENnZNyKvNWFTFbcc0iWMlO/bNSezsQpNgpfj9YD+a0KzaK7VGVzjTqXtzQ90R4upNpcIAjsaAT8ZmsEOMHC0E0AD5WOtyYHh803L+o2VYkfA3OPRAje8lbhW8Wp5dV1ByFA8k0ULJzEQ2dk3Iq81YVMVtxzSJYyU79s6gx6PzSFm2O"},
}
func TestTea_Encrypt_String(t *testing.T) {
for index, test := range teaTests {
e1 := Encrypt.FromString(test.input).ByTea(test.key, test.rounds)
if test.rounds == 0 {
e1 = Encrypt.FromString(test.input).ByTea(test.key)
}
assert.Nil(t, e1.Error)
assert.Equal(t, test.toHex, e1.ToHexString(), "Hex test index is "+strconv.Itoa(index))
assert.Equal(t, test.toBase64, e1.ToBase64String(), "Base64 test index is "+strconv.Itoa(index))
e2 := Encrypt.FromString(test.input).ByTea(test.key, test.rounds)
if test.rounds == 0 {
e2 = Encrypt.FromString(test.input).ByTea(test.key)
}
assert.Nil(t, e2.Error)
assert.Equal(t, test.toHex, e2.ToHexString(), "Hex test index is "+strconv.Itoa(index))
assert.Equal(t, test.toBase64, e2.ToBase64String(), "Base64 test index is "+strconv.Itoa(index))
}
}
func TestTea_Decrypt_String(t *testing.T) {
for index, test := range teaTests {
d1 := Decrypt.FromHexString(test.toHex).ByTea(test.key, test.rounds)
if test.rounds == 0 {
d1 = Decrypt.FromHexString(test.toHex).ByTea(test.key)
}
assert.Nil(t, d1.Error)
assert.Equal(t, test.input, d1.ToString(), "Hex test index is "+strconv.Itoa(index))
d2 := Decrypt.FromBase64String(test.toBase64).ByTea(test.key, test.rounds)
if test.rounds == 0 {
d2 = Decrypt.FromBase64String(test.toBase64).ByTea(test.key)
}
assert.Nil(t, d2.Error)
assert.Equal(t, test.input, d2.ToString(), "base64 test index is "+strconv.Itoa(index))
}
}
func TestTea_Encrypt_Bytes(t *testing.T) {
for index, test := range teaTests {
e := Encrypt.FromBytes([]byte(test.input)).ByTea([]byte(test.key), test.rounds)
if test.rounds == 0 {
e = Encrypt.FromBytes([]byte(test.input)).ByTea([]byte(test.key))
}
assert.Nil(t, e.Error)
assert.Equal(t, []byte(test.toHex), e.ToHexBytes(), "Hex test test index is "+strconv.Itoa(index))
assert.Equal(t, []byte(test.toBase64), e.ToBase64Bytes(), "Base64 test index is "+strconv.Itoa(index))
}
}
func TestTea_Decrypt_Bytes(t *testing.T) {
for index, test := range teaTests {
e1 := Decrypt.FromHexBytes([]byte(test.toHex)).ByTea([]byte(test.key), test.rounds)
if test.rounds == 0 {
e1 = Decrypt.FromHexBytes([]byte(test.toHex)).ByTea([]byte(test.key))
}
assert.Nil(t, e1.Error)
assert.Equal(t, []byte(test.input), e1.ToBytes(), "Hex test index is "+strconv.Itoa(index))
e2 := Decrypt.FromBase64Bytes([]byte(test.toBase64)).ByTea([]byte(test.key), test.rounds)
if test.rounds == 0 {
e2 = Decrypt.FromBase64Bytes([]byte(test.toBase64)).ByTea([]byte(test.key))
}
assert.Nil(t, e2.Error)
assert.Equal(t, []byte(test.input), e2.ToBytes(), "base64 test index is "+strconv.Itoa(index))
}
}
func TestTea_Key_Error(t *testing.T) {
e := Encrypt.FromString("hello go").ByTea("xxxx", 8)
assert.Equal(t, invalidTeaKeyError(), e.Error)
d := Decrypt.FromRawString("hello go").ByTea("xxxx", 8)
assert.Equal(t, invalidTeaKeyError(), d.Error)
}
func TestTea_Rounds_Error(t *testing.T) {
e := Encrypt.FromString("hello go").ByTea("0123456789abcdefghijklmn", 1)
assert.Equal(t, invalidTeaRoundsError(), e.Error)
d := Decrypt.FromRawString("hello go").ByTea("0123456789abcdefghijklmn", 1)
assert.Equal(t, invalidTeaRoundsError(), d.Error)
}