mirror of https://github.com/jech/galene.git
138 lines
3.0 KiB
Go
138 lines
3.0 KiB
Go
package group
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
var key1 = ""
|
|
var pw1 = Password{
|
|
Type: "plain",
|
|
Key: &key1,
|
|
}
|
|
var key2 = "pass"
|
|
var pw2 = Password{
|
|
Type: "plain",
|
|
Key: &key2,
|
|
}
|
|
var key3 = "fe499504e8f144693fae828e8e371d50e019d0e4c84994fa03f7f445bd8a570a"
|
|
var pw3 = Password{
|
|
Type: "pbkdf2",
|
|
Hash: "sha-256",
|
|
Key: &key3,
|
|
Salt: "bcc1717851030776",
|
|
Iterations: 4096,
|
|
}
|
|
var key4 = "$2a$10$afOr2f33onT/nDFFyT3mbOq5FMSw1wWXfyTXQTBMbKvZpBkoD3Qwu"
|
|
var pw4 = Password{
|
|
Type: "bcrypt",
|
|
Key: &key4,
|
|
}
|
|
var pw5 = Password{}
|
|
var pw6 = Password{
|
|
Type: "bad",
|
|
}
|
|
|
|
func TestGood(t *testing.T) {
|
|
if match, err := pw1.Match(""); err != nil || !match {
|
|
t.Errorf("pw2 doesn't match (%v)", err)
|
|
}
|
|
if match, err := pw2.Match("pass"); err != nil || !match {
|
|
t.Errorf("pw2 doesn't match (%v)", err)
|
|
}
|
|
if match, err := pw3.Match("pass"); err != nil || !match {
|
|
t.Errorf("pw3 doesn't match (%v)", err)
|
|
}
|
|
if match, err := pw4.Match("pass"); err != nil || !match {
|
|
t.Errorf("pw4 doesn't match (%v)", err)
|
|
}
|
|
}
|
|
|
|
func TestBad(t *testing.T) {
|
|
if match, err := pw2.Match("bad"); err != nil || match {
|
|
t.Errorf("pw2 matches")
|
|
}
|
|
if match, err := pw3.Match("bad"); err != nil || match {
|
|
t.Errorf("pw3 matches")
|
|
}
|
|
if match, err := pw4.Match("bad"); err != nil || match {
|
|
t.Errorf("pw4 matches")
|
|
}
|
|
if match, err := pw5.Match(""); err != nil || match {
|
|
t.Errorf("pw5 matches")
|
|
}
|
|
if match, err := pw5.Match("bad"); err != nil || match {
|
|
t.Errorf("pw5 matches")
|
|
}
|
|
if match, err := pw6.Match("bad"); err == nil || match {
|
|
t.Errorf("pw6 matches")
|
|
}
|
|
}
|
|
|
|
func TestEmptyKey(t *testing.T) {
|
|
for _, tpe := range []string{"plain", "pbkdf2", "bcrypt", "bad"} {
|
|
pw := Password{Type: tpe}
|
|
if match, err := pw.Match(""); err == nil || match {
|
|
t.Errorf("empty password of type %v didn't error", tpe)
|
|
}
|
|
}
|
|
|
|
pw := Password{}
|
|
if match, err := pw.Match(""); err != nil || match {
|
|
t.Errorf("empty password empty type matched")
|
|
}
|
|
}
|
|
|
|
func TestJSON(t *testing.T) {
|
|
plain, err := json.Marshal(pw2)
|
|
if err != nil || string(plain) != `"pass"` {
|
|
t.Errorf("Expected \"pass\", got %v", string(plain))
|
|
}
|
|
|
|
for _, pw := range []Password{pw1, pw2, pw3, pw4, pw5} {
|
|
j, err := json.Marshal(pw)
|
|
if err != nil {
|
|
t.Fatalf("Marshal: %v", err)
|
|
}
|
|
if testing.Verbose() {
|
|
log.Printf("%v", string(j))
|
|
}
|
|
var pw2 Password
|
|
err = json.Unmarshal(j, &pw2)
|
|
if err != nil {
|
|
t.Errorf("Unmarshal: %v", err)
|
|
} else if !reflect.DeepEqual(pw, pw2) {
|
|
t.Errorf("Expected %v, got %v", pw, pw2)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkPlain(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
match, err := pw2.Match("bad")
|
|
if err != nil || match {
|
|
b.Errorf("pw2 matched")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkPBKDF2(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
match, err := pw3.Match("bad")
|
|
if err != nil || match {
|
|
b.Errorf("pw3 matched")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkBCrypt(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
match, err := pw4.Match("bad")
|
|
if err != nil || match {
|
|
b.Errorf("pw3 matched")
|
|
}
|
|
}
|
|
}
|