mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-11-14 15:04:03 -08:00
33 lines
752 B
Go
33 lines
752 B
Go
|
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())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|