mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-05-07 00:37:36 +00:00
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMigrateToMultiTenant(t *testing.T) {
|
|
// Setup temp dir
|
|
dataDir := t.TempDir()
|
|
|
|
// Create dummy legacy files
|
|
legacyFiles := []string{"nodes.enc", "system.json", "alerts.json"}
|
|
for _, f := range legacyFiles {
|
|
err := os.WriteFile(filepath.Join(dataDir, f), []byte("dummy content"), 0644)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Run migration
|
|
err := MigrateToMultiTenant(dataDir)
|
|
require.NoError(t, err)
|
|
|
|
// Verify files moved to default org
|
|
defaultOrgDir := filepath.Join(dataDir, "orgs", "default")
|
|
for _, f := range legacyFiles {
|
|
// Check file exists in new location
|
|
content, err := os.ReadFile(filepath.Join(defaultOrgDir, f))
|
|
require.NoError(t, err, "File %s should exist in default org dir", f)
|
|
require.Equal(t, "dummy content", string(content))
|
|
|
|
// Check symlink exists in old location
|
|
info, err := os.Lstat(filepath.Join(dataDir, f))
|
|
require.NoError(t, err, "Symlink %s should exist in root data dir", f)
|
|
require.Equal(t, os.ModeSymlink, info.Mode()&os.ModeSymlink, "%s should be a symlink", f)
|
|
}
|
|
|
|
// Verify marker
|
|
_, err = os.Stat(filepath.Join(defaultOrgDir, ".migrated"))
|
|
require.NoError(t, err, "Migration marker should exist")
|
|
|
|
// Run again (should be idempotent)
|
|
err = MigrateToMultiTenant(dataDir)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestIsMigrationNeeded(t *testing.T) {
|
|
dataDir := t.TempDir()
|
|
|
|
// Empty dir - no migration needed
|
|
require.False(t, IsMigrationNeeded(dataDir))
|
|
|
|
// With files - migration needed
|
|
_ = os.WriteFile(filepath.Join(dataDir, "system.json"), []byte("{}"), 0644)
|
|
require.True(t, IsMigrationNeeded(dataDir))
|
|
|
|
// After migration - not needed
|
|
_ = MigrateToMultiTenant(dataDir)
|
|
require.False(t, IsMigrationNeeded(dataDir))
|
|
}
|
|
|
|
func TestMigrateToMultiTenant_SetsOwnerOnlyPermissions(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("permission mode assertions are not stable on windows")
|
|
}
|
|
|
|
dataDir := t.TempDir()
|
|
legacyPath := filepath.Join(dataDir, "system.json")
|
|
_ = os.WriteFile(legacyPath, []byte("{}"), 0o644)
|
|
|
|
require.NoError(t, MigrateToMultiTenant(dataDir))
|
|
|
|
defaultOrgDir := filepath.Join(dataDir, "orgs", "default")
|
|
orgInfo, err := os.Stat(defaultOrgDir)
|
|
require.NoError(t, err)
|
|
require.Equal(t, os.FileMode(0o700), orgInfo.Mode().Perm())
|
|
|
|
migratedInfo, err := os.Stat(filepath.Join(defaultOrgDir, "system.json"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, os.FileMode(0o600), migratedInfo.Mode().Perm())
|
|
|
|
markerInfo, err := os.Stat(filepath.Join(defaultOrgDir, ".migrated"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, os.FileMode(0o600), markerInfo.Mode().Perm())
|
|
}
|