Function TestPerformUpdate_FullFlow

internal/update/update_test.go:371–429

(t *testing.T)

}

func TestPerformUpdate_FullFlow(t *testing.T) {
	// Create a fake "current binary"
	dir := t.TempDir()
	binaryPath := filepath.Join(dir, "chief")
	os.WriteFile(binaryPath, []byte("old binary"), 0o755)

	// New binary content
	newContent := []byte("new binary v0.6.0")
	h := sha256.Sum256(newContent)
	expectedHash := hex.EncodeToString(h[:])

	binaryName := fmt.Sprintf("chief-%s-%s", runtime.GOOS, runtime.GOARCH)

	// Set up download server
	downloadSrv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path == "/binary" {
			w.Write(newContent)
		} else if r.URL.Path == "/checksum" {
			fmt.Fprintf(w, "%s  %s\n", expectedHash, binaryName)
		}
	}))
	defer downloadSrv.Close()

	release := Release{
		TagName: "v0.6.0",
		Assets: []Asset{
			{Name: binaryName, BrowserDownloadURL: downloadSrv.URL + "/binary"},
			{Name: binaryName + ".sha256", BrowserDownloadURL: downloadSrv.URL + "/checksum"},
		},
	}
	releaseSrv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		json.NewEncoder(w).Encode(release)
	}))
	defer releaseSrv.Close()

	// We can't easily test PerformUpdate because it calls os.Executable()
	// Instead, test the components used by PerformUpdate individually
	// (CheckForUpdate, findAssets, downloadToTemp, verifyChecksum are all tested above)

	// Test the download + checksum flow manually
	tmpFile, err := downloadToTemp(downloadSrv.URL+"/binary", dir)
	if err != nil {
		t.Fatalf("download failed: %v", err)
	}
	defer os.Remove(tmpFile)

	if err := verifyChecksum(tmpFile, downloadSrv.URL+"/checksum"); err != nil {
		t.Fatalf("checksum failed: %v", err)
	}

	// Verify file content
	data, err := os.ReadFile(tmpFile)
	if err != nil {
		t.Fatalf("reading downloaded file: %v", err)
	}
	if string(data) != string(newContent) {
		t.Errorf("downloaded content mismatch")
	}