81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
cli "v/cli"
|
|
python "v/python"
|
|
state "v/state"
|
|
testutils "v/testutils"
|
|
)
|
|
|
|
func TestWriteShim(t *testing.T) {
|
|
defer testutils.SetupAndCleanupEnvironment(t)()
|
|
|
|
os.Mkdir(state.GetStatePath("shims"), 0775)
|
|
testShimPath := state.GetStatePath("shims", "testshim")
|
|
e := writeShim(testShimPath, "testcommand")
|
|
|
|
shimContent, _ := ioutil.ReadFile(testShimPath)
|
|
|
|
if e != nil {
|
|
t.Errorf("Errored while writing shim")
|
|
}
|
|
|
|
if !strings.Contains(string(shimContent), "testcommand") {
|
|
t.Errorf("Expected shim to contain pass-through via 'which', got %s", shimContent)
|
|
}
|
|
|
|
}
|
|
|
|
func TestInitializeCreatesStateDirectories(t *testing.T) {
|
|
defer testutils.SetupAndCleanupEnvironment(t)()
|
|
|
|
err := Initialize([]string{}, cli.Flags{}, state.State{})
|
|
|
|
if err != nil {
|
|
t.Errorf("Unexpected error initializing")
|
|
}
|
|
|
|
if _, err = os.Stat(state.GetStatePath()); os.IsNotExist(err) {
|
|
t.Errorf("Root state directory not found")
|
|
}
|
|
|
|
if _, err = os.Stat(state.GetStatePath("shims")); os.IsNotExist(err) {
|
|
t.Errorf("Shims directory not found")
|
|
}
|
|
|
|
if _, err = os.Stat(state.GetStatePath("cache")); os.IsNotExist(err) {
|
|
t.Errorf("Cache directory not found")
|
|
}
|
|
|
|
if _, err = os.Stat(state.GetStatePath("runtimes")); os.IsNotExist(err) {
|
|
t.Errorf("Runtimes directory not found")
|
|
}
|
|
}
|
|
|
|
func TestInitializeCreatesAllPythonShims(t *testing.T) {
|
|
defer testutils.SetupAndCleanupEnvironment(t)()
|
|
|
|
err := Initialize([]string{}, cli.Flags{}, state.State{})
|
|
|
|
if err != nil {
|
|
t.Errorf("Unexpected error initializing")
|
|
}
|
|
|
|
expectedShims := python.Shims
|
|
|
|
for shimLabel, shimCall := range expectedShims {
|
|
shimContent, err := os.ReadFile(state.GetStatePath("shims", shimLabel))
|
|
|
|
if os.IsNotExist(err) {
|
|
t.Errorf("%s shim not created", shimLabel)
|
|
}
|
|
|
|
if !strings.Contains(string(shimContent), shimCall) {
|
|
t.Errorf("%s shim does not contain expected call (%s not in %s)", shimLabel, shimCall, shimContent)
|
|
}
|
|
}
|
|
}
|