mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-09 00:56:37 +08:00
fix(auth): add missing masked_token column to w_access_tokens table in migrations
This commit is contained in:
@@ -38,3 +38,47 @@ func TestAdminPluginUnit(t *testing.T) {
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, "0 4 * * *", setting.Default)
|
||||
}
|
||||
|
||||
func TestAdminMigrationsIncludeTaskExecutionsAndSchedules(t *testing.T) {
|
||||
ctx := core.NewContext(context.Background())
|
||||
p := admin.New()
|
||||
require.NoError(t, p.Apply(ctx))
|
||||
|
||||
entry, ok := ctx.Migrations().Get("admin")
|
||||
require.True(t, ok, "admin plugin must register migrations")
|
||||
assert.Equal(t, "admin", entry.PluginID)
|
||||
|
||||
// Verify sqlite migration files include w_schedules and w_task_executions
|
||||
sqliteDir, err := entry.FS.Open("migrations/sqlite/00001_initial.sql")
|
||||
require.NoError(t, err)
|
||||
defer sqliteDir.Close()
|
||||
|
||||
stat, err := sqliteDir.Stat()
|
||||
require.NoError(t, err)
|
||||
buf := make([]byte, stat.Size())
|
||||
_, err = sqliteDir.Read(buf)
|
||||
require.NoError(t, err)
|
||||
content := string(buf)
|
||||
|
||||
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_task_executions")
|
||||
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_schedules")
|
||||
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_system_configs")
|
||||
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_templates")
|
||||
|
||||
// Verify postgres migration files include w_schedules and w_task_executions
|
||||
pgDir, err := entry.FS.Open("migrations/postgres/00001_initial.sql")
|
||||
require.NoError(t, err)
|
||||
defer pgDir.Close()
|
||||
|
||||
stat, err = pgDir.Stat()
|
||||
require.NoError(t, err)
|
||||
buf = make([]byte, stat.Size())
|
||||
_, err = pgDir.Read(buf)
|
||||
require.NoError(t, err)
|
||||
pgContent := string(buf)
|
||||
|
||||
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_task_executions")
|
||||
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_schedules")
|
||||
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_system_configs")
|
||||
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_templates")
|
||||
}
|
||||
|
||||
@@ -35,6 +35,7 @@ CREATE TABLE IF NOT EXISTS w_access_tokens (
|
||||
user_id BIGINT NOT NULL,
|
||||
token_hash VARCHAR(64) NOT NULL UNIQUE,
|
||||
name VARCHAR(128) NOT NULL,
|
||||
masked_token VARCHAR(64) NOT NULL DEFAULT '',
|
||||
description VARCHAR(255),
|
||||
is_admin BOOLEAN DEFAULT FALSE,
|
||||
expires_at TIMESTAMPTZ,
|
||||
|
||||
@@ -35,6 +35,7 @@ CREATE TABLE IF NOT EXISTS w_access_tokens (
|
||||
user_id BIGINT NOT NULL,
|
||||
token_hash VARCHAR(64) NOT NULL UNIQUE,
|
||||
name VARCHAR(128) NOT NULL,
|
||||
masked_token VARCHAR(64) NOT NULL DEFAULT '',
|
||||
description VARCHAR(255),
|
||||
is_admin BOOLEAN DEFAULT 0,
|
||||
expires_at DATETIME,
|
||||
|
||||
@@ -151,6 +151,7 @@ func TestUserPlugin(t *testing.T) {
|
||||
require.NoError(t, db.New(db.WithDB(testDB)).Apply(ctx))
|
||||
require.NoError(t, cache.New().Apply(ctx))
|
||||
require.NoError(t, logger.New().Apply(ctx))
|
||||
require.NoError(t, auth.New().Apply(ctx))
|
||||
|
||||
p := user.New()
|
||||
assert.Equal(t, "user", p.Name())
|
||||
|
||||
Reference in New Issue
Block a user