core/cli_test.go at main · dAppCore/core · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package core_test

import (
	"bytes"
	"testing"

	. "dappco.re/go/core"
	"github.com/stretchr/testify/assert"
)

// --- Cli Surface ---

func TestCli_Good(t *testing.T) {
	c := New()
	assert.NotNil(t, c.Cli())
}

func TestCli_Banner_Good(t *testing.T) {
	c := New(WithOption("name", "myapp"))
	assert.Equal(t, "myapp", c.Cli().Banner())
}

func TestCli_SetBanner_Good(t *testing.T) {
	c := New()
	c.Cli().SetBanner(func(_ *Cli) string { return "Custom Banner" })
	assert.Equal(t, "Custom Banner", c.Cli().Banner())
}

func TestCli_Run_Good(t *testing.T) {
	c := New()
	executed := false
	c.Command("hello", Command{Action: func(_ Options) Result {
		executed = true
		return Result{Value: "world", OK: true}
	}})
	r := c.Cli().Run("hello")
	assert.True(t, r.OK)
	assert.Equal(t, "world", r.Value)
	assert.True(t, executed)
}

func TestCli_Run_Nested_Good(t *testing.T) {
	c := New()
	executed := false
	c.Command("deploy/to/homelab", Command{Action: func(_ Options) Result {
		executed = true
		return Result{OK: true}
	}})
	r := c.Cli().Run("deploy", "to", "homelab")
	assert.True(t, r.OK)
	assert.True(t, executed)
}

func TestCli_Run_WithFlags_Good(t *testing.T) {
	c := New()
	var received Options
	c.Command("serve", Command{Action: func(opts Options) Result {
		received = opts
		return Result{OK: true}
	}})
	c.Cli().Run("serve", "--port=8080", "--debug")
	assert.Equal(t, "8080", received.String("port"))
	assert.True(t, received.Bool("debug"))
}

func TestCli_Run_NoCommand_Good(t *testing.T) {
	c := New()
	r := c.Cli().Run()
	assert.False(t, r.OK)
}

func TestCli_PrintHelp_Good(t *testing.T) {
	c := New(WithOption("name", "myapp"))
	c.Command("deploy", Command{Action: func(_ Options) Result { return Result{OK: true} }})
	c.Command("serve", Command{Action: func(_ Options) Result { return Result{OK: true} }})
	c.Cli().PrintHelp()
}

func TestCli_SetOutput_Good(t *testing.T) {
	c := New()
	var buf bytes.Buffer
	c.Cli().SetOutput(&buf)
	c.Cli().Print("hello %s", "world")
	assert.Contains(t, buf.String(), "hello world")
}