mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2025-01-07 09:17:26 -08:00
95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
package segments
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"oh-my-posh/environment"
|
|
"oh-my-posh/mock"
|
|
"oh-my-posh/properties"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type mockedLanguageParams struct {
|
|
cmd string
|
|
versionParam string
|
|
versionOutput string
|
|
extension string
|
|
}
|
|
|
|
func getMockedLanguageEnv(params *mockedLanguageParams) (*mock.MockedEnvironment, properties.Map) {
|
|
env := new(mock.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("Pwd").Return("/usr/home/project")
|
|
env.On("Home").Return("/usr/home")
|
|
env.On("TemplateCache").Return(&environment.TemplateCache{
|
|
Env: make(map[string]string),
|
|
})
|
|
props := properties.Map{
|
|
properties.FetchVersion: true,
|
|
}
|
|
return env, props
|
|
}
|
|
|
|
func TestGolang(t *testing.T) {
|
|
cases := []struct {
|
|
Case string
|
|
ExpectedString string
|
|
Version string
|
|
ParseModFile bool
|
|
HasModFileInParentDir bool
|
|
InvalidModfile bool
|
|
}{
|
|
{Case: "Go 1.15", ExpectedString: "1.15.8", Version: "go version go1.15.8 darwin/amd64"},
|
|
{Case: "Go 1.16", ExpectedString: "1.16", Version: "go version go1.16 darwin/amd64"},
|
|
{Case: "go.mod 1.18", ParseModFile: true, HasModFileInParentDir: true, ExpectedString: "1.18"},
|
|
{Case: "no go.mod file fallback", ParseModFile: true, ExpectedString: "1.16", Version: "go version go1.16 darwin/amd64"},
|
|
{
|
|
Case: "invalid go.mod file fallback",
|
|
ParseModFile: true,
|
|
HasModFileInParentDir: true,
|
|
InvalidModfile: true,
|
|
ExpectedString: "../go.mod:1: unknown directive: invalid",
|
|
Version: "go version go1.16 darwin/amd64",
|
|
},
|
|
}
|
|
for _, tc := range cases {
|
|
params := &mockedLanguageParams{
|
|
cmd: "go",
|
|
versionParam: "version",
|
|
versionOutput: tc.Version,
|
|
extension: "*.go",
|
|
}
|
|
env, props := getMockedLanguageEnv(params)
|
|
if tc.ParseModFile {
|
|
props[ParseModFile] = tc.ParseModFile
|
|
fileInfo := &environment.FileInfo{
|
|
Path: "../go.mod",
|
|
ParentFolder: "./",
|
|
IsDir: false,
|
|
}
|
|
var err error
|
|
if !tc.HasModFileInParentDir {
|
|
err = errors.New("no match")
|
|
}
|
|
env.On("HasParentFilePath", "go.mod").Return(fileInfo, err)
|
|
var content string
|
|
if tc.InvalidModfile {
|
|
content = "invalid go.mod file"
|
|
} else {
|
|
tmp, _ := ioutil.ReadFile(fileInfo.Path)
|
|
content = string(tmp)
|
|
}
|
|
env.On("FileContent", fileInfo.Path).Return(content)
|
|
}
|
|
g := &Golang{}
|
|
g.Init(props, env)
|
|
assert.True(t, g.Enabled(), fmt.Sprintf("Failed in case: %s", tc.Case))
|
|
assert.Equal(t, tc.ExpectedString, renderTemplate(env, g.Template(), g), fmt.Sprintf("Failed in case: %s", tc.Case))
|
|
}
|
|
}
|