mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2025-01-05 08:17:25 -08:00
94 lines
3 KiB
Go
94 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"oh-my-posh/environment"
|
|
"oh-my-posh/mock"
|
|
"oh-my-posh/properties"
|
|
"testing"
|
|
|
|
"github.com/alecthomas/assert"
|
|
)
|
|
|
|
func TestPythonTemplate(t *testing.T) {
|
|
cases := []struct {
|
|
Case string
|
|
Expected string
|
|
ExpectedDisabled bool
|
|
Template string
|
|
VirtualEnvName string
|
|
FetchVersion bool
|
|
}{
|
|
{Case: "No virtual env present", FetchVersion: true, Expected: "3.8.4", Template: "{{ if .Venv }}{{ .Venv }} {{ end }}{{ .Full }}"},
|
|
{Case: "Virtual env present", FetchVersion: true, Expected: "VENV 3.8.4", VirtualEnvName: "VENV", Template: "{{ if .Venv }}{{ .Venv }} {{ end }}{{ .Full }}"},
|
|
{
|
|
Case: "Virtual env major and minor dot",
|
|
FetchVersion: true,
|
|
Expected: "VENV 3.8",
|
|
VirtualEnvName: "VENV",
|
|
Template: "{{ if .Venv }}{{ .Venv }} {{ end }}{{ .Major }}.{{ .Minor }}",
|
|
},
|
|
{
|
|
Case: "Virtual env hide on default",
|
|
FetchVersion: true,
|
|
Expected: "3.8",
|
|
VirtualEnvName: "default",
|
|
Template: "{{ if ne .Venv \"default\" }}{{ .Venv }} {{ end }}{{ .Major }}.{{ .Minor }}",
|
|
},
|
|
{
|
|
Case: "Virtual env show non default",
|
|
FetchVersion: true,
|
|
Expected: "billy 3.8",
|
|
VirtualEnvName: "billy",
|
|
Template: "{{ if ne .Venv \"default\" }}{{ .Venv }} {{ end }}{{ .Major }}.{{ .Minor }}",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
env := new(mock.MockedEnvironment)
|
|
env.On("HasCommand", "python").Return(true)
|
|
env.On("RunCommand", "python", []string{"--version"}).Return("Python 3.8.4", nil)
|
|
env.On("HasFiles", "*.py").Return(true)
|
|
env.On("Getenv", "VIRTUAL_ENV").Return(tc.VirtualEnvName)
|
|
env.On("Getenv", "CONDA_ENV_PATH").Return(tc.VirtualEnvName)
|
|
env.On("Getenv", "CONDA_DEFAULT_ENV").Return(tc.VirtualEnvName)
|
|
env.On("Getenv", "PYENV_VERSION").Return(tc.VirtualEnvName)
|
|
env.On("PathSeperator").Return("")
|
|
env.On("Pwd").Return("/usr/home/project")
|
|
env.On("Home").Return("/usr/home")
|
|
props := properties.Map{
|
|
properties.FetchVersion: tc.FetchVersion,
|
|
DisplayMode: DisplayModeAlways,
|
|
}
|
|
env.On("TemplateCache").Return(&environment.TemplateCache{
|
|
Env: make(map[string]string),
|
|
})
|
|
python := &Python{}
|
|
python.Init(props, env)
|
|
assert.Equal(t, !tc.ExpectedDisabled, python.Enabled(), tc.Case)
|
|
assert.Equal(t, tc.Expected, renderTemplate(env, tc.Template, python), tc.Case)
|
|
}
|
|
}
|
|
|
|
func TestPythonPythonInContext(t *testing.T) {
|
|
cases := []struct {
|
|
Expected bool
|
|
VirtualEnvName string
|
|
}{
|
|
{Expected: true, VirtualEnvName: "VENV"},
|
|
{Expected: false, VirtualEnvName: ""},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
env := new(mock.MockedEnvironment)
|
|
env.On("PathSeperator").Return("")
|
|
env.On("Getenv", "VIRTUAL_ENV").Return(tc.VirtualEnvName)
|
|
env.On("Getenv", "CONDA_ENV_PATH").Return("")
|
|
env.On("Getenv", "CONDA_DEFAULT_ENV").Return("")
|
|
env.On("Getenv", "PYENV_VERSION").Return("")
|
|
python := &Python{}
|
|
python.Init(properties.Map{}, env)
|
|
python.loadContext()
|
|
assert.Equal(t, tc.Expected, python.inContext())
|
|
}
|
|
}
|