feat: seed default dataset sources

backend/internal/initdb/dataset_test.go

@@ -0,0 +1,84 @@
+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
+}