119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestModelStoreSQLiteCreatesSchemaAndPersists(t *testing.T) {
|
|
dbPath := filepath.Join(t.TempDir(), "models.sqlite")
|
|
store := NewModelStore(ModelStoreConfig{
|
|
Type: databaseTypeSQLite,
|
|
DSN: dbPath,
|
|
SQLitePath: dbPath,
|
|
})
|
|
defer store.Close()
|
|
|
|
if err := store.Load(); err != nil {
|
|
t.Fatalf("load sqlite store: %v", err)
|
|
}
|
|
if _, err := os.Stat(dbPath); err != nil {
|
|
t.Fatalf("sqlite file not created: %v", err)
|
|
}
|
|
|
|
model, err := store.EnsureByHostModelKey("chaturbate.com", "alice")
|
|
if err != nil {
|
|
t.Fatalf("ensure model: %v", err)
|
|
}
|
|
if model.ID != "chaturbate.com:alice" {
|
|
t.Fatalf("unexpected model id: %q", model.ID)
|
|
}
|
|
|
|
if err := store.SetLastSeenOnline("chaturbate.com", "alice", true, time.Now().UTC().Format(time.RFC3339Nano)); err != nil {
|
|
t.Fatalf("set last seen online: %v", err)
|
|
}
|
|
|
|
models := store.List()
|
|
if len(models) != 1 {
|
|
t.Fatalf("expected 1 model, got %d", len(models))
|
|
}
|
|
if models[0].LastSeenOnline == nil || !*models[0].LastSeenOnline {
|
|
t.Fatalf("lastSeenOnline not persisted: %#v", models[0].LastSeenOnline)
|
|
}
|
|
|
|
meta := store.Meta()
|
|
if meta.Count != 1 {
|
|
t.Fatalf("expected meta count 1, got %d", meta.Count)
|
|
}
|
|
if meta.UpdatedAt == "" {
|
|
t.Fatal("expected meta updatedAt")
|
|
}
|
|
}
|
|
|
|
func TestMigrateModelStoreDataSQLiteToSQLiteReplacesTarget(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
sourcePath := filepath.Join(tmp, "source.sqlite")
|
|
targetPath := filepath.Join(tmp, "target.sqlite")
|
|
|
|
sourceConfig := ModelStoreConfig{Type: databaseTypeSQLite, DSN: sourcePath, SQLitePath: sourcePath}
|
|
targetConfig := ModelStoreConfig{Type: databaseTypeSQLite, DSN: targetPath, SQLitePath: targetPath}
|
|
|
|
source := NewModelStore(sourceConfig)
|
|
defer source.Close()
|
|
if err := source.Load(); err != nil {
|
|
t.Fatalf("load source: %v", err)
|
|
}
|
|
if _, err := source.EnsureByHostModelKey("chaturbate.com", "alice"); err != nil {
|
|
t.Fatalf("ensure source model: %v", err)
|
|
}
|
|
if err := source.SetProfileImage("chaturbate.com", "alice", "https://example.test/a.jpg", "image/jpeg", []byte{1, 2, 3, 4}, time.Now().UTC().Format(time.RFC3339Nano)); err != nil {
|
|
t.Fatalf("set profile image: %v", err)
|
|
}
|
|
if err := source.SetBioContext("chaturbate.com", "alice", `{"ok":true}`, time.Now().UTC().Format(time.RFC3339Nano)); err != nil {
|
|
t.Fatalf("set bio context: %v", err)
|
|
}
|
|
|
|
target := NewModelStore(targetConfig)
|
|
if err := target.Load(); err != nil {
|
|
t.Fatalf("load target: %v", err)
|
|
}
|
|
if _, err := target.EnsureByHostModelKey("chaturbate.com", "stale"); err != nil {
|
|
t.Fatalf("ensure stale target model: %v", err)
|
|
}
|
|
_ = target.Close()
|
|
_ = source.Close()
|
|
|
|
count, err := migrateModelStoreData(sourceConfig, targetConfig)
|
|
if err != nil {
|
|
t.Fatalf("migrate sqlite to sqlite: %v", err)
|
|
}
|
|
if count != 1 {
|
|
t.Fatalf("expected 1 migrated row, got %d", count)
|
|
}
|
|
|
|
migrated := NewModelStore(targetConfig)
|
|
defer migrated.Close()
|
|
if err := migrated.Load(); err != nil {
|
|
t.Fatalf("load migrated target: %v", err)
|
|
}
|
|
models := migrated.List()
|
|
if len(models) != 1 {
|
|
t.Fatalf("expected target replacement with 1 row, got %d", len(models))
|
|
}
|
|
if models[0].ModelKey != "alice" {
|
|
t.Fatalf("unexpected migrated model: %q", models[0].ModelKey)
|
|
}
|
|
if models[0].ProfileImageCached == "" {
|
|
t.Fatal("expected migrated profile image blob marker")
|
|
}
|
|
bio, _, ok, err := migrated.GetBioContext("chaturbate.com", "alice")
|
|
if err != nil {
|
|
t.Fatalf("get migrated bio context: %v", err)
|
|
}
|
|
if !ok || bio != `{"ok":true}` {
|
|
t.Fatalf("unexpected migrated bio context ok=%v bio=%q", ok, bio)
|
|
}
|
|
}
|