wrouter/auth/authenticator_test.go

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()
		}
	})
}