85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package initdb
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/gorm"
|
|
"senlinai-agent/backend/internal/models"
|
|
)
|
|
|
|
func TestInitDatasetCreatesDefaultSourcesWhenTableIsEmpty(t *testing.T) {
|
|
database := newDatasetInitTestDatabase(t)
|
|
require.NoError(t, InitUser(database))
|
|
|
|
require.NoError(t, InitDataset(database))
|
|
require.NoError(t, InitDataset(database))
|
|
|
|
var root models.SaUser
|
|
require.NoError(t, database.Where("email = ?", rootUsername).First(&root).Error)
|
|
var sources []models.SaDatasetSource
|
|
require.NoError(t, database.Order("id asc").Find(&sources).Error)
|
|
require.Len(t, sources, len(defaultDatasetSources))
|
|
for index, expected := range defaultDatasetSources {
|
|
require.Equal(t, expected.Name, sources[index].Name)
|
|
require.Equal(t, expected.URL, sources[index].URL)
|
|
require.Equal(t, expected.IconURL, sources[index].IconURL)
|
|
require.Equal(t, "rss", sources[index].Kind)
|
|
require.True(t, sources[index].Enabled)
|
|
require.Equal(t, root.ID, sources[index].CreatedBy)
|
|
require.Equal(t, root.Identity, sources[index].CreatedByIdentity)
|
|
}
|
|
}
|
|
|
|
func TestInitDatasetDoesNothingWhenSourceTableIsNotEmpty(t *testing.T) {
|
|
database := newDatasetInitTestDatabase(t)
|
|
require.NoError(t, InitUser(database))
|
|
var root models.SaUser
|
|
require.NoError(t, database.Where("email = ?", rootUsername).First(&root).Error)
|
|
existing := models.SaDatasetSource{
|
|
CreatedBy: root.ID,
|
|
Name: "Existing source",
|
|
Kind: "manual",
|
|
Enabled: true,
|
|
}
|
|
require.NoError(t, database.Create(&existing).Error)
|
|
|
|
require.NoError(t, InitDataset(database))
|
|
|
|
var sources []models.SaDatasetSource
|
|
require.NoError(t, database.Find(&sources).Error)
|
|
require.Len(t, sources, 1)
|
|
require.Equal(t, existing.Identity, sources[0].Identity)
|
|
}
|
|
|
|
func TestInitDatasetUsesFirstExistingUserWhenRootIsAbsent(t *testing.T) {
|
|
database := newDatasetInitTestDatabase(t)
|
|
user := models.SaUser{
|
|
Email: "existing@example.com",
|
|
DisplayName: "Existing",
|
|
PasswordHash: "hash",
|
|
Role: "user",
|
|
}
|
|
require.NoError(t, database.Create(&user).Error)
|
|
|
|
require.NoError(t, InitDataset(database))
|
|
|
|
var sources []models.SaDatasetSource
|
|
require.NoError(t, database.Find(&sources).Error)
|
|
require.Len(t, sources, len(defaultDatasetSources))
|
|
for _, source := range sources {
|
|
require.Equal(t, user.ID, source.CreatedBy)
|
|
require.Equal(t, user.Identity, source.CreatedByIdentity)
|
|
}
|
|
}
|
|
|
|
func newDatasetInitTestDatabase(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, database.AutoMigrate(&models.SaUser{}, &models.SaDatasetSource{}))
|
|
return database
|
|
}
|