user_service_basic_test.go

package gimlet

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestBasicUserManager(t *testing.T) {
	const expectedToken = "0:baz:[56 88 246 34 48 172 60 145 95 48 12 102 67 18 198 63]"
	assert := assert.New(t)
	assert.Implements((*UserManager)(nil), &BasicUserManager{})

	u := BasicUserManager{users: []BasicUser{
		{
			ID:           "foo",
			Password:     "bar",
			EmailAddress: "baz",
		},
	}}
	user, err := u.GetUserByToken(context.Background(), expectedToken)
	assert.NoError(err)
	assert.NotNil(user)
	assert.Equal("foo", user.Username())
	assert.Equal("baz", user.Email())

	user, err = u.GetUserByToken(context.Background(), "")
	assert.Error(err)
	assert.Nil(user)

	token, err := u.CreateUserToken("foo", "bar")
	assert.NoError(err)
	assert.Equal(token, expectedToken)

	assert.Nil(u.GetLoginHandler(""))
	assert.Nil(u.GetLoginCallbackHandler())
	assert.False(u.IsRedirect())
	assert.Nil(u.ReauthorizeUser(&u.users[0]))
	assert.NotNil(u.ReauthorizeUser(&BasicUser{ID: "bar"}))

	user, err = u.GetUserByID("bar")
	assert.Error(err)
	assert.Nil(user)

	user, err = u.GetUserByID("foo")
	assert.NoError(err)
	assert.NotNil(user)
	assert.Equal("foo", user.Username())
	assert.Equal("baz", user.Email())

	newUser := &BasicUser{ID: "foo"}
	user, err = u.GetOrCreateUser(newUser)
	assert.NoError(err)
	assert.NotNil(user)
	assert.Equal("foo", user.Username())
	assert.Equal("baz", user.Email())

	newUser = &BasicUser{ID: "new_user", Password: "password", EmailAddress: "email@example.com"}
	user, err = u.GetOrCreateUser(newUser)
	assert.NoError(err)
	assert.NotNil(user)
	assert.Equal("new_user", user.Username())
	assert.Equal("email@example.com", user.Email())

	assert.False(u.isInvalid(user.Username()))
	u.setInvalid(user.Username(), true)
	assert.True(u.isInvalid(user.Username()))
	returnedUser, err := u.GetUserByID(user.Username())
	assert.Nil(returnedUser)
	assert.Error(err)
	u.setInvalid(user.Username(), false)
	assert.False(u.isInvalid(user.Username()))
	assert.True(u.isInvalid("DNE"))

	assert.Error(u.ClearUser(newUser, false))
}