feat(observabillity): #153 instrument sqlx
This commit was merged in pull request #160.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package test_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"spend-sparrow/internal/db"
|
||||
"spend-sparrow/internal/types"
|
||||
"testing"
|
||||
@@ -26,7 +27,7 @@ func setupDb(t *testing.T) *sqlx.DB {
|
||||
}
|
||||
})
|
||||
|
||||
err = db.RunMigrations(d, "../")
|
||||
err = db.RunMigrations(context.Background(), d, "../")
|
||||
if err != nil {
|
||||
t.Fatalf("Error running migrations: %v", err)
|
||||
}
|
||||
@@ -47,14 +48,14 @@ func TestUser(t *testing.T) {
|
||||
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
||||
expected := types.NewUser(uuid.New(), "some@email.de", true, &verifiedAt, false, []byte("somePass"), []byte("someSalt"), createAt)
|
||||
|
||||
err := underTest.InsertUser(expected)
|
||||
err := underTest.InsertUser(context.Background(), expected)
|
||||
require.NoError(t, err)
|
||||
|
||||
actual, err := underTest.GetUser(expected.Id)
|
||||
actual, err := underTest.GetUser(context.Background(), expected.Id)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, expected, actual)
|
||||
|
||||
actual, err = underTest.GetUserByEmail(expected.Email)
|
||||
actual, err = underTest.GetUserByEmail(context.Background(), expected.Email)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, expected, actual)
|
||||
})
|
||||
@@ -64,7 +65,7 @@ func TestUser(t *testing.T) {
|
||||
|
||||
underTest := db.NewAuthSqlite(d)
|
||||
|
||||
_, err := underTest.GetUserByEmail("nonExistentEmail")
|
||||
_, err := underTest.GetUserByEmail(context.Background(), "nonExistentEmail")
|
||||
assert.Equal(t, db.ErrNotFound, err)
|
||||
})
|
||||
t.Run("should return ErrUserExist", func(t *testing.T) {
|
||||
@@ -77,10 +78,10 @@ func TestUser(t *testing.T) {
|
||||
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
||||
user := types.NewUser(uuid.New(), "some@email.de", true, &verifiedAt, false, []byte("somePass"), []byte("someSalt"), createAt)
|
||||
|
||||
err := underTest.InsertUser(user)
|
||||
err := underTest.InsertUser(context.Background(), user)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = underTest.InsertUser(user)
|
||||
err = underTest.InsertUser(context.Background(), user)
|
||||
assert.Equal(t, db.ErrAlreadyExists, err)
|
||||
})
|
||||
t.Run("should return ErrInternal on missing NOT NULL fields", func(t *testing.T) {
|
||||
@@ -92,7 +93,7 @@ func TestUser(t *testing.T) {
|
||||
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
||||
user := types.NewUser(uuid.New(), "some@email.de", false, nil, false, []byte("somePass"), nil, createAt)
|
||||
|
||||
err := underTest.InsertUser(user)
|
||||
err := underTest.InsertUser(context.Background(), user)
|
||||
assert.Equal(t, types.ErrInternal, err)
|
||||
})
|
||||
}
|
||||
@@ -110,21 +111,21 @@ func TestToken(t *testing.T) {
|
||||
expiresAt := createAt.Add(24 * time.Hour)
|
||||
expected := types.NewToken(uuid.New(), "sessionId", "token", types.TokenTypeCsrf, createAt, expiresAt)
|
||||
|
||||
err := underTest.InsertToken(expected)
|
||||
err := underTest.InsertToken(context.Background(), expected)
|
||||
require.NoError(t, err)
|
||||
|
||||
actual, err := underTest.GetToken(expected.Token)
|
||||
actual, err := underTest.GetToken(context.Background(), expected.Token)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, expected, actual)
|
||||
|
||||
expected.SessionId = ""
|
||||
actuals, err := underTest.GetTokensByUserIdAndType(expected.UserId, expected.Type)
|
||||
actuals, err := underTest.GetTokensByUserIdAndType(context.Background(), expected.UserId, expected.Type)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, []*types.Token{expected}, actuals)
|
||||
|
||||
expected.SessionId = "sessionId"
|
||||
expected.UserId = uuid.Nil
|
||||
actuals, err = underTest.GetTokensBySessionIdAndType(expected.SessionId, expected.Type)
|
||||
actuals, err = underTest.GetTokensBySessionIdAndType(context.Background(), expected.SessionId, expected.Type)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, []*types.Token{expected}, actuals)
|
||||
})
|
||||
@@ -140,14 +141,14 @@ func TestToken(t *testing.T) {
|
||||
expected1 := types.NewToken(userId, "sessionId", "token1", types.TokenTypeCsrf, createAt, expiresAt)
|
||||
expected2 := types.NewToken(userId, "sessionId", "token2", types.TokenTypeCsrf, createAt, expiresAt)
|
||||
|
||||
err := underTest.InsertToken(expected1)
|
||||
err := underTest.InsertToken(context.Background(), expected1)
|
||||
require.NoError(t, err)
|
||||
err = underTest.InsertToken(expected2)
|
||||
err = underTest.InsertToken(context.Background(), expected2)
|
||||
require.NoError(t, err)
|
||||
|
||||
expected1.UserId = uuid.Nil
|
||||
expected2.UserId = uuid.Nil
|
||||
actuals, err := underTest.GetTokensBySessionIdAndType(expected1.SessionId, expected1.Type)
|
||||
actuals, err := underTest.GetTokensBySessionIdAndType(context.Background(), expected1.SessionId, expected1.Type)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, []*types.Token{expected1, expected2}, actuals)
|
||||
|
||||
@@ -155,7 +156,7 @@ func TestToken(t *testing.T) {
|
||||
expected2.SessionId = ""
|
||||
expected1.UserId = userId
|
||||
expected2.UserId = userId
|
||||
actuals, err = underTest.GetTokensByUserIdAndType(userId, expected1.Type)
|
||||
actuals, err = underTest.GetTokensByUserIdAndType(context.Background(), userId, expected1.Type)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, []*types.Token{expected1, expected2}, actuals)
|
||||
})
|
||||
@@ -165,13 +166,13 @@ func TestToken(t *testing.T) {
|
||||
|
||||
underTest := db.NewAuthSqlite(d)
|
||||
|
||||
_, err := underTest.GetToken("nonExistent")
|
||||
_, err := underTest.GetToken(context.Background(), "nonExistent")
|
||||
assert.Equal(t, db.ErrNotFound, err)
|
||||
|
||||
_, err = underTest.GetTokensByUserIdAndType(uuid.New(), types.TokenTypeEmailVerify)
|
||||
_, err = underTest.GetTokensByUserIdAndType(context.Background(), uuid.New(), types.TokenTypeEmailVerify)
|
||||
assert.Equal(t, db.ErrNotFound, err)
|
||||
|
||||
_, err = underTest.GetTokensBySessionIdAndType("sessionId", types.TokenTypeEmailVerify)
|
||||
_, err = underTest.GetTokensBySessionIdAndType(context.Background(), "sessionId", types.TokenTypeEmailVerify)
|
||||
assert.Equal(t, db.ErrNotFound, err)
|
||||
})
|
||||
t.Run("should return ErrAlreadyExists", func(t *testing.T) {
|
||||
@@ -184,10 +185,10 @@ func TestToken(t *testing.T) {
|
||||
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
||||
user := types.NewUser(uuid.New(), "some@email.de", true, &verifiedAt, false, []byte("somePass"), []byte("someSalt"), createAt)
|
||||
|
||||
err := underTest.InsertUser(user)
|
||||
err := underTest.InsertUser(context.Background(), user)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = underTest.InsertUser(user)
|
||||
err = underTest.InsertUser(context.Background(), user)
|
||||
assert.Equal(t, db.ErrAlreadyExists, err)
|
||||
})
|
||||
t.Run("should return ErrInternal on missing NOT NULL fields", func(t *testing.T) {
|
||||
@@ -199,7 +200,7 @@ func TestToken(t *testing.T) {
|
||||
createAt := time.Date(2020, 1, 5, 12, 0, 0, 0, time.UTC)
|
||||
user := types.NewUser(uuid.New(), "some@email.de", false, nil, false, []byte("somePass"), nil, createAt)
|
||||
|
||||
err := underTest.InsertUser(user)
|
||||
err := underTest.InsertUser(context.Background(), user)
|
||||
assert.Equal(t, types.ErrInternal, err)
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user