128 lines
3.1 KiB
Go
128 lines
3.1 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"me-fit/types"
|
|
"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 = RunMigrations(db, "../")
|
|
if err != nil {
|
|
t.Fatalf("Error running migrations: %v", err)
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func TestUser(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("should insert and get the same", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{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.Id)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, expected, actual)
|
|
|
|
actual, err = underTest.GetUserByEmail(expected.Email)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, expected, actual)
|
|
})
|
|
t.Run("should return UserNotFound", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{db: db}
|
|
|
|
_, err := underTest.GetUserByEmail("nonExistentEmail")
|
|
assert.Equal(t, ErrNotFound, err)
|
|
})
|
|
t.Run("should return ErrUserExist", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{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)
|
|
})
|
|
t.Run("should return ErrInternal on missing NOT NULL fields", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{db: db}
|
|
|
|
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
|
user := NewUser(uuid.New(), "some@email.de", false, nil, false, []byte("somePass"), nil, createAt)
|
|
|
|
err := underTest.InsertUser(user)
|
|
assert.Equal(t, types.ErrInternal, err)
|
|
})
|
|
}
|
|
|
|
func TestEmailVerification(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("should return NotFound", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{db: db}
|
|
|
|
token, err := underTest.GetToken("someNonExistentToken")
|
|
|
|
assert.Equal(t, ErrNotFound, err)
|
|
assert.Nil(t, token)
|
|
})
|
|
t.Run("should insert and return token", func(t *testing.T) {
|
|
t.Parallel()
|
|
db := setupDb(t)
|
|
|
|
underTest := AuthSqlite{db: db}
|
|
tokenStr := "some secure token"
|
|
createdAt := time.Date(2020, 1, 5, 13, 0, 0, 0, time.UTC)
|
|
|
|
expectedToken := NewToken(uuid.New(), tokenStr, TokenTypeEmailVerify, createdAt, createdAt.Add(24*time.Hour))
|
|
|
|
err := underTest.InsertToken(expectedToken)
|
|
assert.Nil(t, err)
|
|
|
|
actualToken, err := underTest.GetToken(tokenStr)
|
|
assert.Nil(t, err)
|
|
|
|
t.Logf("expectedToken: %v", expectedToken)
|
|
t.Logf("actualToken: %v", actualToken)
|
|
assert.Equal(t, expectedToken, actualToken)
|
|
})
|
|
}
|