55 lines
989 B
Go
55 lines
989 B
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type User struct {
|
|
user string
|
|
pass string
|
|
hashed string
|
|
}
|
|
|
|
func TestHtpasswdSuccess(t *testing.T) {
|
|
path := "/tmp/.htpasswd"
|
|
user1 := User{
|
|
user: "user",
|
|
pass: "password",
|
|
hashed: "$2a$14$SQSscaF4fVO3e5dp2/.VPuVQDPKqxSagLQnN6OncTRtoQw0ie9ByK",
|
|
}
|
|
err := ioutil.WriteFile(path,
|
|
[]byte(fmt.Sprintf("%s:%s", user1.user, user1.hashed)), 0640)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store, err := NewHtpasswd(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = store.Authenticate(user1.user, user1.pass)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
user2 := User{
|
|
user: "foo",
|
|
pass: "bar",
|
|
}
|
|
err = store.Register(user2.user, user2.pass)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
data, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
for _, u := range []User{user1, user2} {
|
|
if !strings.Contains(string(data), u.user) {
|
|
t.Errorf("%s not found in htpasswd file: %s", u.user, string(data))
|
|
}
|
|
}
|
|
}
|