// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package interact

import (
	"os"
	"testing"
)

func TestIsAgitPromptEnabled(t *testing.T) {
	t.Run("default enabled", func(t *testing.T) {
		previous, hadPrevious := os.LookupEnv("TEA_USE_AGIT")
		if err := os.Unsetenv("TEA_USE_AGIT"); err != nil {
			t.Fatal(err)
		}
		t.Cleanup(func() {
			if hadPrevious {
				if err := os.Setenv("TEA_USE_AGIT", previous); err != nil {
					t.Fatal(err)
				}
			}
		})

		if !isAgitPromptEnabled() {
			t.Fatal("isAgitPromptEnabled() = false, want true")
		}
	})

	tests := []struct {
		name  string
		value string
		want  bool
	}{
		{name: "zero disables", value: "0", want: false},
		{name: "false disables", value: "false", want: false},
		{name: "true enables", value: "true", want: true},
		{name: "invalid enables", value: "invalid", want: true},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			t.Setenv("TEA_USE_AGIT", test.value)

			if got := isAgitPromptEnabled(); got != test.want {
				t.Fatalf("isAgitPromptEnabled() = %v, want %v", got, test.want)
			}
		})
	}
}