package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestPoshGitSegment(t *testing.T) {
	cases := []struct {
		Case          string
		PoshGitPrompt string
		Expected      string
		Enabled       bool
	}{
		{Case: "regular prompt", PoshGitPrompt: "my prompt", Expected: "my prompt", Enabled: true},
		{Case: "prompt with spaces", PoshGitPrompt: "   my prompt", Expected: "my prompt", Enabled: true},
		{Case: "no prompt", PoshGitPrompt: "", Enabled: false},
	}

	for _, tc := range cases {
		env := new(MockedEnvironment)
		env.On("getenv", poshGitEnv).Return(tc.PoshGitPrompt)
		p := &poshgit{
			env: env,
		}
		assert.Equal(t, tc.Enabled, p.enabled())
		if tc.Enabled {
			assert.Equal(t, tc.Expected, p.string())
		}
	}
}