113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
package db
|
|
|
|
import (
|
|
"me-fit/utils"
|
|
|
|
"database/sql"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func setupDb(t *testing.T) *sql.DB {
|
|
db, err := sql.Open("sqlite3", ":memory:")
|
|
if err != nil {
|
|
t.Fatalf("Error opening database: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
db.Close()
|
|
})
|
|
|
|
err = utils.RunMigrations(db, "../")
|
|
if err != nil {
|
|
t.Fatalf("Error running migrations: %v", err)
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func TestUser(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("should return UserNotFound", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthDbSqlite{db: db}
|
|
|
|
_, err := underTest.GetUser("someNonExistentEmail")
|
|
assert.Equal(t, ErrUserNotFound, err)
|
|
})
|
|
|
|
t.Run("should insert and get user", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthDbSqlite{db: db}
|
|
|
|
verifiedAt := time.Date(2020, 1, 5, 13, 0, 0, 0, time.UTC)
|
|
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
|
expected := NewUser(uuid.New(), "some@email.de", true, &verifiedAt, false, []byte("somePass"), []byte("someSalt"), createAt)
|
|
|
|
err := underTest.InsertUser(expected)
|
|
assert.Nil(t, err)
|
|
|
|
actual, err := underTest.GetUser(expected.Email)
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, expected, actual)
|
|
})
|
|
|
|
t.Run("should throw error if user already exists", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthDbSqlite{db: db}
|
|
|
|
verifiedAt := time.Date(2020, 1, 5, 13, 0, 0, 0, time.UTC)
|
|
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
|
user := NewUser(uuid.New(), "some@email.de", true, &verifiedAt, false, []byte("somePass"), []byte("someSalt"), createAt)
|
|
|
|
err := underTest.InsertUser(user)
|
|
assert.Nil(t, err)
|
|
|
|
err = underTest.InsertUser(user)
|
|
assert.Equal(t, ErrUserExists, err)
|
|
})
|
|
}
|
|
|
|
func TestEmailVerification(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("should return empty string if no token is safed", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthDbSqlite{db: db}
|
|
|
|
token, err := underTest.GetEmailVerificationToken(uuid.New())
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "", token)
|
|
})
|
|
t.Run("should insert and return token", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthDbSqlite{db: db}
|
|
|
|
userId := uuid.New()
|
|
expectedToken := "someToken"
|
|
|
|
err := underTest.InsertEmailVerificationToken(userId, expectedToken)
|
|
assert.Nil(t, err)
|
|
|
|
actualToken, err := underTest.GetEmailVerificationToken(userId)
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, expectedToken, actualToken)
|
|
})
|
|
}
|