mirror of https://github.com/databricks/cli.git
334 lines
11 KiB
Go
334 lines
11 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/databricks/bricks/git"
|
|
"github.com/databricks/bricks/libs/testfile"
|
|
"github.com/databricks/bricks/project"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func assertKeysOfMap(t *testing.T, m map[string]time.Time, expectedKeys []string) {
|
|
keys := make([]string, len(m))
|
|
i := 0
|
|
for k := range m {
|
|
keys[i] = k
|
|
i++
|
|
}
|
|
assert.ElementsMatch(t, expectedKeys, keys)
|
|
}
|
|
|
|
func TestDiff(t *testing.T) {
|
|
// Create temp project dir
|
|
projectDir := t.TempDir()
|
|
fileSet := git.NewFileSet(projectDir)
|
|
state := Snapshot{
|
|
LastUpdatedTimes: make(map[string]time.Time),
|
|
LocalToRemoteNames: make(map[string]string),
|
|
RemoteToLocalNames: make(map[string]string),
|
|
}
|
|
|
|
f1 := testfile.CreateFile(t, filepath.Join(projectDir, "hello.txt"))
|
|
defer f1.Close(t)
|
|
worldFilePath := filepath.Join(projectDir, "world.txt")
|
|
f2 := testfile.CreateFile(t, worldFilePath)
|
|
defer f2.Close(t)
|
|
|
|
// New files are put
|
|
files, err := fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err := state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 0)
|
|
assert.Len(t, change.put, 2)
|
|
assert.Contains(t, change.put, "hello.txt")
|
|
assert.Contains(t, change.put, "world.txt")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"hello.txt", "world.txt"})
|
|
assert.Equal(t, map[string]string{"hello.txt": "hello.txt", "world.txt": "world.txt"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"hello.txt": "hello.txt", "world.txt": "world.txt"}, state.RemoteToLocalNames)
|
|
|
|
// world.txt is editted
|
|
f2.Overwrite(t, "bunnies are cute.")
|
|
assert.NoError(t, err)
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Len(t, change.delete, 0)
|
|
assert.Len(t, change.put, 1)
|
|
assert.Contains(t, change.put, "world.txt")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"hello.txt", "world.txt"})
|
|
assert.Equal(t, map[string]string{"hello.txt": "hello.txt", "world.txt": "world.txt"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"hello.txt": "hello.txt", "world.txt": "world.txt"}, state.RemoteToLocalNames)
|
|
|
|
// hello.txt is deleted
|
|
f1.Remove(t)
|
|
assert.NoError(t, err)
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 1)
|
|
assert.Len(t, change.put, 0)
|
|
assert.Contains(t, change.delete, "hello.txt")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"world.txt"})
|
|
assert.Equal(t, map[string]string{"world.txt": "world.txt"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"world.txt": "world.txt"}, state.RemoteToLocalNames)
|
|
}
|
|
|
|
func TestFolderDiff(t *testing.T) {
|
|
// Create temp project dir
|
|
projectDir := t.TempDir()
|
|
fileSet := git.NewFileSet(projectDir)
|
|
state := Snapshot{
|
|
LastUpdatedTimes: make(map[string]time.Time),
|
|
LocalToRemoteNames: make(map[string]string),
|
|
RemoteToLocalNames: make(map[string]string),
|
|
}
|
|
|
|
err := os.Mkdir(filepath.Join(projectDir, "foo"), os.ModePerm)
|
|
assert.NoError(t, err)
|
|
f1 := testfile.CreateFile(t, filepath.Join(projectDir, "foo", "bar.py"))
|
|
defer f1.Close(t)
|
|
f1.Overwrite(t, "# Databricks notebook source\nprint(\"abc\")")
|
|
|
|
files, err := fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err := state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 0)
|
|
assert.Len(t, change.put, 1)
|
|
assert.Contains(t, change.put, "foo/bar.py")
|
|
|
|
f1.Remove(t)
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 1)
|
|
assert.Len(t, change.put, 0)
|
|
assert.Contains(t, change.delete, "foo/bar")
|
|
}
|
|
|
|
func TestPythonNotebookDiff(t *testing.T) {
|
|
// Create temp project dir
|
|
projectDir := t.TempDir()
|
|
fileSet := git.NewFileSet(projectDir)
|
|
state := Snapshot{
|
|
LastUpdatedTimes: make(map[string]time.Time),
|
|
LocalToRemoteNames: make(map[string]string),
|
|
RemoteToLocalNames: make(map[string]string),
|
|
}
|
|
|
|
foo := testfile.CreateFile(t, filepath.Join(projectDir, "foo.py"))
|
|
defer foo.Close(t)
|
|
|
|
// Case 1: notebook foo.py is uploaded
|
|
files, err := fileSet.All()
|
|
assert.NoError(t, err)
|
|
foo.Overwrite(t, "# Databricks notebook source\nprint(\"abc\")")
|
|
change, err := state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 0)
|
|
assert.Len(t, change.put, 1)
|
|
assert.Contains(t, change.put, "foo.py")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"foo.py"})
|
|
assert.Equal(t, map[string]string{"foo.py": "foo"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"foo": "foo.py"}, state.RemoteToLocalNames)
|
|
|
|
// Case 2: notebook foo.py is converted to python script by removing
|
|
// magic keyword
|
|
foo.Overwrite(t, "print(\"abc\")")
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 1)
|
|
assert.Len(t, change.put, 1)
|
|
assert.Contains(t, change.put, "foo.py")
|
|
assert.Contains(t, change.delete, "foo")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"foo.py"})
|
|
assert.Equal(t, map[string]string{"foo.py": "foo.py"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"foo.py": "foo.py"}, state.RemoteToLocalNames)
|
|
|
|
// Case 3: Python script foo.py is converted to a databricks notebook
|
|
foo.Overwrite(t, "# Databricks notebook source\nprint(\"def\")")
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 1)
|
|
assert.Len(t, change.put, 1)
|
|
assert.Contains(t, change.put, "foo.py")
|
|
assert.Contains(t, change.delete, "foo.py")
|
|
assertKeysOfMap(t, state.LastUpdatedTimes, []string{"foo.py"})
|
|
assert.Equal(t, map[string]string{"foo.py": "foo"}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{"foo": "foo.py"}, state.RemoteToLocalNames)
|
|
|
|
// Case 4: Python notebook foo.py is deleted, and its remote name is used in change.delete
|
|
foo.Remove(t)
|
|
assert.NoError(t, err)
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 1)
|
|
assert.Len(t, change.put, 0)
|
|
assert.Contains(t, change.delete, "foo")
|
|
assert.Len(t, state.LastUpdatedTimes, 0)
|
|
assert.Equal(t, map[string]string{}, state.LocalToRemoteNames)
|
|
assert.Equal(t, map[string]string{}, state.RemoteToLocalNames)
|
|
}
|
|
|
|
func TestErrorWhenIdenticalRemoteName(t *testing.T) {
|
|
// Create temp project dir
|
|
projectDir := t.TempDir()
|
|
fileSet := git.NewFileSet(projectDir)
|
|
state := Snapshot{
|
|
LastUpdatedTimes: make(map[string]time.Time),
|
|
LocalToRemoteNames: make(map[string]string),
|
|
RemoteToLocalNames: make(map[string]string),
|
|
}
|
|
|
|
// upload should work since they point to different destinations
|
|
pythonFoo := testfile.CreateFile(t, filepath.Join(projectDir, "foo.py"))
|
|
defer pythonFoo.Close(t)
|
|
vanillaFoo := testfile.CreateFile(t, filepath.Join(projectDir, "foo"))
|
|
defer vanillaFoo.Close(t)
|
|
files, err := fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err := state.diff(files)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, change.delete, 0)
|
|
assert.Len(t, change.put, 2)
|
|
assert.Contains(t, change.put, "foo.py")
|
|
assert.Contains(t, change.put, "foo")
|
|
|
|
// errors out because they point to the same destination
|
|
pythonFoo.Overwrite(t, "# Databricks notebook source\nprint(\"def\")")
|
|
files, err = fileSet.All()
|
|
assert.NoError(t, err)
|
|
change, err = state.diff(files)
|
|
assert.ErrorContains(t, err, "both foo and foo.py point to the same remote file location foo. Please remove one of them from your local project")
|
|
}
|
|
|
|
func TestNewSnapshotDefaults(t *testing.T) {
|
|
ctx := setupProject(t)
|
|
snapshot, err := newSnapshot(ctx, "/Repos/foo/bar")
|
|
prj := project.Get(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, LatestSnapshotVersion, snapshot.Version)
|
|
assert.Equal(t, "/Repos/foo/bar", snapshot.RemotePath)
|
|
assert.Equal(t, prj.WorkspacesClient().Config.Host, snapshot.Host)
|
|
assert.Empty(t, snapshot.LastUpdatedTimes)
|
|
assert.Empty(t, snapshot.RemoteToLocalNames)
|
|
assert.Empty(t, snapshot.LocalToRemoteNames)
|
|
}
|
|
|
|
func getEmptySnapshot() Snapshot {
|
|
return Snapshot{
|
|
LastUpdatedTimes: make(map[string]time.Time),
|
|
LocalToRemoteNames: make(map[string]string),
|
|
RemoteToLocalNames: make(map[string]string),
|
|
}
|
|
}
|
|
|
|
func setupProject(t *testing.T) context.Context {
|
|
projectDir := t.TempDir()
|
|
ctx := context.TODO()
|
|
t.Setenv("DATABRICKS_HOST", "www.foobar.com")
|
|
ctx, err := project.Initialize(ctx, projectDir, "development")
|
|
assert.NoError(t, err)
|
|
return ctx
|
|
}
|
|
|
|
func TestOldSnapshotInvalidation(t *testing.T) {
|
|
oldVersionSnapshot := `{
|
|
"version": "v0",
|
|
"host": "www.foobar.com",
|
|
"remote_path": "/Repos/foo/bar",
|
|
"last_modified_times": {},
|
|
"local_to_remote_names": {},
|
|
"remote_to_local_names": {}
|
|
}`
|
|
ctx := setupProject(t)
|
|
emptySnapshot := getEmptySnapshot()
|
|
snapshotPath, err := emptySnapshot.getPath(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
snapshotFile := testfile.CreateFile(t, snapshotPath)
|
|
snapshotFile.Overwrite(t, oldVersionSnapshot)
|
|
snapshotFile.Close(t)
|
|
|
|
assert.FileExists(t, snapshotPath)
|
|
snapshot := emptySnapshot
|
|
err = snapshot.loadSnapshot(ctx)
|
|
assert.NoError(t, err)
|
|
// assert snapshot did not get loaded
|
|
assert.Equal(t, emptySnapshot, snapshot)
|
|
}
|
|
|
|
func TestNoVersionSnapshotInvalidation(t *testing.T) {
|
|
noVersionSnapshot := `{
|
|
"host": "www.foobar.com",
|
|
"remote_path": "/Repos/foo/bar",
|
|
"last_modified_times": {},
|
|
"local_to_remote_names": {},
|
|
"remote_to_local_names": {}
|
|
}`
|
|
ctx := setupProject(t)
|
|
emptySnapshot := getEmptySnapshot()
|
|
snapshotPath, err := emptySnapshot.getPath(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
snapshotFile := testfile.CreateFile(t, snapshotPath)
|
|
snapshotFile.Overwrite(t, noVersionSnapshot)
|
|
snapshotFile.Close(t)
|
|
|
|
assert.FileExists(t, snapshotPath)
|
|
snapshot := emptySnapshot
|
|
err = snapshot.loadSnapshot(ctx)
|
|
assert.NoError(t, err)
|
|
// assert snapshot did not get loaded
|
|
assert.Equal(t, emptySnapshot, snapshot)
|
|
}
|
|
|
|
func TestLatestVersionSnapshotGetsLoaded(t *testing.T) {
|
|
latestVersionSnapshot := fmt.Sprintf(`{
|
|
"version": "%s",
|
|
"host": "www.foobar.com",
|
|
"remote_path": "/Repos/foo/bar",
|
|
"last_modified_times": {},
|
|
"local_to_remote_names": {},
|
|
"remote_to_local_names": {}
|
|
}`, LatestSnapshotVersion)
|
|
|
|
ctx := setupProject(t)
|
|
emptySnapshot := getEmptySnapshot()
|
|
snapshotPath, err := emptySnapshot.getPath(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
snapshotFile := testfile.CreateFile(t, snapshotPath)
|
|
snapshotFile.Overwrite(t, latestVersionSnapshot)
|
|
snapshotFile.Close(t)
|
|
|
|
assert.FileExists(t, snapshotPath)
|
|
snapshot := emptySnapshot
|
|
err = snapshot.loadSnapshot(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
// assert snapshot gets loaded
|
|
assert.NotEqual(t, emptySnapshot, snapshot)
|
|
assert.Equal(t, LatestSnapshotVersion, snapshot.Version)
|
|
assert.Equal(t, "www.foobar.com", snapshot.Host)
|
|
assert.Equal(t, "/Repos/foo/bar", snapshot.RemotePath)
|
|
}
|