databricks-cli/cmd/sync/sync_test.go

package sync

import (
	"flag"
	"path/filepath"
	"testing"

	"github.com/databricks/bricks/bundle"
	"github.com/databricks/bricks/bundle/config"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestSyncOptionsFromBundle(t *testing.T) {
	tempDir := t.TempDir()
	b := &bundle.Bundle{
		Config: config.Root{
			Path: tempDir,

			Bundle: config.Bundle{
				Environment: "default",
			},

			Workspace: config.Workspace{
				FilePath: config.PathLike{
					Workspace: "/Users/jane@doe.com/path",
				},
			},
		},
	}

	opts, err := syncOptionsFromBundle(syncCmd, []string{}, b)
	require.NoError(t, err)
	assert.Equal(t, tempDir, opts.LocalPath)
	assert.Equal(t, "/Users/jane@doe.com/path", opts.RemotePath)
	assert.Equal(t, filepath.Join(tempDir, ".databricks", "bundle", "default"), opts.SnapshotBasePath)
	assert.NotNil(t, opts.WorkspaceClient)
}

func TestSyncOptionsFromArgsRequiredTwoArgs(t *testing.T) {
	var err error
	_, err = syncOptionsFromArgs(syncCmd, []string{})
	require.ErrorIs(t, err, flag.ErrHelp)
	_, err = syncOptionsFromArgs(syncCmd, []string{"foo"})
	require.ErrorIs(t, err, flag.ErrHelp)
	_, err = syncOptionsFromArgs(syncCmd, []string{"foo", "bar", "qux"})
	require.ErrorIs(t, err, flag.ErrHelp)
}

func TestSyncOptionsFromArgs(t *testing.T) {
	opts, err := syncOptionsFromArgs(syncCmd, []string{"/local", "/remote"})
	require.NoError(t, err)
	assert.Equal(t, "/local", opts.LocalPath)
	assert.Equal(t, "/remote", opts.RemotePath)
}