summaryrefslogtreecommitdiff
path: root/internal/tmux/status_more_test.go
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2025-09-17 21:33:45 +0300
committerPaul Buetow <paul@buetow.org>2025-09-17 21:33:45 +0300
commit88103657fb230bb41217a06aa5602ae23e7acb8b (patch)
tree524c437e4e40ee5d6713b6ea5414ad975654cc52 /internal/tmux/status_more_test.go
parent2b6232704ecc90630196b9f829f966533e5cdccd (diff)
feat(stats,tmux): global Σ@window stats across processes with flocked cache; width mitigation (narrow/maxlen); configurable [stats] window_minutes; robust coverage parsing; docs update\n\n- Add internal/stats with windowed event cache + flock + atomic writes\n- Wire stats into LSP/CLI/Tmux Action; tmux shows Σ@window with per-model tail\n- HEXAI_TMUX_STATUS_NARROW and HEXAI_TMUX_STATUS_MAXLEN for width control\n- Add [stats] window_minutes to config and apply on startup\n- Improve Magefile coverage handling; add tests to lift coverage >85%\n- Update docs/tmux.md and config example
Diffstat (limited to 'internal/tmux/status_more_test.go')
-rw-r--r--internal/tmux/status_more_test.go62
1 files changed, 62 insertions, 0 deletions
diff --git a/internal/tmux/status_more_test.go b/internal/tmux/status_more_test.go
new file mode 100644
index 0000000..deaf57d
--- /dev/null
+++ b/internal/tmux/status_more_test.go
@@ -0,0 +1,62 @@
+package tmux
+
+import (
+ "os"
+ "testing"
+ "time"
+)
+
+func TestFormatLLMStatsStatus_Basic(t *testing.T) {
+ s := FormatLLMStatsStatus("gpt-4.1", 5, 0.8, 1234, 5678)
+ if s == "" || !containsAll(s, []string{"LLM:gpt-4.1", "5r", "0.8rpm"}) {
+ t.Fatalf("unexpected status: %q", s)
+ }
+}
+
+func TestFormatLLMStatsStatusColored_Basic(t *testing.T) {
+ s := FormatLLMStatsStatusColored("openai", "gpt-4.1", 2, 1.2, 100, 200)
+ if s == "" || !containsAll(s, []string{"LLM:openai:gpt-4.1", "rpm", "2r"}) {
+ t.Fatalf("colored status missing parts: %q", s)
+ }
+}
+
+func TestFormatGlobalStatusColored_NarrowAndMaxLen(t *testing.T) {
+ // Narrow mode should elide the tail
+ os.Setenv("HEXAI_TMUX_STATUS_NARROW", "1")
+ defer os.Unsetenv("HEXAI_TMUX_STATUS_NARROW")
+ s := FormatGlobalStatusColored(10, 3.3, 1000, 2000, "prov", "model", 1.1, 4, 30*time.Minute)
+ if containsAll(s, []string{"|", "prov:model"}) {
+ t.Fatalf("narrow mode should not include tail: %q", s)
+ }
+ // Max length should also drop the tail when it would overflow
+ os.Unsetenv("HEXAI_TMUX_STATUS_NARROW")
+ os.Setenv("HEXAI_TMUX_STATUS_MAXLEN", "10")
+ defer os.Unsetenv("HEXAI_TMUX_STATUS_MAXLEN")
+ s2 := FormatGlobalStatusColored(10, 3.3, 1000, 2000, "prov", "model", 1.1, 4, 30*time.Minute)
+ if containsAll(s2, []string{"|", "prov:model"}) {
+ t.Fatalf("maxlen should drop tail when overflowing: %q", s2)
+ }
+}
+
+func containsAll(s string, parts []string) bool {
+ for _, p := range parts {
+ if !contains(s, p) {
+ return false
+ }
+ }
+ return true
+}
+
+func contains(s, sub string) bool {
+ return len(s) >= len(sub) && (len(sub) == 0 || (len(s) > 0 && indexOf(s, sub) >= 0))
+}
+
+func indexOf(s, sub string) int {
+ // tiny wrapper to avoid importing strings
+ for i := 0; i+len(sub) <= len(s); i++ {
+ if s[i:i+len(sub)] == sub {
+ return i
+ }
+ }
+ return -1
+}