webhooker/internal/config/config_test.go

package config

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/fx"
	"go.uber.org/fx/fxtest"
	"sneak.berlin/go/webhooker/internal/globals"
	"sneak.berlin/go/webhooker/internal/logger"
)

func TestEnvironmentConfig(t *testing.T) {
	tests := []struct {
		name        string
		envValue    string
		envVars     map[string]string
		expectError bool
		isDev       bool
		isProd      bool
	}{
		{
			name:        "default is dev",
			envValue:    "",
			envVars:     map[string]string{},
			expectError: false,
			isDev:       true,
			isProd:      false,
		},
		{
			name:        "explicit dev",
			envValue:    "dev",
			envVars:     map[string]string{},
			expectError: false,
			isDev:       true,
			isProd:      false,
		},
		{
			name:        "explicit prod",
			envValue:    "prod",
			envVars:     map[string]string{},
			expectError: false,
			isDev:       false,
			isProd:      true,
		},
		{
			name:        "invalid environment",
			envValue:    "staging",
			envVars:     map[string]string{},
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Set environment variable if specified
			if tt.envValue != "" {
				os.Setenv("WEBHOOKER_ENVIRONMENT", tt.envValue)
				defer os.Unsetenv("WEBHOOKER_ENVIRONMENT")
			} else {
				os.Unsetenv("WEBHOOKER_ENVIRONMENT")
			}

			// Set additional environment variables
			for k, v := range tt.envVars {
				os.Setenv(k, v)
				defer os.Unsetenv(k)
			}

			if tt.expectError {
				// Use regular fx.New for error cases since fxtest doesn't expose errors the same way
				var cfg *Config
				app := fx.New(
					fx.NopLogger, // Suppress fx logs in tests
					fx.Provide(
						globals.New,
						logger.New,
						New,
					),
					fx.Populate(&cfg),
				)
				assert.Error(t, app.Err())
			} else {
				// Use fxtest for success cases
				var cfg *Config
				app := fxtest.New(
					t,
					fx.Provide(
						globals.New,
						logger.New,
						New,
					),
					fx.Populate(&cfg),
				)
				require.NoError(t, app.Err())
				app.RequireStart()
				defer app.RequireStop()

				assert.Equal(t, tt.isDev, cfg.IsDev())
				assert.Equal(t, tt.isProd, cfg.IsProd())
			}
		})
	}
}

func TestDefaultDataDir(t *testing.T) {
	// Verify that when DATA_DIR is unset, the default is /var/lib/webhooker
	// regardless of the environment setting.
	for _, env := range []string{"", "dev", "prod"} {
		name := env
		if name == "" {
			name = "unset"
		}
		t.Run("env="+name, func(t *testing.T) {
			if env != "" {
				os.Setenv("WEBHOOKER_ENVIRONMENT", env)
				defer os.Unsetenv("WEBHOOKER_ENVIRONMENT")
			} else {
				os.Unsetenv("WEBHOOKER_ENVIRONMENT")
			}
			os.Unsetenv("DATA_DIR")

			var cfg *Config
			app := fxtest.New(
				t,
				fx.Provide(
					globals.New,
					logger.New,
					New,
				),
				fx.Populate(&cfg),
			)
			require.NoError(t, app.Err())
			app.RequireStart()
			defer app.RequireStop()

			assert.Equal(t, "/var/lib/webhooker", cfg.DataDir)
		})
	}
}