console/command_test.go at master · eidolon/console · 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
package console_test

import (
	"fmt"
	"math/rand"
	"testing"

	"github.com/eidolon/console"
	"github.com/seeruk/assert"
)

func TestCommand(t *testing.T) {
	t.Run("AddCommands()", func(t *testing.T) {
		t.Run("should add all of the given sub-commands", func(t *testing.T) {
			inCommands := []*console.Command{
				{
					Name: fmt.Sprintf("test%d", rand.Int()),
				},
				{
					Name: fmt.Sprintf("test%d", rand.Int()),
				},
			}

			command := console.Command{}
			command.AddCommands(inCommands)

			assert.True(t, len(command.Commands()) == 2, "Expected two commands")

			for i, cmd := range command.Commands() {
				assert.Equal(t, inCommands[i], cmd)
			}
		})
	})

	t.Run("AddCommand()", func(t *testing.T) {
		t.Run("should add a given sub-command", func(t *testing.T) {
			inCommand := &console.Command{
				Name: fmt.Sprintf("test%d", rand.Int()),
			}

			command := console.Command{}
			command.AddCommand(inCommand)

			assert.True(t, len(command.Commands()) == 1, "Expected one command")
			assert.Equal(t, inCommand, command.Commands()[0])
		})
	})

	t.Run("Commands()", func(t *testing.T) {
		t.Run("should return all sub-commands", func(t *testing.T) {
			inCommands := []*console.Command{
				{
					Name: fmt.Sprintf("test%d", rand.Int()),
				},
				{
					Name: fmt.Sprintf("test%d", rand.Int()),
				},
			}

			command := console.Command{}
			command.AddCommands(inCommands)

			assert.True(t, len(command.Commands()) == 2, "Expected two commands")

			for i, cmd := range command.Commands() {
				assert.Equal(t, inCommands[i], cmd)
			}
		})
	})
}