-
Notifications
You must be signed in to change notification settings - Fork 170
/
generator_test.go
74 lines (59 loc) · 2 KB
/
generator_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
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package password
import (
"strings"
"testing"
"github.com/stretchr/testify/require"
)
func TestZeroLengthPasswordError(t *testing.T) {
_, err := Generate(PasswordComposition{})
require.Error(t, err)
}
func TestOneCharPassword(t *testing.T) {
// Really weak, so not practically usable, but we should be able to generate these 1-char passwords without errors
var pwd string
var err error
pwd, err = Generate(PasswordComposition{NumLowercase: 1})
require.NoError(t, err)
require.Len(t, pwd, 1)
require.Equal(t, 1, countCharsFrom(pwd, LowercaseLetters))
pwd, err = Generate(PasswordComposition{NumUppercase: 1})
require.NoError(t, err)
require.Len(t, pwd, 1)
require.Equal(t, 1, countCharsFrom(pwd, UppercaseLetters))
pwd, err = Generate(PasswordComposition{NumDigits: 1})
require.NoError(t, err)
require.Len(t, pwd, 1)
require.Equal(t, 1, countCharsFrom(pwd, Digits))
pwd, err = Generate(PasswordComposition{NumSymbols: 1})
require.NoError(t, err)
require.Len(t, pwd, 1)
require.Equal(t, 1, countCharsFrom(pwd, Symbols))
}
func TestPasswordContainsRequestedChars(t *testing.T) {
pwd, err := Generate(PasswordComposition{
NumLowercase: 3,
NumUppercase: 4,
NumDigits: 5,
NumSymbols: 6,
})
require.NoError(t, err)
require.Equal(t, 3, countCharsFrom(pwd, LowercaseLetters))
require.Equal(t, 4, countCharsFrom(pwd, UppercaseLetters))
require.Equal(t, 5, countCharsFrom(pwd, Digits))
require.Equal(t, 6, countCharsFrom(pwd, Symbols))
}
func TestPasswordShuffled(t *testing.T) {
pwd, err := Generate(PasswordComposition{NumLowercase: 10, NumUppercase: 20})
require.NoError(t, err)
// Should be super improbable for the lowercase letters to remain at the front
require.Less(t, countCharsFrom(string(pwd[0:10]), LowercaseLetters), 10)
}
func countCharsFrom(s, choices string) int {
count := 0
for i := 0; i < len(choices); i++ {
count += strings.Count(s, string(choices[i]))
}
return count
}