Skip to content

Commit db8033c

Browse files
committed
fix: Add cmd/tinybrain directory that was ignored by gitignore
1 parent cd8c6d4 commit db8033c

3 files changed

Lines changed: 1553 additions & 0 deletions

File tree

cmd/tinybrain/cli_test.go

Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
1+
package main
2+
3+
import (
4+
"os"
5+
"testing"
6+
)
7+
8+
func TestMainWithHelp(t *testing.T) {
9+
// Save original args
10+
oldArgs := os.Args
11+
defer func() { os.Args = oldArgs }()
12+
13+
// Test --help doesn't panic
14+
os.Args = []string{"server", "--help"}
15+
// We can't easily test main() since it calls os.Exit
16+
// Instead we test the logic directly
17+
18+
if len(os.Args) > 1 && (os.Args[1] == "--help" || os.Args[1] == "-h" || os.Args[1] == "help") {
19+
// This branch works
20+
t.Log("Help flag detected correctly")
21+
} else {
22+
t.Error("Help flag not detected")
23+
}
24+
}
25+
26+
func TestMainDefaultsToServe(t *testing.T) {
27+
oldArgs := os.Args
28+
defer func() { os.Args = oldArgs }()
29+
30+
os.Args = []string{"server"}
31+
32+
// Test that with no args, we'd add "serve"
33+
if len(os.Args) == 1 {
34+
testArgs := append(os.Args, "serve")
35+
if testArgs[1] != "serve" {
36+
t.Error("Expected serve to be added as default command")
37+
}
38+
}
39+
}
40+
41+
func TestMainWithFlags(t *testing.T) {
42+
oldArgs := os.Args
43+
defer func() { os.Args = oldArgs }()
44+
45+
os.Args = []string{"server", "--http=:9000"}
46+
47+
// Test that flags get prepended with serve
48+
if os.Args[1] != "serve" {
49+
testArgs := append([]string{os.Args[0], "serve"}, os.Args[1:]...)
50+
if testArgs[1] != "serve" || testArgs[2] != "--http=:9000" {
51+
t.Error("Expected flags to be prepended with serve")
52+
}
53+
}
54+
}

0 commit comments

Comments
 (0)