,,

Author: nfx

cmd/labs/project/installer_test.go

@@ -121,7 +121,47 @@ func respondWithJSON(t *testing.T, w http.ResponseWriter, v any) {
 	w.Write(raw)
 }
 
+type fileTree struct {
+	Path     string
+	MaxDepth int
+}
+
+func (ft fileTree) String() string {
+	lines := ft.listFiles(ft.Path, ft.MaxDepth)
+	return strings.Join(lines, "\n")
+}
+
+func (ft fileTree) listFiles(dir string, depth int) (lines []string) {
+	if ft.MaxDepth > 0 && depth > ft.MaxDepth {
+		return []string{fmt.Sprintf("deeper than %d levels", ft.MaxDepth)}
+	}
+	fileInfo, err := os.ReadDir(dir)
+	if err != nil {
+		return []string{err.Error()}
+	}
+	for _, entry := range fileInfo {
+		lines = append(lines, fmt.Sprintf("%s%s", ft.getIndent(depth), entry.Name()))
+		if entry.IsDir() {
+			subdir := filepath.Join(dir, entry.Name())
+			lines = append(lines, ft.listFiles(subdir, depth+1)...)
+		}
+	}
+	return lines
+}
+
+func (ft fileTree) getIndent(depth int) string {
+	return "│" + strings.Repeat(" ", depth*2) + "├─ "
+}
+
 func TestInstallerWorksForReleases(t *testing.T) {
+	defer func() {
+		if !t.Failed() {
+			return
+		}
+		t.Logf("file tree:\n%s", fileTree{
+			Path: filepath.Dir(t.TempDir()),
+		})
+	}()
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if r.URL.Path == "/databrickslabs/blueprint/v0.3.15/labs.yml" {
 			raw, err := os.ReadFile("testdata/installed-in-home/.databricks/labs/blueprint/lib/labs.yml")
@@ -192,6 +232,14 @@ func TestInstallerWorksForReleases(t *testing.T) {
 }
 
 func TestInstallerWorksForDevelopment(t *testing.T) {
+	defer func() {
+		if !t.Failed() {
+			return
+		}
+		t.Logf("file tree:\n%s", fileTree{
+			Path: filepath.Dir(t.TempDir()),
+		})
+	}()
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if r.URL.Path == "/api/2.0/clusters/list" {
 			respondWithJSON(t, w, compute.ListClustersResponse{
@@ -302,3 +350,65 @@ account_id = abc
 	r.RunBackground()
 	r.WaitForTextPrinted("setting up important infrastructure", 5*time.Second)
 }
+
+func TestUpgraderWorksForReleases(t *testing.T) {
+	defer func() {
+		if !t.Failed() {
+			return
+		}
+		t.Logf("file tree:\n%s", fileTree{
+			Path: filepath.Dir(t.TempDir()),
+		})
+	}()
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/databrickslabs/blueprint/v0.4.0/labs.yml" {
+			raw, err := os.ReadFile("testdata/installed-in-home/.databricks/labs/blueprint/lib/labs.yml")
+			if err != nil {
+				panic(err)
+			}
+			w.Write(raw)
+			return
+		}
+		if r.URL.Path == "/repos/databrickslabs/blueprint/zipball/v0.4.0" {
+			raw, err := zipballFromFolder("testdata/installed-in-home/.databricks/labs/blueprint/lib")
+			if err != nil {
+				panic(err)
+			}
+			w.Header().Add("Content-Type", "application/octet-stream")
+			w.Write(raw)
+			return
+		}
+		if r.URL.Path == "/api/2.0/clusters/get" {
+			respondWithJSON(t, w, &compute.ClusterDetails{
+				State: compute.StateRunning,
+			})
+			return
+		}
+		t.Logf("Requested: %s", r.URL.Path)
+		t.FailNow()
+	}))
+	defer server.Close()
+
+	ctx := installerContext(t, server)
+
+	newHome := copyTestdata(t, "testdata/installed-in-home")
+	ctx = env.WithUserHomeDir(ctx, newHome)
+
+	py, _ := python.DetectExecutable(ctx)
+	py, _ = filepath.Abs(py)
+	ctx = env.Set(ctx, "PYTHON_BIN", py)
+
+	cachePath := project.PathInLabs(ctx, "blueprint", "cache")
+	bs := []byte(`{"refreshed_at": "2033-01-01T00:00:00.92857+02:00","data": [{"tag_name": "v0.4.0"}]}`)
+	err := os.WriteFile(filepath.Join(cachePath, "databrickslabs-blueprint-releases.json"), bs, ownerRW)
+	require.NoError(t, err)
+
+	// simulate the case of GitHub Actions
+	ctx = env.Set(ctx, "DATABRICKS_HOST", server.URL)
+	ctx = env.Set(ctx, "DATABRICKS_TOKEN", "...")
+	ctx = env.Set(ctx, "DATABRICKS_CLUSTER_ID", "installer-cluster")
+	ctx = env.Set(ctx, "DATABRICKS_WAREHOUSE_ID", "installer-warehouse")
+
+	r := internal.NewCobraTestRunnerWithContext(t, ctx, "labs", "upgrade", "blueprint")
+	r.RunAndExpectOutput("setting up important infrastructure")
+}