diff --git a/src/segment_crystal_test.go b/src/segment_crystal_test.go new file mode 100644 index 000000000000..90a783bc1b70 --- /dev/null +++ b/src/segment_crystal_test.go @@ -0,0 +1,31 @@ +package main + +import ( + "fmt" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestCrystal(t *testing.T) { + cases := []struct { + Case string + ExpectedString string + Version string + }{ + {Case: "Crystal 1.0.0", ExpectedString: "1.0.0", Version: "Crystal 1.0.0 (2021-03-22)"}, + } + for _, tc := range cases { + params := &mockedLanguageParams{ + cmd: "crystal", + versionParam: "--version", + versionOutput: tc.Version, + extension: "*.cr", + } + env, props := getMockedLanguageEnv(params) + c := &crystal{} + c.init(props, env) + assert.True(t, c.enabled(), fmt.Sprintf("Failed in case: %s", tc.Case)) + assert.Equal(t, tc.ExpectedString, c.string(), fmt.Sprintf("Failed in case: %s", tc.Case)) + } +} diff --git a/src/segment_golang_test.go b/src/segment_golang_test.go index b7d8e18b13a6..4ac29f9098ff 100644 --- a/src/segment_golang_test.go +++ b/src/segment_golang_test.go @@ -7,6 +7,28 @@ import ( "github.com/stretchr/testify/assert" ) +type mockedLanguageParams struct { + cmd string + versionParam string + versionOutput string + extension string +} + +func getMockedLanguageEnv(params *mockedLanguageParams) (*MockedEnvironment, *properties) { + env := new(MockedEnvironment) + env.On("hasCommand", params.cmd).Return(true) + env.On("runCommand", params.cmd, []string{params.versionParam}).Return(params.versionOutput, nil) + env.On("hasFiles", params.extension).Return(true) + env.On("getcwd", nil).Return("/usr/home/project") + env.On("homeDir", nil).Return("/usr/home") + props := &properties{ + values: map[Property]interface{}{ + DisplayVersion: true, + }, + } + return env, props +} + func TestGolang(t *testing.T) { cases := []struct { Case string @@ -17,17 +39,13 @@ func TestGolang(t *testing.T) { {Case: "Go 1.16", ExpectedString: "1.16", Version: "go version go1.16 darwin/amd64"}, } for _, tc := range cases { - env := new(MockedEnvironment) - env.On("hasCommand", "go").Return(true) - env.On("runCommand", "go", []string{"version"}).Return(tc.Version, nil) - env.On("hasFiles", "*.go").Return(true) - env.On("getcwd", nil).Return("/usr/home/project") - env.On("homeDir", nil).Return("/usr/home") - props := &properties{ - values: map[Property]interface{}{ - DisplayVersion: true, - }, + params := &mockedLanguageParams{ + cmd: "go", + versionParam: "version", + versionOutput: tc.Version, + extension: "*.go", } + env, props := getMockedLanguageEnv(params) g := &golang{} g.init(props, env) assert.True(t, g.enabled(), fmt.Sprintf("Failed in case: %s", tc.Case)) diff --git a/src/segment_julia_test.go b/src/segment_julia_test.go new file mode 100644 index 000000000000..212aa16b1c0b --- /dev/null +++ b/src/segment_julia_test.go @@ -0,0 +1,32 @@ +package main + +import ( + "fmt" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestJulia(t *testing.T) { + cases := []struct { + Case string + ExpectedString string + Version string + }{ + {Case: "Julia 1.6.0", ExpectedString: "1.6.0", Version: "julia version 1.6.0"}, + {Case: "Julia 1.6.1", ExpectedString: "1.6.1", Version: "julia version 1.6.1"}, + } + for _, tc := range cases { + params := &mockedLanguageParams{ + cmd: "julia", + versionParam: "--version", + versionOutput: tc.Version, + extension: "*.jl", + } + env, props := getMockedLanguageEnv(params) + j := &julia{} + j.init(props, env) + assert.True(t, j.enabled(), fmt.Sprintf("Failed in case: %s", tc.Case)) + assert.Equal(t, tc.ExpectedString, j.string(), fmt.Sprintf("Failed in case: %s", tc.Case)) + } +}