package auth import ( "errors" "strings" "testing" ) var ErrExists = errors.New("auth: user exists") var ErrLogin = errors.New("auth: login failed") type testAuther struct { FullName string users []*User passwords map[string]string } func (ta *testAuther) ID() string { return strings.ToLower(ta.FullName) } func (ta *testAuther) Name() string { return ta.FullName } func (ta *testAuther) Find(userid string) *User { for _, user := range ta.users { if user.ID == userid { return user } } return nil } func (ta *testAuther) Login(userid, password string) (*User, error) { for _, user := range ta.users { if user.ID == userid && password == ta.passwords[user.ID] { return user, nil } } return nil, ErrLogin } func (ta *testAuther) Register(userid, password string, data map[string]string) (*User, error) { if ta.Find(userid) != nil { return nil, ErrExists } if ta.passwords == nil { ta.passwords = make(map[string]string) } user := NewUser(ta, userid, "member", data) ta.users = append(ta.users, user) ta.passwords[userid] = password return user, nil } func TestAuthenticator(t *testing.T) { ta1 := testAuther{FullName: "Auth1"} ta2 := testAuther{FullName: "Auth2"} Register(&ta1) Register(&ta2) if ta1.ID() != "auth1" { t.Errorf("ta1.ID() = %s", ta1.ID()) t.Fail() } if ta2.ID() != "auth2" { t.Errorf("ta2.ID() = %s", ta2.ID()) t.Fail() } t.Run("Find", func(t *testing.T) { fa1 := FindAuthenticator("auth1") fa2 := FindAuthenticator("auth2") if &ta1 != fa1 { t.Errorf("%s != %s", ta1.ID(), fa1.ID()) t.Fail() } if &ta2 != fa2 { t.Errorf("%s != %s", ta2.ID(), fa2.ID()) t.Fail() } }) t.Run("Register", func(t *testing.T) { user, err := ta1.Register("Test", "CakesAndStuff", nil) if err != nil || user.ID != "Test" { t.Logf("err = %v; name = \"%s\"", err, user.ID) t.Fail() } if ta1.Find("Test") == nil { t.Log("Registered user does not exist") t.Fail() } user2, err := ta1.Register("Test", "CakesAndStuff", nil) if err == nil || user2 != nil { t.Logf("err = %s; name = \"%s\"", err, user2.ID) t.Fail() } }) t.Run("Login", func(t *testing.T) { user, err := ta1.Login("Test", "CakesAndStuff") if err != nil || user.ID != "Test" { t.Logf("err = %v; user = %+v", err, user) t.Fail() } user2, err := ta1.Login("Test", "WrongPassword") if err == nil || user2 != nil { t.Logf("err = %v; user = %+v", err, user2) t.Fail() } }) }