mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2025-01-18 22:51:10 -08:00
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package segments
|
|
|
|
import (
|
|
"oh-my-posh/environment"
|
|
"oh-my-posh/mock"
|
|
"oh-my-posh/properties"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestOSInfo(t *testing.T) {
|
|
cases := []struct {
|
|
Case string
|
|
ExpectedString string
|
|
GOOS string
|
|
IsWSL bool
|
|
Platform string
|
|
DisplayDistroName bool
|
|
}{
|
|
{
|
|
Case: "WSL debian - icon",
|
|
ExpectedString: "WSL at \uf306",
|
|
GOOS: "linux",
|
|
IsWSL: true,
|
|
Platform: "debian",
|
|
},
|
|
{
|
|
Case: "WSL debian - name",
|
|
ExpectedString: "WSL at debian",
|
|
GOOS: "linux",
|
|
IsWSL: true,
|
|
Platform: "debian",
|
|
DisplayDistroName: true,
|
|
},
|
|
{
|
|
Case: "plain linux - icon",
|
|
ExpectedString: "\uf306",
|
|
GOOS: "linux",
|
|
Platform: "debian",
|
|
},
|
|
{
|
|
Case: "plain linux - name",
|
|
ExpectedString: "debian",
|
|
GOOS: "linux",
|
|
Platform: "debian",
|
|
DisplayDistroName: true,
|
|
},
|
|
{
|
|
Case: "windows",
|
|
ExpectedString: "windows",
|
|
GOOS: "windows",
|
|
},
|
|
{
|
|
Case: "darwin",
|
|
ExpectedString: "darwin",
|
|
GOOS: "darwin",
|
|
},
|
|
{
|
|
Case: "unknown",
|
|
ExpectedString: "unknown",
|
|
GOOS: "unknown",
|
|
},
|
|
}
|
|
for _, tc := range cases {
|
|
env := new(mock.MockedEnvironment)
|
|
env.On("GOOS").Return(tc.GOOS)
|
|
env.On("Platform").Return(tc.Platform)
|
|
env.On("TemplateCache").Return(&environment.TemplateCache{
|
|
Env: make(map[string]string),
|
|
WSL: tc.IsWSL,
|
|
})
|
|
osInfo := &Os{
|
|
env: env,
|
|
props: properties.Map{
|
|
DisplayDistroName: tc.DisplayDistroName,
|
|
Windows: "windows",
|
|
MacOS: "darwin",
|
|
},
|
|
}
|
|
_ = osInfo.Enabled()
|
|
assert.Equal(t, tc.ExpectedString, renderTemplate(env, osInfo.Template(), osInfo), tc.Case)
|
|
}
|
|
}
|