92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package testutils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"veza-backend-api/internal/models"
|
|
)
|
|
|
|
func TestSetupTestDB(t *testing.T) {
|
|
db := SetupTestDB()
|
|
require.NotNil(t, db)
|
|
defer CleanupTestDB(db)
|
|
|
|
// Vérifier que les tables sont créées
|
|
assert.True(t, db.Migrator().HasTable(&models.User{}))
|
|
assert.True(t, db.Migrator().HasTable(&models.Track{}))
|
|
assert.True(t, db.Migrator().HasTable(&models.Playlist{}))
|
|
assert.True(t, db.Migrator().HasTable(&models.Room{}))
|
|
assert.True(t, db.Migrator().HasTable(&models.Message{}))
|
|
}
|
|
|
|
func TestCleanupTestDB(t *testing.T) {
|
|
db := SetupTestDB()
|
|
require.NotNil(t, db)
|
|
|
|
err := CleanupTestDB(db)
|
|
assert.NoError(t, err)
|
|
|
|
// Vérifier que la DB est fermée en essayant une requête
|
|
sqlDB, _ := db.DB()
|
|
assert.Error(t, sqlDB.Ping())
|
|
}
|
|
|
|
func TestResetTestDB(t *testing.T) {
|
|
db := SetupTestDB()
|
|
require.NotNil(t, db)
|
|
defer CleanupTestDB(db)
|
|
|
|
// Créer des données de test
|
|
user, err := CreateTestUser(db)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, user)
|
|
|
|
// Vérifier que les données existent
|
|
var count int64
|
|
db.Model(&models.User{}).Count(&count)
|
|
assert.Equal(t, int64(1), count)
|
|
|
|
// Réinitialiser la DB
|
|
err = ResetTestDB(db)
|
|
assert.NoError(t, err)
|
|
|
|
// Vérifier que les données ont été supprimées
|
|
db.Model(&models.User{}).Count(&count)
|
|
assert.Equal(t, int64(0), count)
|
|
}
|
|
|
|
func TestGetDBStats(t *testing.T) {
|
|
db := SetupTestDB()
|
|
require.NotNil(t, db)
|
|
defer CleanupTestDB(db)
|
|
|
|
stats, err := GetDBStats(db)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, stats)
|
|
|
|
// Vérifier que les stats sont valides
|
|
assert.GreaterOrEqual(t, stats.MaxOpenConnections, 0)
|
|
}
|
|
|
|
func TestSetupTestDB_CanCreateRecords(t *testing.T) {
|
|
db := SetupTestDB()
|
|
require.NotNil(t, db)
|
|
defer CleanupTestDB(db)
|
|
|
|
// Créer un utilisateur de test
|
|
user, err := CreateTestUser(db)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, user)
|
|
|
|
// Vérifier que l'utilisateur a un ID
|
|
assert.Greater(t, user.ID, int64(0))
|
|
|
|
// Vérifier que l'utilisateur peut être récupéré
|
|
var retrievedUser models.User
|
|
err = db.First(&retrievedUser, user.ID).Error
|
|
require.NoError(t, err)
|
|
assert.Equal(t, user.Username, retrievedUser.Username)
|
|
assert.Equal(t, user.Email, retrievedUser.Email)
|
|
}
|