test: add AwaitPrompt parser and shell prompt marker tests

Cover AwaitPrompt parsing with and without @timeout variants, and
verify that all 9 supported shells embed the OSC 7777 prompt marker.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: MattieTK

parser/parser_test.go

@@ -33,7 +33,9 @@ Sleep 100ms
 Sleep 3
 Wait
 Wait+Screen
-Wait@100ms /foobar/`
+Wait@100ms /foobar/
+AwaitPrompt
+AwaitPrompt@30s`
 
 	expected := []Command{
 		{Type: token.SET, Options: "TypingSpeed", Args: "100ms"},
@@ -59,6 +61,8 @@ Wait@100ms /foobar/`
 		{Type: token.WAIT, Args: "Line"},
 		{Type: token.WAIT, Args: "Screen"},
 		{Type: token.WAIT, Options: "100ms", Args: "Line foobar"},
+		{Type: token.AWAIT_PROMPT, Options: "", Args: ""},
+		{Type: token.AWAIT_PROMPT, Options: "30s", Args: ""},
 	}
 
 	l := lexer.New(input)
@@ -291,6 +295,66 @@ func TestParseCtrl(t *testing.T) {
 	}
 }
 
+func TestParseAwaitPrompt(t *testing.T) {
+	t.Run("bare AwaitPrompt", func(t *testing.T) {
+		l := lexer.New("AwaitPrompt")
+		p := New(l)
+		cmds := p.Parse()
+
+		if len(cmds) != 1 {
+			t.Fatalf("Expected 1 command, got %d", len(cmds))
+		}
+		if cmds[0].Type != token.AWAIT_PROMPT {
+			t.Errorf("Expected AWAIT_PROMPT, got %s", cmds[0].Type)
+		}
+		if cmds[0].Options != "" {
+			t.Errorf("Expected empty options, got %s", cmds[0].Options)
+		}
+	})
+
+	t.Run("AwaitPrompt with timeout", func(t *testing.T) {
+		l := lexer.New("AwaitPrompt@30s")
+		p := New(l)
+		cmds := p.Parse()
+
+		if len(cmds) != 1 {
+			t.Fatalf("Expected 1 command, got %d", len(cmds))
+		}
+		if cmds[0].Type != token.AWAIT_PROMPT {
+			t.Errorf("Expected AWAIT_PROMPT, got %s", cmds[0].Type)
+		}
+		if cmds[0].Options != "30s" {
+			t.Errorf("Expected options '30s', got %s", cmds[0].Options)
+		}
+	})
+
+	t.Run("AwaitPrompt with millisecond timeout", func(t *testing.T) {
+		l := lexer.New("AwaitPrompt@500ms")
+		p := New(l)
+		cmds := p.Parse()
+
+		if len(cmds) != 1 {
+			t.Fatalf("Expected 1 command, got %d", len(cmds))
+		}
+		if cmds[0].Options != "500ms" {
+			t.Errorf("Expected options '500ms', got %s", cmds[0].Options)
+		}
+	})
+
+	t.Run("AwaitPrompt with minute timeout", func(t *testing.T) {
+		l := lexer.New("AwaitPrompt@2m")
+		p := New(l)
+		cmds := p.Parse()
+
+		if len(cmds) != 1 {
+			t.Fatalf("Expected 1 command, got %d", len(cmds))
+		}
+		if cmds[0].Options != "2m" {
+			t.Errorf("Expected options '2m', got %s", cmds[0].Options)
+		}
+	})
+}
+
 type parseSourceTest struct {
 	tape      string
 	srcTape   string

shell_test.go

@@ -0,0 +1,64 @@
+package main
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestHexToRGB(t *testing.T) {
+	tests := []struct {
+		hex     string
+		r, g, b int
+	}{
+		{"#5B56E0", 91, 86, 224},
+		{"5B56E0", 91, 86, 224},
+		{"#F6821F", 246, 130, 31},
+		{"#000000", 0, 0, 0},
+		{"#FFFFFF", 255, 255, 255},
+		{"", 0, 0, 0},
+		{"#FFF", 0, 0, 0}, // invalid length
+	}
+
+	for _, tc := range tests {
+		r, g, b := hexToRGB(tc.hex)
+		if r != tc.r || g != tc.g || b != tc.b {
+			t.Errorf("hexToRGB(%q) = (%d,%d,%d), want (%d,%d,%d)",
+				tc.hex, r, g, b, tc.r, tc.g, tc.b)
+		}
+	}
+}
+
+func TestShellConfigPromptMarker(t *testing.T) {
+	// Every shell configuration should embed the OSC 7777 prompt marker
+	// so that AwaitPrompt can detect when a command has finished.
+	//
+	// The marker format varies by shell:
+	//   - Most shells: \e]7777;\a (ESC ] 7777 ; BEL)
+	//   - cmd.exe: $E]7777;$E\ (using ST terminator instead of BEL)
+	shellNames := []string{
+		bash,
+		zsh,
+		fish,
+		powershell,
+		pwsh,
+		cmdexe,
+		nushell,
+		osh,
+		xonsh,
+	}
+
+	color := "#5B56E0"
+
+	for _, name := range shellNames {
+		t.Run(name, func(t *testing.T) {
+			env, command := ShellConfig(name, color)
+
+			// Combine env and command into a single string to search
+			combined := strings.Join(env, " ") + " " + strings.Join(command, " ")
+
+			if !strings.Contains(combined, "7777") {
+				t.Errorf("ShellConfig(%q) does not contain OSC 7777 marker.\nenv: %v\ncommand: %v", name, env, command)
+			}
+		})
+	}
+}