50 lines
947 B
Go
50 lines
947 B
Go
package tokens_test
|
|
|
|
import (
|
|
"context"
|
|
"gitea.deepak.science/deepak/gogmagog/tokens"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestGoodContext(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
idToUse := 3
|
|
username := "username"
|
|
|
|
ctx := tokens.GetContextForUserValues(idToUse, username)
|
|
|
|
receivedID, err := tokens.GetUserID(ctx)
|
|
assert.Nil(err)
|
|
assert.EqualValues(idToUse, receivedID)
|
|
|
|
receivedUsername, err := tokens.GetUsername(ctx)
|
|
assert.Nil(err)
|
|
assert.Equal(username, receivedUsername)
|
|
|
|
}
|
|
|
|
func TestBadContext(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
ctx := context.Background()
|
|
|
|
_, err := tokens.GetUserID(ctx)
|
|
assert.NotNil(err)
|
|
|
|
_, err = tokens.GetUsername(ctx)
|
|
assert.NotNil(err)
|
|
|
|
}
|
|
|
|
func TestSetContext(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
idToUse := 3
|
|
ctx := tokens.SetUserID(context.Background(), 3)
|
|
receivedID, err := tokens.GetUserID(ctx)
|
|
assert.Nil(err)
|
|
assert.EqualValues(idToUse, receivedID)
|
|
}
|