mirror of https://github.com/databricks/cli.git
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
|
package python
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestExecAndPassError(t *testing.T) {
|
||
|
|
_, err := execAndPassErr(context.Background(), "which", "__non_existing__")
|
||
|
|
assert.EqualError(t, err, "exit status 1")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDetectPython(t *testing.T) {
|
||
|
|
pyExec = ""
|
||
|
|
py, err := detectExecutable(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Contains(t, py, "python3")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDetectPythonCache(t *testing.T) {
|
||
|
|
pyExec = "abc"
|
||
|
|
py, err := detectExecutable(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "abc", py)
|
||
|
|
pyExec = ""
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDetectVirtualEnvFalse(t *testing.T) {
|
||
|
|
venvDir, err := detectVirtualEnv()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "", venvDir)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMakeDetectableVenv(t *testing.T) {
|
||
|
|
var temp string
|
||
|
|
defer testTempdir(t, &temp)()
|
||
|
|
|
||
|
|
// TODO: rewrite with t.TempDir() and arguments
|
||
|
|
err := createVirtualEnv(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
venv, err := detectVirtualEnv()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, fmt.Sprintf("%s/.venv", temp), venv)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testTempdir(t *testing.T, dir *string) func() {
|
||
|
|
wd, _ := os.Getwd()
|
||
|
|
temp, err := os.MkdirTemp(os.TempDir(), "brickstest")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
os.Chdir(temp)
|
||
|
|
wd2, _ := os.Getwd()
|
||
|
|
*dir = wd2
|
||
|
|
return func() {
|
||
|
|
os.Chdir(wd)
|
||
|
|
os.RemoveAll(temp)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPyError(t *testing.T) {
|
||
|
|
_, err := Py(context.Background(), "__non_existing__.py")
|
||
|
|
assert.Contains(t, err.Error(), "can't open file")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPyInline(t *testing.T) {
|
||
|
|
hello, err := PyInline(context.Background(), "print('Hello, world!')")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "Hello, world!", hello)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPyInlineStderr(t *testing.T) {
|
||
|
|
detectExecutable(context.Background())
|
||
|
|
inline := "import sys; sys.stderr.write('___msg___'); sys.exit(1)"
|
||
|
|
_, err := PyInline(context.Background(), inline)
|
||
|
|
assert.EqualError(t, err, "___msg___")
|
||
|
|
}
|