chore: add debug line for user query

internal/engine/engine.go

@@ -277,16 +277,21 @@ func (be *BotEngine) NetworkStatus() (*network.NetStatus, error) {
 }
 
 func (be *BotEngine) GetUser(appID entity.PlatformID, platformUserID string) (*entity.User, error) {
-	if u, _ := be.db.GetUserByApp(appID, platformUserID); u != nil {
-		return u, nil
+	existingUser, _ := be.db.GetUserByPlatformID(appID, platformUserID)
+	if existingUser != nil {
+		return existingUser, nil
 	}
 
-	user := &entity.User{PlatformID: appID, PlatformUserID: platformUserID}
-	if err := be.db.AddUser(user); err != nil {
+	newUser := &entity.User{
+		PlatformID:     appID,
+		PlatformUserID: platformUserID,
+		Role:           entity.BasicUser,
+	}
+	if err := be.db.AddUser(newUser); err != nil {
 		return nil, err
 	}
 
-	return user, nil
+	return newUser, nil
 }
 
 func (be *BotEngine) Stop() {

internal/repository/user.go

@@ -3,7 +3,7 @@ package repository
 import "github.com/pagu-project/pagu/internal/entity"
 
 func (db *Database) AddUser(u *entity.User) error {
-	tx := db.gormDB.Create(u)
+	tx := db.gormDB.Debug().Create(u)
 	if tx.Error != nil {
 		return WriteError{
 			Message: tx.Error.Error(),
@@ -13,7 +13,7 @@ func (db *Database) AddUser(u *entity.User) error {
 	return nil
 }
 
-func (db *Database) HasUser(id string) bool {
+func (db *Database) HasUser(id uint) bool {
 	var exists bool
 
 	_ = db.gormDB.Model(&entity.User{}).
@@ -25,9 +25,9 @@ func (db *Database) HasUser(id string) bool {
 	return exists
 }
 
-func (db *Database) GetUserByApp(appID entity.PlatformID, callerID string) (*entity.User, error) {
+func (db *Database) GetUserByPlatformID(appID entity.PlatformID, callerID string) (*entity.User, error) {
 	var user *entity.User
-	tx := db.gormDB.Model(&entity.User{}).
+	tx := db.gormDB.Debug().Model(&entity.User{}).
 		Where("platform_id = ?", appID).
 		Where("platform_user_id = ?", callerID).
 		First(&user)

internal/repository/user_test.go

@@ -0,0 +1,32 @@
+package repository
+
+import (
+	"testing"
+
+	"github.com/pagu-project/pagu/internal/entity"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestGetUserByID(t *testing.T) {
+	db, err := NewDB("sqlite:file::memory:")
+	require.NoError(t, err)
+
+	user := entity.User{
+		PlatformID:     entity.PlatformIDDiscord,
+		PlatformUserID: "1234",
+		Role:           entity.BasicUser,
+	}
+	err = db.AddUser(&user)
+	assert.NoError(t, err)
+
+	foundUser, err := db.GetUserByPlatformID(user.PlatformID, user.PlatformUserID)
+	require.NoError(t, err)
+
+	assert.Equal(t, foundUser.PlatformID, user.PlatformID)
+	assert.Equal(t, foundUser.PlatformUserID, user.PlatformUserID)
+	assert.Equal(t, foundUser.Role, user.Role)
+
+	hasUser := db.HasUser(foundUser.DBModel.ID)
+	assert.True(t, hasUser)
+}