67 lines
2.8 KiB
Go
67 lines
2.8 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/Tencent/WeKnora/internal/resumable"
|
||
|
|
"github.com/Tencent/WeKnora/internal/types"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"gorm.io/driver/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestResolveDigitalHumanUploadSourcePrefersMaterialThenLegacyThenReparse(t *testing.T) {
|
||
|
|
db, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "mat.db")), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
|
||
|
|
require.NoError(t, err)
|
||
|
|
sqlDB, err := db.DB()
|
||
|
|
require.NoError(t, err)
|
||
|
|
t.Cleanup(func() { _ = sqlDB.Close() })
|
||
|
|
require.NoError(t, db.AutoMigrate(&resumable.Upload{}))
|
||
|
|
root := t.TempDir()
|
||
|
|
uploads, err := resumable.NewService(db, root, func(context.Context, string, string) error { return nil },
|
||
|
|
func(ctx context.Context, _ *resumable.Upload) (context.Context, error) { return ctx, nil },
|
||
|
|
func(context.Context, *resumable.Upload, string) (int64, types.JSON, bool, error) {
|
||
|
|
return 0, nil, true, nil
|
||
|
|
})
|
||
|
|
require.NoError(t, err)
|
||
|
|
humans := &DigitalHumanService{}
|
||
|
|
|
||
|
|
row := &resumable.Upload{
|
||
|
|
ID: "u1", Name: "note.txt", Size: 11, Kind: "digital-human",
|
||
|
|
State: "completed", ExpiresAt: time.Now().Add(time.Hour),
|
||
|
|
Result: types.JSON(`{"selection":"full","source":{"name":"note.txt"}}`),
|
||
|
|
}
|
||
|
|
require.NoError(t, os.WriteFile(uploads.FilePath(row.ID), []byte("from-upload"), 0600))
|
||
|
|
|
||
|
|
// 1) material file wins
|
||
|
|
require.NoError(t, uploads.SaveMaterial(row.ID, types.DigitalHumanSource{Name: "note.txt", Text: "from-material"}))
|
||
|
|
source, err := resolveDigitalHumanUploadSource(context.Background(), humans, uploads, row, uploadResult{Selection: "full", Source: &types.DigitalHumanSource{Name: "note.txt"}})
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "from-material", source.Text)
|
||
|
|
|
||
|
|
// 2) legacy jsonb text when material missing
|
||
|
|
require.NoError(t, os.Remove(filepath.Join(root, row.ID+".material")))
|
||
|
|
legacy := uploadResult{Selection: "full", Source: &types.DigitalHumanSource{Name: "note.txt", Text: "from-jsonb"}}
|
||
|
|
source, err = resolveDigitalHumanUploadSource(context.Background(), humans, uploads, row, legacy)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "from-jsonb", source.Text)
|
||
|
|
|
||
|
|
// 3) reparse upload and backfill material
|
||
|
|
source, err = resolveDigitalHumanUploadSource(context.Background(), humans, uploads, row, uploadResult{Selection: "full", Source: &types.DigitalHumanSource{Name: "note.txt"}})
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "from-upload", source.Text)
|
||
|
|
loaded, err := uploads.LoadMaterial(row.ID)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "from-upload", loaded.Text)
|
||
|
|
|
||
|
|
meta, err := json.Marshal(uploadResult{Selection: "full", Source: &types.DigitalHumanSource{Name: "note.txt"}})
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotContains(t, string(meta), "from-upload")
|
||
|
|
}
|