From 812267c2d82d887dc143ec10abda6d4b8474f757 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Fri, 27 Feb 2026 17:18:32 +0200
Subject: ior: route live mode through trace pipeline

---
 internal/ior_mode_test.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'internal/ior_mode_test.go')

diff --git a/internal/ior_mode_test.go b/internal/ior_mode_test.go
index 42b0a48..35d7f43 100644
--- a/internal/ior_mode_test.go
+++ b/internal/ior_mode_test.go
@@ -34,6 +34,12 @@ func TestShouldRunTraceMode(t *testing.T) {
 	if !shouldRunTraceMode(withPprof) {
 		t.Fatalf("expected pprof mode to use trace mode")
 	}
+
+	withLive := base
+	withLive.LiveFlamegraph = true
+	if !shouldRunTraceMode(withLive) {
+		t.Fatalf("expected live mode to use trace mode")
+	}
 }
 
 func TestShouldAutoStopByDuration(t *testing.T) {
@@ -59,6 +65,12 @@ func TestShouldAutoStopByDuration(t *testing.T) {
 	if !shouldAutoStopByDuration(withPprof) {
 		t.Fatalf("expected pprof mode to auto-stop by duration")
 	}
+
+	withLive := base
+	withLive.LiveFlamegraph = true
+	if shouldAutoStopByDuration(withLive) {
+		t.Fatalf("expected live mode not to auto-stop by duration")
+	}
 }
 
 func TestDispatchRunUsesTraceModeWhenRequested(t *testing.T) {
-- 
cgit v1.2.3