steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test fix

This commit is contained in:
kartikgupta-db 2023-09-05 19:10:36 +02:00
parent e01578663b
commit 6c16dc2bef
No known key found for this signature in database
GPG Key ID: 6AD5FA11FACDEA39
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bundle/config/mutator/trampoline.go
View File

@ -3,7 +3,6 @@ package mutator
import (
"context"
"fmt"
"log"
"os"
"path"
"path/filepath"
@ -59,7 +58,6 @@ func (m *trampoline) Name() string {
func (m *trampoline) Apply(ctx context.Context, b *bundle.Bundle) error {
tasks := m.functions.GetTasks(b)
for _, task := range tasks {
log.Default().Printf("%s, %s task", task.Task.TaskKey, task.Task.NotebookTask.NotebookPath)
err := m.generateNotebookWrapper(b, task)
if err != nil {
return err

4
bundle/config/mutator/trampoline_test.go
View File

@ -90,7 +90,7 @@ func TestGenerateTrampoline(t *testing.T) {
dir, err := b.InternalDir()
require.NoError(t, err)
filename := filepath.Join(dir, "notebook_test_to_trampoline.py")
filename := filepath.Join(dir, "notebook_test_trampoline_test_to_trampoline.py")
bytes, err := os.ReadFile(filename)
require.NoError(t, err)
@ -98,6 +98,6 @@ func TestGenerateTrampoline(t *testing.T) {
require.Equal(t, "Hello from Trampoline", string(bytes))
task := b.Config.Resources.Jobs["test"].Tasks[0]
require.Equal(t, task.NotebookTask.NotebookPath, ".databricks/bundle/development/.internal/notebook_test_to_trampoline")
require.Equal(t, task.NotebookTask.NotebookPath, ".databricks/bundle/development/.internal/notebook_test_trampoline_test_to_trampoline")
require.Nil(t, task.PythonWheelTask)
}