root_test.go

package cmd

import (
	"os"
	"testing"

	"github.com/spf13/viper"
	"github.com/timescale/tiger-cli/internal/tiger/config"
)

func TestMain(m *testing.M) {
	// Clean up any global state before tests
	config.ResetGlobalConfig()
	code := m.Run()
	os.Exit(code)
}

func setupTestCommand(t *testing.T) (string, func()) {
	t.Helper()

	// Create temporary directory for test config
	tmpDir, err := os.MkdirTemp("", "tiger-test-cmd-*")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}

	// Clean up function
	cleanup := func() {
		os.RemoveAll(tmpDir)
		config.ResetGlobalConfig()
	}

	t.Cleanup(cleanup)

	return tmpDir, cleanup
}

func TestFlagPrecedence(t *testing.T) {
	tmpDir, _ := setupTestCommand(t)

	// Create config file with some values
	configContent := `api_url: https://file.api.com/v1
project_id: file-project
service_id: file-service
output: table
analytics: true
`
	configFile := config.GetConfigFile(tmpDir)
	if err := os.WriteFile(configFile, []byte(configContent), 0644); err != nil {
		t.Fatalf("Failed to write config file: %v", err)
	}

	// Set environment variables
	os.Setenv("TIGER_CONFIG_DIR", tmpDir)
	os.Setenv("TIGER_PROJECT_ID", "env-project")
	os.Setenv("TIGER_SERVICE_ID", "env-service")
	os.Setenv("TIGER_OUTPUT", "json")
	os.Setenv("TIGER_ANALYTICS", "false")

	defer func() {
		os.Unsetenv("TIGER_CONFIG_DIR")
		os.Unsetenv("TIGER_PROJECT_ID")
		os.Unsetenv("TIGER_SERVICE_ID")
		os.Unsetenv("TIGER_OUTPUT")
		os.Unsetenv("TIGER_ANALYTICS")
	}()

	// Use buildRootCmd() to get a complete root command
	testCmd := buildRootCmd()

	// Set CLI flags (these should take precedence)
	args := []string{
		"--config-dir", tmpDir,
		"--project-id", "flag-project",
		"--service-id", "flag-service",
		"--output", "yaml",
		"--analytics=false",
		"--debug",
		"version", // Need a subcommand to execute
	}

	testCmd.SetArgs(args)

	// Execute the command to trigger PersistentPreRunE
	err := testCmd.Execute()
	if err != nil {
		t.Fatalf("Command execution failed: %v", err)
	}

	// Verify Viper reflects the CLI flag values (highest precedence)
	if viper.GetString("project_id") != "flag-project" {
		t.Errorf("Expected Viper project_id 'flag-project', got '%s'", viper.GetString("project_id"))
	}
	if viper.GetString("output") != "yaml" {
		t.Errorf("Expected Viper output 'yaml', got '%s'", viper.GetString("output"))
	}
}

func TestFlagBindingWithViper(t *testing.T) {
	tmpDir, _ := setupTestCommand(t)

	// Set environment variable
	os.Setenv("TIGER_CONFIG_DIR", tmpDir)
	os.Setenv("TIGER_OUTPUT", "json")

	defer func() {
		os.Unsetenv("TIGER_CONFIG_DIR")
		os.Unsetenv("TIGER_OUTPUT")
	}()

	// Test 1: Environment variable should be used when no flag is set
	testCmd1 := buildRootCmd()
	testCmd1.SetArgs([]string{"version"}) // Need a subcommand
	err := testCmd1.Execute()
	if err != nil {
		t.Fatalf("Command execution failed: %v", err)
	}

	if viper.GetString("output") != "json" {
		t.Errorf("Expected output 'json' from env var, got '%s'", viper.GetString("output"))
	}

	// Reset for next test
	config.ResetGlobalConfig()

	// Test 2: Flag should override environment variable
	testCmd2 := buildRootCmd()
	testCmd2.SetArgs([]string{"--output", "table", "version"})
	err = testCmd2.Execute()
	if err != nil {
		t.Fatalf("Command execution failed: %v", err)
	}

	if viper.GetString("output") != "table" {
		t.Errorf("Expected output 'table' from flag, got '%s'", viper.GetString("output"))
	}
}

func TestConfigFilePrecedence(t *testing.T) {
	tmpDir, _ := setupTestCommand(t)

	// Create config file
	configContent := `output: json
analytics: false
`
	configFile := config.GetConfigFile(tmpDir)
	if err := os.WriteFile(configFile, []byte(configContent), 0644); err != nil {
		t.Fatalf("Failed to write config file: %v", err)
	}

	// Set environment that should be overridden by config file
	os.Setenv("TIGER_CONFIG_DIR", tmpDir)

	defer os.Unsetenv("TIGER_CONFIG_DIR")

	// Use buildRootCmd() to get a complete root command
	testCmd := buildRootCmd()

	// Execute with config file specified
	testCmd.SetArgs([]string{"--config-dir", tmpDir, "version"})
	err := testCmd.Execute()
	if err != nil {
		t.Fatalf("Command execution failed: %v", err)
	}

	// Values should come from config file since no flags were set
	if viper.GetString("output") != "json" {
		t.Errorf("Expected output 'json' from config file, got '%s'", viper.GetString("output"))
	}
	if viper.GetBool("analytics") != false {
		t.Errorf("Expected analytics false from config file, got %t", viper.GetBool("analytics"))
	}
}