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

add unit test

This commit is contained in:
Shreyas Goenka 2024-12-31 12:53:25 +05:30
parent 9d8ba099ba
commit 09bf4fa90c
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
integration/libs/filer/filer_test.go
View File

@ -914,10 +914,35 @@ func TestDbfsFilerForStreamingUploads(t *testing.T) {
err := os.WriteFile(filepath.Join(tmpDir, "foo.txt"), []byte("foobar"), 0o644)
require.NoError(t, err)
fd, err := os.Open(filepath.Join(tmpDir, "foo.txt"))
require.NoError(t, err)
defer fd.Close()
// Write a file with streaming upload
err = f.Write(ctx, "foo.txt", strings.NewReader("foo"))
err = f.Write(ctx, "foo.txt", fd)
require.NoError(t, err)
// Assert contents
filerTest{t, f}.assertContents(ctx, "foo.txt", "foo")
filerTest{t, f}.assertContents(ctx, "foo.txt", "foobar")
}
func TestDbfsFilerForPutUploads(t *testing.T) {
ctx := context.Background()
f, _ := setupDbfsFiler(t)
// Write a file to local disk.
tmpDir := t.TempDir()
err := os.WriteFile(filepath.Join(tmpDir, "foo.txt"), []byte("foobar"), 0o644)
require.NoError(t, err)
fd, err := os.Open(filepath.Join(tmpDir, "foo.txt"))
require.NoError(t, err)
defer fd.Close()
// Write a file with PUT upload
err = f.Write(ctx, "foo.txt", fd)
require.NoError(t, err)
// Assert contents
filerTest{t, f}.assertContents(ctx, "foo.txt", "foobar")
}