mirror of https://github.com/databricks/cli.git
96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package config_tests
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
pathlib "path"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle/config/resources"
|
|
"github.com/databricks/cli/libs/dyn"
|
|
"github.com/databricks/databricks-sdk-go/service/jobs"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/exp/maps"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPythonImport_dataclass_no_wheel(t *testing.T) {
|
|
activateVEnv(t)
|
|
setPythonPath(t, "python_import/dataclass_no_wheel/src")
|
|
|
|
expected := &resources.Job{
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "Test Job",
|
|
JobClusters: []jobs.JobCluster{
|
|
{
|
|
JobClusterKey: "my_cluster",
|
|
},
|
|
},
|
|
Tasks: []jobs.Task{
|
|
{
|
|
NotebookTask: &jobs.NotebookTask{
|
|
NotebookPath: "notebooks/my_notebook.py",
|
|
},
|
|
JobClusterKey: "my_cluster",
|
|
TaskKey: "my_notebook_task",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
b := load(t, "./python_import/dataclass_no_wheel")
|
|
|
|
assert.Equal(t, []string{"my_job"}, maps.Keys(b.Config.Resources.Jobs))
|
|
|
|
myJob := b.Config.Resources.Jobs["my_job"]
|
|
assert.Equal(t, expected, myJob)
|
|
|
|
// NewCluster is reference to a variable and needs to be checked separately
|
|
err := b.Config.Mutate(func(value dyn.Value) (dyn.Value, error) {
|
|
path := dyn.MustPathFromString("resources.jobs.my_job.job_clusters[0].new_cluster")
|
|
value, err := dyn.GetByPath(value, path)
|
|
if err != nil {
|
|
return dyn.InvalidValue, err
|
|
}
|
|
|
|
assert.Equal(t, "${var.default_cluster_spec}", value.AsAny())
|
|
|
|
return value, nil
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func setPythonPath(t *testing.T, path string) {
|
|
wd, err := os.Getwd()
|
|
require.NoError(t, err)
|
|
t.Setenv("PYTHONPATH", pathlib.Join(wd, path))
|
|
}
|
|
|
|
func activateVEnv(t *testing.T) {
|
|
dir := t.TempDir()
|
|
venvDir := pathlib.Join(dir, "venv")
|
|
|
|
err := exec.Command("python3", "-m", "venv", venvDir).Run()
|
|
require.NoError(t, err, "failed to create venv")
|
|
|
|
// we don't have shell to activate venv, updating PATH is enough
|
|
|
|
var venvBinDir string
|
|
if runtime.GOOS == "windows" {
|
|
venvBinDir = pathlib.Join(venvDir, "Scripts")
|
|
t.Setenv("PATH", venvBinDir+";"+os.Getenv("PATH"))
|
|
} else {
|
|
venvBinDir = pathlib.Join(venvDir, "bin")
|
|
t.Setenv("PATH", venvBinDir+":"+os.Getenv("PATH"))
|
|
}
|
|
|
|
err = exec.Command(
|
|
pathlib.Join(venvBinDir, "pip"),
|
|
"install",
|
|
"databricks-pydabs==0.5.1",
|
|
).Run()
|
|
require.NoError(t, err, "failed to install databricks-pydabs")
|
|
}
|