【Go言語】TDDで大規模リファクタリング！散らばった環境変数を一箇所にまとめる技術

// ファイルA: ユーザーサービス
userPoolID := os.Getenv("COGNITO_USER_POOL_ID")

// ファイルB: 認証サービス
poolId := os.Getenv("COGNITO_USER_POOL_ID") // あれ、変数名が違う…

// ファイルC: 共通ライブラリ
cognitoPool := os.Getenv("COGNITO_USER_POOL_ID")
if cognitoPool == "" {
    // デフォルト値を設定…でも、他の場所と違う値だったら？
}

// pkg/config/config_test.go
func TestLoadConfig_Success(t *testing.T) {
    // 準備: テスト用の環境変数を設定
    os.Setenv("COGNITO_USER_POOL_ID", "ap-northeast-1_TestPool123")
    os.Setenv("USERS_TABLE_NAME", "test-users-table")
    os.Setenv("DEBUG_LOGGING", "true")
    // テストが終わったら環境変数を元に戻す
    defer os.Clearenv()

    // 実行: 理想の関数 `LoadConfig` を呼び出す（まだ存在しない！）
    config, err := LoadConfig()

    // 検証:
    assert.NoError(t, err) // エラーなく読み込めるはず
    assert.NotNil(t, config) // configオブジェクトが作られるはず

    // 型安全なゲッターメソッドで値を取得できるはず
    assert.Equal(t, "ap-northeast-1_TestPool123", config.GetCognitoUserPoolID())
    assert.Equal(t, "test-users-table", config.GetUsersTableName())
    assert.True(t, config.IsDebugEnabled()) // bool型として扱える！
}

// pkg/config/config_test.go
func TestLoadConfig_MissingRequiredVariable(t *testing.T) {
    // USERS_TABLE_NAME を設定しないでおく
    os.Setenv("COGNITO_USER_POOL_ID", "ap-northeast-1_TestPool123")
    defer os.Clearenv()

    // 実行
    _, err := LoadConfig()

    // 検証:
    assert.Error(t, err) // 必ずエラーになるはず
    assert.Contains(t, err.Error(), "USERS_TABLE_NAME") // エラーメッセージに変数名が含まれるはず
}

// pkg/config/config.go
package config

// Config は、アプリケーションの全設定を保持する構造体
type Config struct {
    // 必須の設定
    CognitoUserPoolID string
    CognitoClientID   string
    UsersTableName    string

    // オプション（デフォルト値あり）
    AWSRegion      string
    AllowedOrigins string
    DebugLogging   bool
}

// pkg/config/config.go
import (
    "fmt"
    "os"
    "strings"
)

// LoadConfig は環境変数から設定を読み込み、検証する
func LoadConfig() (*Config, error) {
    config := &Config{}

    // 必須の変数を読み込む
    config.CognitoUserPoolID = os.Getenv("COGNITO_USER_POOL_ID")
    if config.CognitoUserPoolID == "" {
        return nil, fmt.Errorf("必須の環境変数が見つかりません: COGNITO_USER_POOL_ID")
    }
    config.UsersTableName = os.Getenv("USERS_TABLE_NAME")
    if config.UsersTableName == "" {
        return nil, fmt.Errorf("必須の環境変数が見つかりません: USERS_TABLE_NAME")
    }
    // ...他の必須変数も同様に...

    // オプションの変数を読み込む（なければデフォルト値）
    config.AWSRegion = os.Getenv("AWS_REGION")
    if config.AWSRegion == "" {
        config.AWSRegion = "ap-northeast-1"
    }
    debugStr := strings.ToLower(os.Getenv("DEBUG_LOGGING"))
    config.DebugLogging = (debugStr == "true" || debugStr == "1")

    return config, nil
}

// pkg/config/config.go

// GetCognitoUserPoolID は、CognitoのユーザープールIDを返す
func (c *Config) GetCognitoUserPoolID() string {
    return c.CognitoUserPoolID
}

// IsDebugEnabled は、デバッグログが有効かどうかを返す
func (c *Config) IsDebugEnabled() bool {
    return c.DebugLogging
}
// ...他のゲッターも同様に...

$ go test -v ./pkg/config
PASS
ok      poc-cognite/pkg/config  0.003s

// cmd/update-user/main.go (修正前)
func businessHandler(...) (..., error) {
    // ...
    tableName := os.Getenv("USERS_TABLE_NAME") // 😭 文字列、検証なし
    if tableName == "" {
        // ...
    }
    // ...
}

// cmd/update-user/main.go (修正後)
import "poc-cognite/pkg/config" // 新しいパッケージをインポート

func businessHandler(...) (..., error) {
    // ...
    cfg, err := config.LoadConfig() // ✨ 型安全な設定読み込み
    if err != nil {
        // 起動時に設定ミスがわかる！
        return ..., fmt.Errorf("設定の読み込みに失敗: %w", err)
    }
    tableName := cfg.GetUsersTableName() // ✨ 型安全なゲッター
    // ...
}

$ go test ./...
PASS ✅