Tests

internal/config/config_test.go

@@ -0,0 +1,117 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestDefaultConfig(t *testing.T) {
+	cfg := DefaultConfig()
+
+	if cfg.DBHost != "localhost" {
+		t.Errorf("Expected DBHost to be 'localhost', got %q", cfg.DBHost)
+	}
+
+	if cfg.DBPort != 5432 {
+		t.Errorf("Expected DBPort to be 5432, got %d", cfg.DBPort)
+	}
+
+	if cfg.DBSchema != "public" {
+		t.Errorf("Expected DBSchema to be 'public', got %q", cfg.DBSchema)
+	}
+
+	if cfg.DBUser != "postgres" {
+		t.Errorf("Expected DBUser to be 'postgres', got %q", cfg.DBUser)
+	}
+
+	if cfg.DBPassword != "postgres" {
+		t.Errorf("Expected DBPassword to be 'postgres', got %q", cfg.DBPassword)
+	}
+}
+
+func TestSaveAndLoad(t *testing.T) {
+	// Create a temporary directory and set HOME to it
+	tmpDir := t.TempDir()
+	oldHome := os.Getenv("HOME")
+	os.Setenv("HOME", tmpDir)
+	defer os.Setenv("HOME", oldHome)
+
+	configPath := filepath.Join(tmpDir, ".config", "entity-maker.toml")
+
+	// Create a test config
+	cfg := &Config{
+		DBHost:     "testhost",
+		DBPort:     5433,
+		DBName:     "testdb",
+		DBSchema:   "testschema",
+		DBUser:     "testuser",
+		DBPassword: "testpass",
+		DBTable:    "testtable",
+		OutputDir:  "/tmp/test",
+		EntityName: "TestEntity",
+	}
+
+	// Save config
+	err := cfg.Save()
+	if err != nil {
+		t.Fatalf("Failed to save config: %v", err)
+	}
+
+	// Verify file exists
+	if _, err := os.Stat(configPath); os.IsNotExist(err) {
+		t.Fatalf("Config file was not created: %v", err)
+	}
+
+	// Load config
+	loadedCfg, err := Load()
+	if err != nil {
+		t.Fatalf("Failed to load config: %v", err)
+	}
+
+	// Verify loaded values
+	tests := []struct {
+		name     string
+		got      interface{}
+		expected interface{}
+	}{
+		{"DBHost", loadedCfg.DBHost, cfg.DBHost},
+		{"DBPort", loadedCfg.DBPort, cfg.DBPort},
+		{"DBName", loadedCfg.DBName, cfg.DBName},
+		{"DBSchema", loadedCfg.DBSchema, cfg.DBSchema},
+		{"DBUser", loadedCfg.DBUser, cfg.DBUser},
+		{"DBPassword", loadedCfg.DBPassword, cfg.DBPassword},
+		{"DBTable", loadedCfg.DBTable, cfg.DBTable},
+		{"OutputDir", loadedCfg.OutputDir, cfg.OutputDir},
+		{"EntityName", loadedCfg.EntityName, cfg.EntityName},
+	}
+
+	for _, tt := range tests {
+		if tt.got != tt.expected {
+			t.Errorf("%s mismatch: expected %v, got %v", tt.name, tt.expected, tt.got)
+		}
+	}
+}
+
+func TestLoadNonExistentConfig(t *testing.T) {
+	// Create a temporary directory and set HOME to it
+	tmpDir := t.TempDir()
+	oldHome := os.Getenv("HOME")
+	os.Setenv("HOME", tmpDir)
+	defer os.Setenv("HOME", oldHome)
+
+	// Load should return default config when file doesn't exist
+	cfg, err := Load()
+	if err != nil {
+		t.Fatalf("Load should not error when file doesn't exist: %v", err)
+	}
+
+	// Verify it's the default config
+	defaultCfg := DefaultConfig()
+	if cfg.DBHost != defaultCfg.DBHost {
+		t.Errorf("Expected default DBHost %q, got %q", defaultCfg.DBHost, cfg.DBHost)
+	}
+	if cfg.DBPort != defaultCfg.DBPort {
+		t.Errorf("Expected default DBPort %d, got %d", defaultCfg.DBPort, cfg.DBPort)
+	}
+}