mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-03 07:06:36 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package migrator
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/pkg/config"
|
|
"github.com/pressly/goose/v3"
|
|
)
|
|
|
|
func TestClickHouseMigrationFilesEmbedded(t *testing.T) {
|
|
entries, err := clickhouseMigrationFS.ReadDir(clickhouseMigrationDir)
|
|
if err != nil {
|
|
t.Fatalf("ReadDir(%q) error = %v", clickhouseMigrationDir, err)
|
|
}
|
|
if len(entries) == 0 {
|
|
t.Fatal("expected embedded ClickHouse migrations, got none")
|
|
}
|
|
|
|
found := false
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
continue
|
|
}
|
|
if entry.Name() == "202606190001_create_user_access_logs.sql" {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatal("expected 202606190001_create_user_access_logs.sql in embedded migrations")
|
|
}
|
|
}
|
|
|
|
func TestClickHouseGooseDialect(t *testing.T) {
|
|
if err := goose.SetDialect("clickhouse"); err != nil {
|
|
t.Fatalf("SetDialect(clickhouse) error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMigrateClickHouseSkipsWhenDisabled(t *testing.T) {
|
|
previousEnabled := config.Config.ClickHouse.Enabled
|
|
config.Config.ClickHouse.Enabled = false
|
|
t.Cleanup(func() {
|
|
config.Config.ClickHouse.Enabled = previousEnabled
|
|
})
|
|
|
|
MigrateClickHouse()
|
|
}
|