-
Notifications
You must be signed in to change notification settings - Fork 1
/
charset_test.go
132 lines (116 loc) · 4.82 KB
/
charset_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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package passit
import (
"testing"
"unicode"
"unicode/utf8"
"github.com/stretchr/testify/assert"
"golang.org/x/text/unicode/rangetable"
)
func TestCharset(t *testing.T) {
for _, tc := range []struct{ expect, charset string }{
{"", ""},
{"~~~~~~~~~~~~~~~~~~~~~~~~~", "~"},
{"0110100100010010000000010", "01"},
{"2352984966922368666874797", "0123456789"},
{"yzxeishgyluaruksywwtcxdji", "abcdefghijklmnopqrstuvwxyz"},
{"YZXEISHGYLUARUKSYWWTCXDJI", "ABCDEFGHIJKLMNOPQRSTUVWXYZ"},
{"yXisHgYluArukyWwtCXdjIRmD", "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"},
{"ovNa1Os7MObQ0ruaoUCwj2DdZ", "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"},
{"y+)rvfut_lhnehk%ljjtpk#ji", "abcdefghijklmnopqrstuvwxyz~!@#$%^&*()_+"},
{"ΓΤΞΙΧυχΖτξυΔβΧψΥΠΖΟωωγοοτ", "ΑαΒβΓγΔδΕεΖζΗηΘθΙιΚκΛλΜμΝνΞξΟοΠπΡρΣσςΤτΥυΦφΧχΨψΩω"},
{"👒💩👗🎩🚋🚱💬🚋🌴🌴🍳👒🎩🍉🛰💅🛰🛰🛰💅💻🍈🐝🍳🐝", "🔱🍧👒🍉💬👞🛰🐝💅🍳🐊🐂🎩💩🍈👗🌴💻🚱🚋"},
} {
tr := newTestRand()
pass, err := Repeat(FromCharset(tc.charset), "", 25).Password(tr)
if !assert.NoError(t, err) {
continue
}
assert.Equal(t, tc.expect, pass)
assert.Truef(t, utf8.ValidString(pass),
"utf8.ValidString(%q)", pass)
allRunesAllowed(t, tc.charset, pass)
}
}
func TestFixedCharset(t *testing.T) {
for _, tc := range []struct {
expect string
gen Generator
}{
{"2352984966922368666874797", Digit},
{"yzxeishgyluaruksywwtcxdji", LatinLower},
{"YZXEISHGYLUARUKSYWWTCXDJI", LatinUpper},
{"YxIShGyLUaRUKYwWTcxDJirMd", LatinMixed},
{"4rd6x4ix2e8rwqhkqk08smzst", LatinLowerDigit},
{"4RD6X4IX2E8RWQHKQK08SMZST", LatinUpperDigit},
{"OVnA1oS7moBq0RUAOucWJ2dDz", LatinMixedDigit},
} {
const size = 25
tr := newTestRand()
pass, err := Repeat(tc.gen, "", size).Password(tr)
if !assert.NoError(t, err) {
continue
}
assert.Equal(t, tc.expect, pass)
assert.Equal(t, size, utf8.RuneCountInString(pass),
"utf8.RuneCountInString(%q)", pass)
assert.Truef(t, utf8.ValidString(pass),
"utf8.ValidString(%q)", pass)
allRunesAllowed(t, tc.gen.(*asciiGenerator).s, pass)
}
}
func TestRangeTable(t *testing.T) {
newTable := func(s string) *unicode.RangeTable {
return rangetable.New([]rune(s)...)
}
type testCase struct {
expect string
tab *unicode.RangeTable
}
testCases := []testCase{
{"", new(unicode.RangeTable)},
{"~~~~~~~~~~~~~~~~~~~~~~~~~", newTable("~")},
{"0110100100010010000000010", newTable("01")},
{"2352984966922368666874797", newTable("0123456789")},
{"yzxeishgyluaruksywwtcxdji", newTable("abcdefghijklmnopqrstuvwxyz")},
{"YZXEISHGYLUARUKSYWWTCXDJI", newTable("ABCDEFGHIJKLMNOPQRSTUVWXYZ")},
{"yXisHgYluArukyWwtCXdjIRmD", newTable("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz")},
{"elDQrEixCERGqhkQeK2mZs3TP", newTable("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")},
{"m~yfj(ihz_*b&*^t_@@hd^r@+", newTable("abcdefghijklmnopqrstuvwxyz~!@#$%^&*()_+")},
{"ΕξγΡτρυΛοδρΗΔτχπηΛεωωΖζζο", newTable("ΑαΒβΓγΔδΕεΖζΗηΘθΙιΚκΛλΜμΝνΞξΟοΠπΡρΣσςΤτΥυΦφΧχΨψΩω")},
{"🍉💩💻💅🛰🚱🍳🛰🔱🔱👒🍉💅🍧🐂🐝🐂🐂🐂🐝🚋💬🐊👒🐊", newTable("🔱🍧👒🍉💬👞🛰🐝💅🍳🐊🐂🎩💩🍈👗🌴💻🚱🚋")},
{"ED9Ed60F4A148f2068a49Ab7f", unicode.ASCII_Hex_Digit},
}
const unicodeVersion = "15.0.0"
testCasesUni := []testCase{
{"ʊⅴXǞǟṓŅ𝼈Ņꝧ𐞓Ꞇ𝼕áÆẞḔꝅḩʎĀĮꟑTf", unicode.Latin},
{"ἳ𝈛ῥᵡ𐅺όΫ𐅷𐆎ἓ𐅖Ί𝈂ΗᾁῈϼᴧρὺᵞ𐅰Ϟ𐅬θ", unicode.Greek},
{"₷₼૱₧﷼૱₼𑿟৳௹𞋿₪௹$₻₾₩₵₡𞋿$⃀₲£¢", unicode.Sc},
}
if unicode.Version == unicodeVersion {
testCases = append(testCases, testCasesUni...)
} else {
t.Logf("skipping %d test cases due to mismatched unicode versions; have %s, want %s", len(testCasesUni), unicode.Version, unicodeVersion)
}
for _, tc := range testCases {
tr := newTestRand()
pass, err := Repeat(FromRangeTable(tc.tab), "", 25).Password(tr)
if !assert.NoError(t, err) {
continue
}
assert.Equal(t, tc.expect, pass)
assert.Truef(t, utf8.ValidString(pass),
"utf8.ValidString(%q)", pass)
allRunesAllowed(t, tc.tab, pass)
}
}
func BenchmarkCharsetPassword(b *testing.B) {
for _, tc := range []struct{ name, charset string }{
{"ASCII", "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"},
{"Unicode", "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyzΑαΒβΓγΔδΕεΖζΗηΘθΙιΚκΛλΜμΝνΞξΟοΠπΡρΣσςΤτΥυΦφΧχΨψΩω0123456789"},
} {
tc := tc
b.Run(tc.name, func(b *testing.B) {
benchmarkGeneratorPassword(b, FromCharset(tc.charset))
})
}
}