all: sync with master
This commit is contained in:
83
internal/aghuser/db_test.go
Normal file
83
internal/aghuser/db_test.go
Normal file
@@ -0,0 +1,83 @@
|
||||
package aghuser_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/AdguardTeam/AdGuardHome/internal/aghuser"
|
||||
"github.com/AdguardTeam/golibs/errors"
|
||||
"github.com/AdguardTeam/golibs/testutil"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
)
|
||||
|
||||
func TestDB(t *testing.T) {
|
||||
db := aghuser.NewDefaultDB()
|
||||
|
||||
const (
|
||||
userWithIDPassRaw = "user_with_id_password"
|
||||
userSecondPassRaw = "user_second_password"
|
||||
)
|
||||
|
||||
userWithIDPassHash, err := bcrypt.GenerateFromPassword(
|
||||
[]byte(userWithIDPassRaw),
|
||||
bcrypt.DefaultCost,
|
||||
)
|
||||
require.NoError(t, err)
|
||||
|
||||
userSecondPassHash, err := bcrypt.GenerateFromPassword(
|
||||
[]byte(userSecondPassRaw),
|
||||
bcrypt.DefaultCost,
|
||||
)
|
||||
require.NoError(t, err)
|
||||
|
||||
userWithIDPass := aghuser.NewDefaultPassword(string(userWithIDPassHash))
|
||||
userSecondPass := aghuser.NewDefaultPassword(string(userSecondPassHash))
|
||||
|
||||
var (
|
||||
userWithID = &aghuser.User{
|
||||
ID: aghuser.MustNewUserID(),
|
||||
Login: "user_with_id",
|
||||
Password: userWithIDPass,
|
||||
}
|
||||
userSecond = &aghuser.User{
|
||||
ID: aghuser.MustNewUserID(),
|
||||
Login: "user_second",
|
||||
Password: userSecondPass,
|
||||
}
|
||||
userDuplicateLogin = &aghuser.User{
|
||||
ID: aghuser.MustNewUserID(),
|
||||
Login: userWithID.Login,
|
||||
Password: userWithIDPass,
|
||||
}
|
||||
)
|
||||
|
||||
ctx := testutil.ContextWithTimeout(t, testTimeout)
|
||||
|
||||
err = db.Create(ctx, userWithID)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = db.Create(ctx, userSecond)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = db.Create(ctx, userDuplicateLogin)
|
||||
assert.ErrorIs(t, err, errors.ErrDuplicated)
|
||||
|
||||
got, err := db.ByUUID(ctx, userWithID.ID)
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.Equal(t, userWithID, got)
|
||||
assert.True(t, got.Password.Authenticate(ctx, userWithIDPassRaw))
|
||||
|
||||
got, err = db.ByLogin(ctx, userSecond.Login)
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.Equal(t, userSecond, got)
|
||||
assert.True(t, got.Password.Authenticate(ctx, userSecondPassRaw))
|
||||
|
||||
users, err := db.All(ctx)
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.Len(t, users, 2)
|
||||
assert.Equal(t, []*aghuser.User{userSecond, userWithID}, users)
|
||||
}
|
||||
Reference in New Issue
Block a user