botspec_test.go RawHistoryFolder

1package agent
2
3import (
4	"bytes"
5	"os"
6	"strings"
7	"testing"
8)
9
10func TestSpecForSubcommand(t *testing.T) {
11	if s := specFor("telegram-bot"); s == nil || s.name != "telegram" {
12		t.Fatalf("telegram-bot not matched: %#v", s)
13	}
14	if s := specFor("tg"); s == nil || s.name != "telegram" {
15		t.Fatal("alias tg not matched")
16	}
17	if specFor("nonsense") != nil {
18		t.Fatal("unknown subcommand must not match")
19	}
20}
21
22func TestReadBotTokenDoesNotEchoTerminalInput(t *testing.T) {
23	in, err := os.CreateTemp(t.TempDir(), "stdin")
24	if err != nil {
25		t.Fatal(err)
26	}
27	defer in.Close()
28
29	oldIsTerminal := botTokenIsTerminal
30	oldReadPassword := botTokenReadPassword
31	t.Cleanup(func() {
32		botTokenIsTerminal = oldIsTerminal
33		botTokenReadPassword = oldReadPassword
34	})
35	botTokenIsTerminal = func(fd int) bool {
36		if fd != int(in.Fd()) {
37			t.Fatalf("terminal check fd = %d, want %d", fd, in.Fd())
38		}
39		return true
40	}
41	botTokenReadPassword = func(fd int) ([]byte, error) {
42		if fd != int(in.Fd()) {
43			t.Fatalf("password read fd = %d, want %d", fd, in.Fd())
44		}
45		return []byte("secret-token"), nil
46	}
47
48	var out bytes.Buffer
49	got, err := readBotToken(in, &out)
50	if err != nil {
51		t.Fatalf("readBotToken() error = %v", err)
52	}
53	if got != "secret-token" {
54		t.Fatalf("readBotToken() = %q, want secret token", got)
55	}
56	if strings.Contains(out.String(), got) {
57		t.Fatalf("terminal output exposed bot token: %q", out.String())
58	}
59}