mirror of
https://github.com/hanzoai/base.git
synced 2026-10-21 03:41:50 +00:00
zeekay
3fe483ac1b
Remove all _hz_ prefixes from collection IDs, internal markers, and test fixtures. The default users collection is now _users_auth_ instead of _hz_users_auth_. Internal SQL parser tokens renamed from __hz_join__/__hz_discard__ to __join__/__discard__. Backup prefix changed from @auto_hz_backup_ to @auto_backup_. Test database updated with correct collection IDs and indexes. JWT tokens in test files regenerated with new collection IDs. Add LogoURL field to MetaConfig so the admin UI logo is configurable via app settings instead of hardcoded. FullPage.svelte and App.svelte sidebar now read logo and app name from the settings store, falling back to defaults when not configured.
211 lines
4.9 KiB
Go
211 lines
4.9 KiB
Go
package base
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TestNew(t *testing.T) {
|
|
// copy os.Args
|
|
originalArgs := make([]string, len(os.Args))
|
|
copy(originalArgs, os.Args)
|
|
defer func() {
|
|
// restore os.Args
|
|
os.Args = originalArgs
|
|
}()
|
|
|
|
// change os.Args
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(
|
|
os.Args,
|
|
"--dir=test_dir",
|
|
"--encryptionEnv=test_encryption_env",
|
|
"--debug=true",
|
|
)
|
|
|
|
app := New()
|
|
|
|
if app == nil {
|
|
t.Fatal("Expected initialized Base instance, got nil")
|
|
}
|
|
|
|
if app.RootCmd == nil {
|
|
t.Fatal("Expected RootCmd to be initialized, got nil")
|
|
}
|
|
|
|
if app.App == nil {
|
|
t.Fatal("Expected App to be initialized, got nil")
|
|
}
|
|
|
|
if app.DataDir() != "test_dir" {
|
|
t.Fatalf("Expected app.DataDir() %q, got %q", "test_dir", app.DataDir())
|
|
}
|
|
|
|
if app.EncryptionEnv() != "test_encryption_env" {
|
|
t.Fatalf("Expected app.EncryptionEnv() test_encryption_env, got %q", app.EncryptionEnv())
|
|
}
|
|
}
|
|
|
|
func TestNewWithConfig(t *testing.T) {
|
|
app := NewWithConfig(Config{
|
|
DefaultDataDir: "test_dir",
|
|
DefaultEncryptionEnv: "test_encryption_env",
|
|
HideStartBanner: true,
|
|
})
|
|
|
|
if app == nil {
|
|
t.Fatal("Expected initialized Base instance, got nil")
|
|
}
|
|
|
|
if app.RootCmd == nil {
|
|
t.Fatal("Expected RootCmd to be initialized, got nil")
|
|
}
|
|
|
|
if app.App == nil {
|
|
t.Fatal("Expected App to be initialized, got nil")
|
|
}
|
|
|
|
if app.hideStartBanner != true {
|
|
t.Fatal("Expected app.hideStartBanner to be true, got false")
|
|
}
|
|
|
|
if app.DataDir() != "test_dir" {
|
|
t.Fatalf("Expected app.DataDir() %q, got %q", "test_dir", app.DataDir())
|
|
}
|
|
|
|
if app.EncryptionEnv() != "test_encryption_env" {
|
|
t.Fatalf("Expected app.EncryptionEnv() %q, got %q", "test_encryption_env", app.EncryptionEnv())
|
|
}
|
|
}
|
|
|
|
func TestNewWithConfigAndFlags(t *testing.T) {
|
|
// copy os.Args
|
|
originalArgs := make([]string, len(os.Args))
|
|
copy(originalArgs, os.Args)
|
|
defer func() {
|
|
// restore os.Args
|
|
os.Args = originalArgs
|
|
}()
|
|
|
|
// change os.Args
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(
|
|
os.Args,
|
|
"--dir=test_dir_flag",
|
|
"--encryptionEnv=test_encryption_env_flag",
|
|
"--debug=false",
|
|
)
|
|
|
|
app := NewWithConfig(Config{
|
|
DefaultDataDir: "test_dir",
|
|
DefaultEncryptionEnv: "test_encryption_env",
|
|
HideStartBanner: true,
|
|
})
|
|
|
|
if app == nil {
|
|
t.Fatal("Expected initialized Base instance, got nil")
|
|
}
|
|
|
|
if app.RootCmd == nil {
|
|
t.Fatal("Expected RootCmd to be initialized, got nil")
|
|
}
|
|
|
|
if app.App == nil {
|
|
t.Fatal("Expected App to be initialized, got nil")
|
|
}
|
|
|
|
if app.hideStartBanner != true {
|
|
t.Fatal("Expected app.hideStartBanner to be true, got false")
|
|
}
|
|
|
|
if app.DataDir() != "test_dir_flag" {
|
|
t.Fatalf("Expected app.DataDir() %q, got %q", "test_dir_flag", app.DataDir())
|
|
}
|
|
|
|
if app.EncryptionEnv() != "test_encryption_env_flag" {
|
|
t.Fatalf("Expected app.EncryptionEnv() %q, got %q", "test_encryption_env_flag", app.EncryptionEnv())
|
|
}
|
|
}
|
|
|
|
func TestSkipBootstrap(t *testing.T) {
|
|
// copy os.Args
|
|
originalArgs := make([]string, len(os.Args))
|
|
copy(originalArgs, os.Args)
|
|
defer func() {
|
|
// restore os.Args
|
|
os.Args = originalArgs
|
|
}()
|
|
|
|
tempDir := filepath.Join(os.TempDir(), "temp_data")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
// already bootstrapped
|
|
app0 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
app0.Bootstrap()
|
|
if v := app0.skipBootstrap(); !v {
|
|
t.Fatal("[bootstrapped] Expected true, got false")
|
|
}
|
|
|
|
// unknown command
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(os.Args, "demo")
|
|
app1 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
app1.RootCmd.AddCommand(&cobra.Command{Use: "test"})
|
|
if v := app1.skipBootstrap(); !v {
|
|
t.Fatal("[unknown] Expected true, got false")
|
|
}
|
|
|
|
// default flags
|
|
flagScenarios := []struct {
|
|
name string
|
|
short string
|
|
}{
|
|
{"help", "h"},
|
|
{"version", "v"},
|
|
}
|
|
|
|
for _, s := range flagScenarios {
|
|
// base flag
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(os.Args, "--"+s.name)
|
|
app1 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
if v := app1.skipBootstrap(); !v {
|
|
t.Fatalf("[--%s] Expected true, got false", s.name)
|
|
}
|
|
|
|
// short flag
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(os.Args, "-"+s.short)
|
|
app2 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
if v := app2.skipBootstrap(); !v {
|
|
t.Fatalf("[-%s] Expected true, got false", s.short)
|
|
}
|
|
|
|
customCmd := &cobra.Command{Use: "custom"}
|
|
customCmd.PersistentFlags().BoolP(s.name, s.short, false, "")
|
|
|
|
// base flag in custom command
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(os.Args, "custom")
|
|
os.Args = append(os.Args, "--"+s.name)
|
|
app3 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
app3.RootCmd.AddCommand(customCmd)
|
|
if v := app3.skipBootstrap(); v {
|
|
t.Fatalf("[--%s custom] Expected false, got true", s.name)
|
|
}
|
|
|
|
// short flag in custom command
|
|
os.Args = os.Args[:1]
|
|
os.Args = append(os.Args, "custom")
|
|
os.Args = append(os.Args, "-"+s.short)
|
|
app4 := NewWithConfig(Config{DefaultDataDir: tempDir})
|
|
app4.RootCmd.AddCommand(customCmd)
|
|
if v := app4.skipBootstrap(); v {
|
|
t.Fatalf("[-%s custom] Expected false, got true", s.short)
|
|
}
|
|
}
|
|
}
|