flamegraph: include height/count in view cache key (2p)

2 files changed, 40 insertions, 0 deletions

diff --git a/internal/tui/flamegraph/model.go b/internal/tui/flamegraph/model.go
index 7be2864..b73368f 100644
--- a/internal/tui/flamegraph/model.go
+++ b/internal/tui/flamegraph/model.go
@@ -36,6 +36,8 @@ type flameViewCacheKey struct {
 	searchQuery   string
 	statusMessage string
 	zoomPath      string
+	countField    string
+	heightField   string
 	searchActive  bool
 	showHelp      bool
 	paused        bool
@@ -559,6 +561,8 @@ func (m Model) currentViewCacheKey() flameViewCacheKey {
 		searchQuery:   m.searchQuery,
 		statusMessage: m.statusMessage,
 		zoomPath:      m.zoomPath,
+		countField:    m.countField,
+		heightField:   m.heightField,
 		searchActive:  m.searchActive,
 		showHelp:      m.showHelp,
 		paused:        m.paused,
diff --git a/internal/tui/flamegraph/model_test.go b/internal/tui/flamegraph/model_test.go
index 80a86bb..f31a9a1 100644
--- a/internal/tui/flamegraph/model_test.go
+++ b/internal/tui/flamegraph/model_test.go
@@ -1103,6 +1103,42 @@ func TestNewModelAlignsHeightFieldToLiveTrie(t *testing.T) {
 	}
 }
 
+func TestCurrentViewCacheKeyChangesWhenCountFieldChanges(t *testing.T) {
+	m := NewModel(nil)
+	m.width = 120
+	m.height = 20
+	m.frames = []tuiFrame{{Name: "root", Path: "root"}}
+
+	before := m.currentViewCacheKey()
+	m.countField = "bytes"
+	after := m.currentViewCacheKey()
+
+	if before == after {
+		t.Fatalf("expected cache key to change when count field changes")
+	}
+	if got, want := after.countField, "bytes"; got != want {
+		t.Fatalf("expected cache key count field %q, got %q", want, got)
+	}
+}
+
+func TestCurrentViewCacheKeyChangesWhenHeightFieldChanges(t *testing.T) {
+	m := NewModel(nil)
+	m.width = 120
+	m.height = 20
+	m.frames = []tuiFrame{{Name: "root", Path: "root"}}
+
+	before := m.currentViewCacheKey()
+	m.heightField = "duration"
+	after := m.currentViewCacheKey()
+
+	if before == after {
+		t.Fatalf("expected cache key to change when height field changes")
+	}
+	if got, want := after.heightField, "duration"; got != want {
+		t.Fatalf("expected cache key height field %q, got %q", want, got)
+	}
+}
+
 func TestControlHelpToggle(t *testing.T) {
 	m := NewModel(nil)
 	m = pressFlameKey(t, m, tea.KeyPressMsg{Code: []rune{'?'}[0], Text: "?"})