ik0 replace remaining test seams with DI

2 files changed, 18 insertions, 1 deletions

diff --git a/internal/hexaiaction/tui_custom.go b/internal/hexaiaction/tui_custom.go
index 910242f..876f60a 100644
--- a/internal/hexaiaction/tui_custom.go
+++ b/internal/hexaiaction/tui_custom.go
@@ -29,7 +29,7 @@ func (r tuiRunner) program(m model) teaProgram {
 func (r tuiRunner) RunTUIWithCustom(customs []appconfig.CustomAction, menuHotkey string) (ActionKind, *appconfig.CustomAction, error) {
 	// When no customs, fall back to default menu
 	if len(customs) == 0 {
-		kind, err := RunTUI()
+		kind, err := r.RunTUI()
 		return kind, nil, err
 	}
 	// Build main menu with an extra entry
diff --git a/internal/hexaiaction/tui_custom_test.go b/internal/hexaiaction/tui_custom_test.go
index 7f7e2d3..62d37cb 100644
--- a/internal/hexaiaction/tui_custom_test.go
+++ b/internal/hexaiaction/tui_custom_test.go
@@ -71,3 +71,20 @@ func TestRunTUIWithCustom_SubmenuAndHotkeys(t *testing.T) {
 		t.Fatalf("expected selected custom a, got %+v", selected)
 	}
 }
+
+func TestRunTUIWithCustom_EmptyCustomsUsesRunner(t *testing.T) {
+	r := tuiRunner{newProgram: func(m model) teaProgram {
+		return fakeProg{m: m, onRun: func(mm *model) { mm.chosen = ActionDocument }}
+	}}
+
+	kind, selected, err := r.RunTUIWithCustom(nil, "")
+	if err != nil {
+		t.Fatalf("RunTUIWithCustom error: %v", err)
+	}
+	if kind != ActionDocument {
+		t.Fatalf("kind = %s, want %s", kind, ActionDocument)
+	}
+	if selected != nil {
+		t.Fatalf("selected = %+v, want nil", selected)
+	}
+}