mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2025-01-05 08:17:25 -08:00
92 lines
2 KiB
Go
92 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"oh-my-posh/environment"
|
|
"oh-my-posh/mock"
|
|
"oh-my-posh/properties"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestExitWriterEnabled(t *testing.T) {
|
|
cases := []struct {
|
|
ExitCode int
|
|
Expected bool
|
|
}{
|
|
{ExitCode: 102, Expected: true},
|
|
{ExitCode: 0, Expected: false},
|
|
{ExitCode: -1, Expected: true},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
env := new(mock.MockedEnvironment)
|
|
env.On("ErrorCode").Return(tc.ExitCode)
|
|
e := &Exit{
|
|
env: env,
|
|
props: properties.Map{},
|
|
}
|
|
assert.Equal(t, tc.Expected, e.Enabled())
|
|
}
|
|
}
|
|
|
|
func TestGetMeaningFromExitCode(t *testing.T) {
|
|
errorMap := make(map[int]string)
|
|
errorMap[1] = "ERROR"
|
|
errorMap[2] = "USAGE"
|
|
errorMap[126] = "NOPERM"
|
|
errorMap[127] = "NOTFOUND"
|
|
errorMap[129] = "SIGHUP"
|
|
errorMap[130] = "SIGINT"
|
|
errorMap[131] = "SIGQUIT"
|
|
errorMap[132] = "SIGILL"
|
|
errorMap[133] = "SIGTRAP"
|
|
errorMap[134] = "SIGIOT"
|
|
errorMap[135] = "SIGBUS"
|
|
errorMap[136] = "SIGFPE"
|
|
errorMap[137] = "SIGKILL"
|
|
errorMap[138] = "SIGUSR1"
|
|
errorMap[139] = "SIGSEGV"
|
|
errorMap[140] = "SIGUSR2"
|
|
errorMap[141] = "SIGPIPE"
|
|
errorMap[142] = "SIGALRM"
|
|
errorMap[143] = "SIGTERM"
|
|
errorMap[144] = "SIGSTKFLT"
|
|
errorMap[145] = "SIGCHLD"
|
|
errorMap[146] = "SIGCONT"
|
|
errorMap[147] = "SIGSTOP"
|
|
errorMap[148] = "SIGTSTP"
|
|
errorMap[149] = "SIGTTIN"
|
|
errorMap[150] = "SIGTTOU"
|
|
errorMap[151] = "151"
|
|
errorMap[7000] = "7000"
|
|
for exitcode, want := range errorMap {
|
|
e := &Exit{}
|
|
assert.Equal(t, want, e.getMeaningFromExitCode(exitcode))
|
|
}
|
|
}
|
|
|
|
func TestExitWriterTemplateString(t *testing.T) {
|
|
cases := []struct {
|
|
Case string
|
|
ExitCode int
|
|
Expected string
|
|
Template string
|
|
}{
|
|
{Case: "Only code", ExitCode: 129, Expected: "129", Template: "{{ .Code }}"},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
env := new(mock.MockedEnvironment)
|
|
env.On("ErrorCode").Return(tc.ExitCode)
|
|
env.On("TemplateCache").Return(&environment.TemplateCache{
|
|
Code: tc.ExitCode,
|
|
})
|
|
e := &Exit{
|
|
env: env,
|
|
props: properties.Map{},
|
|
}
|
|
assert.Equal(t, tc.Expected, renderTemplate(env, tc.Template, e), tc.Case)
|
|
}
|
|
}
|