oh-my-posh/segment_exit_test.go
2020-10-05 17:19:05 +02:00

94 lines
2 KiB
Go

package main
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestExitWriterEnabled(t *testing.T) {
env := new(MockedEnvironment)
env.On("lastErrorCode", nil).Return(102)
e := &exit{
env: env,
}
assert.True(t, e.enabled())
}
func TestExitWriterDisabled(t *testing.T) {
env := new(MockedEnvironment)
env.On("lastErrorCode", nil).Return(0)
e := &exit{
env: env,
}
assert.False(t, e.enabled())
}
func TestExitWriterStandardCode(t *testing.T) {
env := new(MockedEnvironment)
env.On("lastErrorCode", nil).Return(129)
props := &properties{
foreground: "#111111",
background: "#ffffff",
}
e := &exit{
env: env,
props: props,
}
assert.Equal(t, "SIGHUP", e.getFormattedText())
}
func TestExitWriterNonStandardCode(t *testing.T) {
env := new(MockedEnvironment)
env.On("lastErrorCode", nil).Return(5001)
props := &properties{
foreground: "#111111",
background: "#ffffff",
}
e := &exit{
env: env,
props: props,
}
assert.Equal(t, "5001", e.getFormattedText())
}
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 {
env := new(MockedEnvironment)
env.On("lastErrorCode", nil).Return(exitcode)
e := &exit{
env: env,
}
assert.Equal(t, want, e.getMeaningFromExitCode())
}
}