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

test: Modify writeProjectFiles to handle errors internally in tests

This commit is contained in:
Denis Bilenko (aider) 2025-03-03 20:58:28 +01:00
parent ea5690eb67
commit 97109ec438
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/patchwheel/patchwheel_test.go
View File

@ -51,17 +51,16 @@ def print_version():
} }
} }
func writeProjectFiles(baseDir string, files map[string]string) error { func writeProjectFiles(t *testing.T, baseDir string, files map[string]string) {
for path, content := range files { for path, content := range files {
fullPath := filepath.Join(baseDir, path) fullPath := filepath.Join(baseDir, path)
if err := os.MkdirAll(filepath.Dir(fullPath), 0o755); err != nil { if err := os.MkdirAll(filepath.Dir(fullPath), 0o755); err != nil {
return err t.Fatalf("Failed to create directory %s: %v", filepath.Dir(fullPath), err)
} }
if err := os.WriteFile(fullPath, []byte(content), 0o644); err != nil { if err := os.WriteFile(fullPath, []byte(content), 0o644); err != nil {
return err t.Fatalf("Failed to write file %s: %v", fullPath, err)
} }
} }
return nil
} }
func runCmd(t *testing.T, dir, name string, args ...string) { func runCmd(t *testing.T, dir, name string, args ...string) {
@ -174,10 +173,7 @@ func TestPatchWheel(t *testing.T) {
tempDir := t.TempDir() tempDir := t.TempDir()
projFiles := minimalPythonProject() projFiles := minimalPythonProject()
// AI TODO: modify writeProjectFiles to accept t as first parameter and fail test on errors in internally writeProjectFiles(t, tempDir, projFiles)
if err := writeProjectFiles(tempDir, projFiles); err != nil {
t.Fatal(err)
}
runCmd(t, tempDir, "uv", "venv", "-q", "--python", py) runCmd(t, tempDir, "uv", "venv", "-q", "--python", py)