[#409] added pocketbase.NewWithConfig factory

2022-09-10 22:53:17 +03:00
parent 4b64e0910b
commit 8c11e2ef01
2 changed files with 161 additions and 114 deletions
@@ -6,14 +6,19 @@ import (
 )

 func TestNew(t *testing.T) {
-	testDir := "./pb_test_data_dir"
-	defer os.RemoveAll(testDir)
+	// copy os.Args
+	originalArgs := []string{}
+	copy(originalArgs, os.Args)
+	defer func() {
+		// restore os.Args
+		copy(os.Args, originalArgs)
+	}()

-	// reset os.Args
+	// change os.Args
 	os.Args = os.Args[0:1]
 	os.Args = append(
 		os.Args,
-		"--dir="+testDir,
+		"--dir=test_dir",
 		"--encryptionEnv=test_encryption_env",
 		"--debug=true",
 	)
@@ -32,8 +37,8 @@ func TestNew(t *testing.T) {
 		t.Fatal("Expected appWrapper to be initialized, got nil")
 	}

-	if app.DataDir() != testDir {
-		t.Fatalf("Expected app.DataDir() %q, got %q", testDir, app.DataDir())
+	if app.DataDir() != "test_dir" {
+		t.Fatalf("Expected app.DataDir() %q, got %q", "test_dir", app.DataDir())
 	}

 	if app.EncryptionEnv() != "test_encryption_env" {
@@ -45,52 +50,93 @@ func TestNew(t *testing.T) {
 	}
 }

-func TestDefaultDebug(t *testing.T) {
-	app := New()
+func TestNewWithConfig(t *testing.T) {
+	app := NewWithConfig(Config{
+		DefaultDebug:         true,
+		DefaultDataDir:       "test_dir",
+		DefaultEncryptionEnv: "test_encryption_env",
+		HideStartBanner:      true,
+	})

-	app.DefaultDebug(true)
-	if app.defaultDebug != true {
-		t.Fatalf("Expected defaultDebug %v, got %v", true, app.defaultDebug)
+	if app == nil {
+		t.Fatal("Expected initialized PocketBase instance, got nil")
 	}

-	app.DefaultDebug(false)
-	if app.defaultDebug != false {
-		t.Fatalf("Expected defaultDebug %v, got %v", false, app.defaultDebug)
+	if app.RootCmd == nil {
+		t.Fatal("Expected RootCmd to be initialized, got nil")
+	}
+
+	if app.appWrapper == nil {
+		t.Fatal("Expected appWrapper 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.DataDir() %q, got %q", "test_encryption_env", app.EncryptionEnv())
+	}
+
+	if app.IsDebug() != true {
+		t.Fatal("Expected app.IsDebug() true, got false")
 	}
 }

-func TestDefaultDataDir(t *testing.T) {
-	app := New()
+func TestNewWithConfigAndFlags(t *testing.T) {
+	// copy os.Args
+	originalArgs := []string{}
+	copy(originalArgs, os.Args)
+	defer func() {
+		// restore os.Args
+		copy(os.Args, originalArgs)
+	}()

-	expected := "test_default"
+	// change os.Args
+	os.Args = os.Args[0:1]
+	os.Args = append(
+		os.Args,
+		"--dir=test_dir_flag",
+		"--encryptionEnv=test_encryption_env_flag",
+		"--debug=false",
+	)

-	app.DefaultDataDir(expected)
-	if app.defaultDataDir != expected {
-		t.Fatalf("Expected defaultDataDir %v, got %v", expected, app.defaultDataDir)
-	}
-}
-
-func TestDefaultEncryptionEnv(t *testing.T) {
-	app := New()
-
-	expected := "test_env"
-
-	app.DefaultEncryptionEnv(expected)
-	if app.defaultEncryptionEnv != expected {
-		t.Fatalf("Expected defaultEncryptionEnv %v, got %v", expected, app.defaultEncryptionEnv)
-	}
-}
-
-func TestShowStartBanner(t *testing.T) {
-	app := New()
-
-	app.ShowStartBanner(true)
-	if app.showStartBanner != true {
-		t.Fatalf("Expected showStartBanner %v, got %v", true, app.showStartBanner)
-	}
-
-	app.ShowStartBanner(false)
-	if app.showStartBanner != false {
-		t.Fatalf("Expected showStartBanner %v, got %v", false, app.showStartBanner)
+	app := NewWithConfig(Config{
+		DefaultDebug:         true,
+		DefaultDataDir:       "test_dir",
+		DefaultEncryptionEnv: "test_encryption_env",
+		HideStartBanner:      true,
+	})
+
+	if app == nil {
+		t.Fatal("Expected initialized PocketBase instance, got nil")
+	}
+
+	if app.RootCmd == nil {
+		t.Fatal("Expected RootCmd to be initialized, got nil")
+	}
+
+	if app.appWrapper == nil {
+		t.Fatal("Expected appWrapper 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.DataDir() %q, got %q", "test_encryption_env_flag", app.EncryptionEnv())
+	}
+
+	if app.IsDebug() != false {
+		t.Fatal("Expected app.IsDebug() false, got true")
 	}
 }