diff options
| -rw-r--r-- | internal/ior.go | 2 | ||||
| -rw-r--r-- | internal/ior_mode_test.go | 12 |
2 files changed, 13 insertions, 1 deletions
diff --git a/internal/ior.go b/internal/ior.go index a19d2f7..bf0fb1f 100644 --- a/internal/ior.go +++ b/internal/ior.go @@ -147,7 +147,7 @@ func dispatchRun(cfg flags.Flags) error { } func shouldRunTraceMode(cfg flags.Flags) bool { - return cfg.PlainMode || cfg.FlamegraphEnable || cfg.PprofEnable + return cfg.PlainMode || cfg.FlamegraphEnable || cfg.LiveFlamegraph || cfg.PprofEnable } func tuiTraceStarterFromRunTrace( 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) { |