databricks-cli/bundle/artifacts/upload_test.go

package artifacts

import (
	"context"
	"os"
	"path/filepath"
	"testing"

	"github.com/databricks/cli/bundle"
	"github.com/databricks/cli/bundle/config"
	"github.com/databricks/cli/bundle/internal/bundletest"
	"github.com/databricks/cli/libs/diag"
	"github.com/databricks/cli/libs/testfile"
	"github.com/stretchr/testify/require"
)

type noop struct{}

func (n *noop) Apply(context.Context, *bundle.Bundle) diag.Diagnostics {
	return nil
}

func (n *noop) Name() string {
	return "noop"
}

func TestExpandGlobFilesSource(t *testing.T) {
	rootPath := t.TempDir()
	err := os.Mkdir(filepath.Join(rootPath, "test"), 0755)
	require.NoError(t, err)

	t1 := testfile.CreateFile(t, filepath.Join(rootPath, "test", "myjar1.jar"))
	t1.Close(t)

	t2 := testfile.CreateFile(t, filepath.Join(rootPath, "test", "myjar2.jar"))
	t2.Close(t)

	b := &bundle.Bundle{
		RootPath: rootPath,
		Config: config.Root{
			Artifacts: map[string]*config.Artifact{
				"test": {
					Type: "custom",
					Files: []config.ArtifactFile{
						{
							Source: filepath.Join("..", "test", "*.jar"),
						},
					},
				},
			},
		},
	}

	bundletest.SetLocation(b, ".", filepath.Join(rootPath, "resources", "artifacts.yml"))

	u := &upload{"test"}
	uploadMutators[config.ArtifactType("custom")] = func(name string) bundle.Mutator {
		return &noop{}
	}

	bm := &build{"test"}
	buildMutators[config.ArtifactType("custom")] = func(name string) bundle.Mutator {
		return &noop{}
	}

	pm := &prepare{"test"}
	prepareMutators[config.ArtifactType("custom")] = func(name string) bundle.Mutator {
		return &noop{}
	}

	diags := bundle.Apply(context.Background(), b, bundle.Seq(pm, bm, u))
	require.NoError(t, diags.Error())

	require.Equal(t, 2, len(b.Config.Artifacts["test"].Files))
	require.Equal(t, filepath.Join(rootPath, "test", "myjar1.jar"), b.Config.Artifacts["test"].Files[0].Source)
	require.Equal(t, filepath.Join(rootPath, "test", "myjar2.jar"), b.Config.Artifacts["test"].Files[1].Source)
}

func TestExpandGlobFilesSourceWithNoMatches(t *testing.T) {
	rootPath := t.TempDir()
	err := os.Mkdir(filepath.Join(rootPath, "test"), 0755)
	require.NoError(t, err)

	b := &bundle.Bundle{
		RootPath: rootPath,
		Config: config.Root{
			Artifacts: map[string]*config.Artifact{
				"test": {
					Type: "custom",
					Files: []config.ArtifactFile{
						{
							Source: filepath.Join("..", "test", "myjar.jar"),
						},
					},
				},
			},
		},
	}

	bundletest.SetLocation(b, ".", filepath.Join(rootPath, "resources", "artifacts.yml"))

	u := &upload{"test"}
	uploadMutators[config.ArtifactType("custom")] = func(name string) bundle.Mutator {
		return &noop{}
	}

	bm := &build{"test"}
	buildMutators[config.ArtifactType("custom")] = func(name string) bundle.Mutator {
		return &noop{}
	}

	diags := bundle.Apply(context.Background(), b, bundle.Seq(bm, u))
	require.ErrorContains(t, diags.Error(), "no files found for")
}