mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-09-19 21:17:31 -07:00
103 lines
2.5 KiB
Go
Executable file
103 lines
2.5 KiB
Go
Executable file
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type venvArgs struct {
|
|
virtualEnvName string
|
|
condaEnvName string
|
|
condaDefaultName string
|
|
pathSeparator string
|
|
}
|
|
|
|
func newVenvArgs() *venvArgs {
|
|
return &venvArgs{
|
|
virtualEnvName: "",
|
|
condaEnvName: "",
|
|
condaDefaultName: "",
|
|
pathSeparator: "/",
|
|
}
|
|
}
|
|
|
|
func bootStrapVenvTest(args *venvArgs) *venv {
|
|
env := new(MockedEnvironment)
|
|
env.On("getenv", "VIRTUAL_ENV").Return(args.virtualEnvName)
|
|
env.On("getenv", "CONDA_ENV_PATH").Return(args.condaEnvName)
|
|
env.On("getenv", "CONDA_DEFAULT_ENV").Return(args.condaDefaultName)
|
|
env.On("getPathSeperator", nil).Return(args.pathSeparator)
|
|
venv := &venv{
|
|
env: env,
|
|
}
|
|
return venv
|
|
}
|
|
|
|
func TestVenvWriterDisabled(t *testing.T) {
|
|
args := newVenvArgs()
|
|
venv := bootStrapVenvTest(args)
|
|
assert.False(t, venv.enabled(), "the virtualenv has no name")
|
|
}
|
|
|
|
func TestVenvWriterEnabledWithVirtualEnv(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.virtualEnvName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
assert.True(t, venv.enabled(), "the virtualenv has a name")
|
|
}
|
|
|
|
func TestVenvWriterEnabledWithCondaEnvPath(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.condaEnvName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
assert.True(t, venv.enabled(), "the virtualenv has a name")
|
|
}
|
|
|
|
func TestVenvWriterEnabledWithCondaDefaultEnv(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.condaDefaultName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
assert.True(t, venv.enabled(), "the virtualenv has a name")
|
|
}
|
|
|
|
func TestVenvWriterEnabledWithTwoValidEnvs(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.virtualEnvName = "venv"
|
|
args.condaDefaultName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
assert.True(t, venv.enabled(), "the virtualenv has a name")
|
|
}
|
|
|
|
func TestVenvWriterNameWithVirtualEnv(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.virtualEnvName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
_ = venv.enabled()
|
|
assert.Equal(t, "venv", venv.venvName)
|
|
}
|
|
|
|
func TestVenvWriterNameWithCondaEnvPath(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.condaEnvName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
_ = venv.enabled()
|
|
assert.Equal(t, "venv", venv.venvName)
|
|
}
|
|
|
|
func TestVenvWriterNameWithCondaDefaultEnv(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.condaDefaultName = "venv"
|
|
venv := bootStrapVenvTest(args)
|
|
_ = venv.enabled()
|
|
assert.Equal(t, "venv", venv.venvName)
|
|
}
|
|
|
|
func TestVenvWriterNameTrailingSlash(t *testing.T) {
|
|
args := newVenvArgs()
|
|
args.virtualEnvName = "venv/"
|
|
venv := bootStrapVenvTest(args)
|
|
_ = venv.enabled()
|
|
assert.Equal(t, "venv", venv.venvName)
|
|
}
|