49 lines
1 KiB
Go
49 lines
1 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func hashedKeyCompare(t *testing.T, a, b string, shouldMatch bool) {
|
||
|
|
err := SetConfigPassword(a)
|
||
|
|
require.NoError(t, err)
|
||
|
|
k1 := configKey
|
||
|
|
|
||
|
|
err = SetConfigPassword(b)
|
||
|
|
require.NoError(t, err)
|
||
|
|
k2 := configKey
|
||
|
|
|
||
|
|
if shouldMatch {
|
||
|
|
assert.Equal(t, k1, k2)
|
||
|
|
} else {
|
||
|
|
assert.NotEqual(t, k1, k2)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPassword(t *testing.T) {
|
||
|
|
defer func() {
|
||
|
|
configKey = nil // reset password
|
||
|
|
}()
|
||
|
|
var err error
|
||
|
|
// Empty password should give error
|
||
|
|
err = SetConfigPassword(" \t ")
|
||
|
|
require.Error(t, err)
|
||
|
|
|
||
|
|
// Test invalid utf8 sequence
|
||
|
|
err = SetConfigPassword(string([]byte{0xff, 0xfe, 0xfd}) + "abc")
|
||
|
|
require.Error(t, err)
|
||
|
|
|
||
|
|
// Simple check of wrong passwords
|
||
|
|
hashedKeyCompare(t, "mis", "match", false)
|
||
|
|
|
||
|
|
// Check that passwords match after unicode normalization
|
||
|
|
hashedKeyCompare(t, "ff\u0041\u030A", "ffÅ", true)
|
||
|
|
|
||
|
|
// Check that passwords preserves case
|
||
|
|
hashedKeyCompare(t, "abcdef", "ABCDEF", false)
|
||
|
|
|
||
|
|
}
|