ior: route live mode through trace pipeline

1 files changed, 12 insertions, 0 deletions

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) {