package deploy import ( "bytes" "encoding/json" "testing" "github.com/databricks/cli/internal/testutil" "github.com/databricks/cli/libs/fileset" "github.com/databricks/cli/libs/vfs" "github.com/stretchr/testify/require" ) func TestFromSlice(t *testing.T) { tmpDir := t.TempDir() fileset := fileset.New(vfs.MustNew(tmpDir)) testutil.Touch(t, tmpDir, "test1.py") testutil.Touch(t, tmpDir, "test2.py") testutil.Touch(t, tmpDir, "test3.py") files, err := fileset.All() require.NoError(t, err) f, err := FromSlice(files) require.NoError(t, err) require.Len(t, f, 3) for _, file := range f { require.Contains(t, []string{"test1.py", "test2.py", "test3.py"}, file.LocalPath) } } func TestToSlice(t *testing.T) { tmpDir := t.TempDir() fileset := fileset.New(vfs.MustNew(tmpDir)) testutil.Touch(t, tmpDir, "test1.py") testutil.Touch(t, tmpDir, "test2.py") testutil.Touch(t, tmpDir, "test3.py") files, err := fileset.All() require.NoError(t, err) f, err := FromSlice(files) require.NoError(t, err) require.Len(t, f, 3) s := f.ToSlice(tmpDir) require.Len(t, s, 3) for _, file := range s { require.Contains(t, []string{"test1.py", "test2.py", "test3.py"}, file.Relative) // If the mtime is not zero we know we produced a valid fs.DirEntry. ts := file.Modified() require.NotZero(t, ts) } } func TestIsLocalStateStale(t *testing.T) { oldState, err := json.Marshal(DeploymentState{ Seq: 1, }) require.NoError(t, err) newState, err := json.Marshal(DeploymentState{ Seq: 2, }) require.NoError(t, err) require.True(t, isLocalStateStale(bytes.NewReader(oldState), bytes.NewReader(newState))) require.False(t, isLocalStateStale(bytes.NewReader(newState), bytes.NewReader(oldState))) }