74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package authorization
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"github.com/zeebo/errs"
|
||
|
|
"go.uber.org/zap/zaptest"
|
||
|
|
|
||
|
|
"storj.io/storj/internal/testcontext"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestService_GetOrCreate(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
authorizationDB := newTestAuthDB(t, ctx)
|
||
|
|
service := NewService(zaptest.NewLogger(t), authorizationDB)
|
||
|
|
require.NotNil(t, service)
|
||
|
|
|
||
|
|
{ // new user, no existing authorization tokens (create)
|
||
|
|
userID := "new@mail.test"
|
||
|
|
group, err := authorizationDB.Get(ctx, userID)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Empty(t, group)
|
||
|
|
|
||
|
|
token, err := service.GetOrCreate(ctx, userID)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotNil(t, token)
|
||
|
|
|
||
|
|
require.Equal(t, userID, token.UserID)
|
||
|
|
}
|
||
|
|
|
||
|
|
{ // existing user with unclaimed authorization token (get)
|
||
|
|
userID := "old@mail.test"
|
||
|
|
group, err := authorizationDB.Create(ctx, userID, 1)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, group)
|
||
|
|
require.Len(t, group, 1)
|
||
|
|
|
||
|
|
existingAuth := group[0]
|
||
|
|
|
||
|
|
token, err := service.GetOrCreate(ctx, userID)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotNil(t, token)
|
||
|
|
|
||
|
|
require.Equal(t, userID, token.UserID)
|
||
|
|
require.Equal(t, existingAuth.Token, *token)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestService_GetOrCreate_error(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
listener, err := net.Listen("tcp", "127.0.0.1:0")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotNil(t, listener)
|
||
|
|
|
||
|
|
authorizationDB := newTestAuthDB(t, ctx)
|
||
|
|
service := NewService(zaptest.NewLogger(t), authorizationDB)
|
||
|
|
require.NotNil(t, service)
|
||
|
|
|
||
|
|
{ // empty user ID
|
||
|
|
token, err := service.GetOrCreate(ctx, "")
|
||
|
|
require.Error(t, errs.Unwrap(err), ErrEmptyUserID.Error())
|
||
|
|
require.Nil(t, token)
|
||
|
|
}
|
||
|
|
}
|